tensorflow / tensorflow/probability
XlaRuntimeError: STS Seasonal component does not work under tfp.substrates.jax.sts
Nobody has claimed this yet.
- Dominant language
- Jupyter Notebook
- Stars
- 4.4k
- Forks
- 1.1k
- PR merge metrics
- No merged PRs in 30d
Description
When running the example in the following blog:
and replace tfp to tfp.substrates.jax and its submodules. The code fails when instantiating the Seasonal component in sts in the co2 example:
seasonal = tfp.sts.Seasonal(
num_seasons=12, observed_time_series=observed_time_series)
Error message:
XlaRuntimeError: INTERNAL: Attempting to perform BLAS operation using StreamExecutor without BLAS support
Environment: Python 3.8 on Ubuntu 22.04 LTS (I also ran this code in colab, as well as with python 3.9 in MacBook m1, and this error message does not raise in both situations)
@murphyk
Contributor guide
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
Research direction
Start with the Structural_Time_Series_Modeling_Case_Studies_Atmospheric_CO2_and_Electricity_Demand.ipynb example and reproduce the failure at the tfp.substrates.jax Seasonal component instantiation. Compare the reported Ubuntu 22.04 environment with the working Colab and MacBook M1 environments. Done means the CO2 example can instantiate Seasonal without the XlaRuntimeError.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- jupyter-notebook, python, tensorflow
- Domain
- machine-learning
- Issue type
- Bug
- Difficulty
- 4/5
- Estimated time
- 3-5 days
- Activity status
- Stale
- Clarity
- Needs clarification
- Newbie friendliness
- 30/100