-
Originally reported as google/jax#20184 and google/jax#16008:
> ### Description
> When inspecting the estimated flop count of a compiled function, dot_general, einsum, '@', jnp.dot, etc show "-1.0…
-
## TPU initialization is failed
When I started tpu v6e-4 TPU Vm with v2-alpha-tpuv6e base image, with pip enviroment and xla updates I can clearly initialized tpus. However when I start to dockeriz…
-
## ❓ Questions and Help
I'm working on adding initial XLA support to ComfyUI https://github.com/comfyanonymous/ComfyUI/pull/5657 and would greatly appreciate any feedback or validation from the commu…
-
## 🐛 Bug
##
After updating XLA pin from 32ebd694c4d0442e241d76324ff1a721831366b4 to 590cd6fcd1ed24ab9cf494789a0fc524b94a4a6a in PR https://github.com/pytorch/xla/pull/8079/files
Our CI h…
qihqi updated
1 month ago
-
## ❓ Questions and Help
Hi, I recieved loss None when training model. Anyone can help?
Simple reproduct kaggle notebook [link](https://www.kaggle.com/code/liondude/notebook548442067d)
```
im…
-
The test is the following:
```julia
x = ConcreteRArray([1.0 + 2.0im, 8.0 + 16.0im, 27.0 + 54.0im, 64.0 + 128.0im])
@test Array(x) .^ (1//3) ≈ @jit Ops.cbrt(x)
```
and yields the following error: htt…
-
### Have I written custom code (as opposed to using a stock example script provided in MediaPipe)
None
### OS Platform and Distribution
Linnux Ubuntu
### MediaPipe Tasks SDK version
0.1…
-
## ❓ Questions and Help
I see that there are docs for triton support but only for GPU? Is it possible for TPU to use triton?
```[tasklist]
### Tasks
```
-
Click to expand!
### Issue Type
Performance
### Source
binary
### Tensorflow Version
2.7.0-rc0
### Custom Code
No
### OS Platform and Distribution
Linux Ubuntu 20.04
### Mobile device
_…
-
## 🐛 Bug
I encountered some correctness issues while using Dynamo + OpenXLA. After investigation, I found that Torch-XLA currently generates the same hash for two different graphs when computing th…