This website requires JavaScript.
Explore
Help
Sign In
mirrors
/
rocm_jax
Watch
1
Star
0
Fork
0
You've already forked rocm_jax
mirror of
https://github.com/ROCm/jax.git
synced
2025-04-16 11:56:07 +00:00
Code
Issues
Packages
Projects
Releases
Wiki
Activity
rocm_jax
/
jaxlib
/
mosaic
History
jax authors
a13b3cedad
Merge pull request
#26691
from h-vetinari:packed
...
PiperOrigin-RevId: 733696873
2025-03-05 05:46:01 -08:00
..
dialect
[Mosaic GPU] Infer layouts (transforms) on memrefs that directly feed into the dialect wgmma op.
2025-03-02 03:17:13 -08:00
gpu
fix member access to packed CUDA struct
2025-02-24 08:03:07 +11:00
python
Fix OSS build for the Mosaic GPU dialect
2025-01-02 15:55:03 +00:00
BUILD
[Mosaic] Rename dep name.
2025-03-03 11:01:25 -08:00
serde.cc
[mosaic] Extracted serialization pass traversal logic into a reusable function
2025-01-16 02:58:06 -08:00
serde.h
[mosaic] Extracted serialization pass traversal logic into a reusable function
2025-01-16 02:58:06 -08:00
