-
StableHLO to TTIR conversion fails to legalize `transpose` op for 3 and higher dimensional tensors. A sample test case can be found in [tt-xla PR](https://github.com/tenstorrent/tt-xla/pull/15)
-
## ❓ Questions and Help
I'm running this official [script here](https://github.com/pytorch/xla/blob/master/test/test_train_mp_imagenet_fsdp.py), but I only see two xla devices being used, xla:0 and…
-
### Description
I sometimes, but not consistently, get the following jaxlib.xla_extension.XlaRuntimeError when training a neural network with sine activations in JAX on an NVIDIA A100 GPU:
```
jax…
-
## Summary of Contributions (9th Feb)
1) **Improve the number of models in TorchBench that work with Dynamo as a tracer:** These passing rates are now comparable to those from torch.compile using I…
-
This is the error I get after running a simple tensorflow command:
`In [3]: import tensorflow as tf
In [4]: print(tf.reduce_sum(tf.random.normal([1000, 1000])))
2021-02-17 08:10:00.365011: I tensor…
-
Hi All,
I'm having some trouble running N2V. I have a computer with NVIDIA RTXx A5000 and Ubuntu 18.04.
I use conda to install N2V according to:
`$ conda create -n 'n2v' python=3.7`
`$ s…
-
Hi,
I noticed a problem with the time grid in euler_sample when XLA compilation (with dynamic shape for times) is used.
https://colab.research.google.com/drive/1kg8RChmJ3TxdONBHE1lIdCAmZK…
-
## 🐛 Bug
```
File "/home/kojoe/EasyAnimate/easyanimate/pipeline/pipeline_easyanimate_inpaint.py", line 1369, in __call__
latent_model_input = xs.mark_sharding(
File "/home/kojoe/.local/lib…
-
## 🐛 Bug
Maxpool operator from xla2 crashes
## To Reproduce
1. download a mnist toy example I prepared from [here](https://gist.githubusercontent.com/Chaosruler972/c8d34310b4a9d892443f81dd5bf…
-
## Description
XLA is an abstraction layer of computation graph for better efficiency, consistency, portability, and a lot as [they claim](https://www.tensorflow.org/xla/architecture). However, the m…