search

pytorch / xla

Enabling PyTorch on XLA Devices (e.g. Google TPU)
https://pytorch.org/xla
Other
2.46k stars 467 forks source link

Cache flash attention tracing #8026

Closed JackCaoG closed 1 week ago

JackCaoG commented 1 week ago

need to add more tests.