-
**Description:**
While using `keras.layers.Embedding` with TensorFlow backend in Keras 3, I encountered issues with XLA optimization during multi-GPU training, leading to `InvalidArgumentError`. This…
-
2024-04-17 11:52:43.470077: E external/local_xla/xla/stream_executor/cuda/cuda_dnn.cc:9261] Unable to register cuDNN factory: Attempting to register factory for plugin cuDNN when one has already been …
-
## 🐛 Bug
When functionalization is on (XLA_DISABLE_FUNCTIONALIZATION=0), I see that there are fewer aliased tensors. Jack has a patch to increase the number of aliased tensors https://github.com/py…
-
### Anything you want to discuss about vllm.
Hi,
We have adapt our harware to pytorch/xla, what should we do to run vllm via pytorch/xla?
-
Following the previous issue (https://github.com/chunmeifeng/T2Net/issues/1) updates. I am faced with the above mentioned runtime error. This happens when I run the code in the GPU runtime. The stack-…
-
## 🐛 Bug Report
### Description
Running the following code results in the error:
```
Status: INVALID_ARGUMENT: The parameter of condition and body, the result of the body, and init must all have…
-
### Issue type
Bug
### Have you reproduced the bug with TensorFlow Nightly?
No
### Source
source
### TensorFlow version
v2.17.0
### Custom code
No
### OS platform and distribution
Ubuntu 22…
-
## Description
We need to implement a testing framework for the TensorFlow XLA (Accelerated Linear Algebra) compiler model. This framework should comprehensively test the XLA compiler, ensuring it …
-
pip install torch_xla[tpu]~=2.3.0 -f https://storage.googleapis.com/libtpu-releases/index.html
ERROR: Could not find a version that satisfies the requirement torch_xla~=2.3.0 (from versions: none)
…
-
Hi, thank you so much for releasing this wonderful codebase. When I'm trying to run pretrain_llama_7b on some v3-tpu pod, I got this error:
```
ERROR: Accessing retired flag 'jax_enable_async_collec…