mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-19 05:16:06 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/jaxlib/mosaic/gpu
History
jax authors 725087e13f Integrate LLVM at llvm/llvm-project@9d24f94379 
Updates LLVM usage to match
[9d24f9437944](https://github.com/llvm/llvm-project/commit/9d24f9437944)

PiperOrigin-RevId: 728265165
2025-02-18 10:30:48 -08:00
..
integrations/c
[mosaic_gpu] Added a serialization pass
2025-01-17 03:12:51 -08:00
BUILD
Integrate LLVM at llvm/llvm-project@912b154f3a
2025-02-17 10:08:37 -08:00
custom_call.cc
Integrate LLVM at llvm/llvm-project@9d24f94379
2025-02-18 10:30:48 -08:00
launch_lowering.cc
[Mosaic GPU] Strengthen cluster-related tests by covering more cluster shapes
2024-07-24 13:43:52 -07:00
launch_lowering.h
Switch Mosaic GPU to a custom pass pipeline and improve the lowering of GPU launch
2024-04-24 03:27:45 -07:00
mosaic_gpu_ext.cc
[mosaic_gpu] Fixed mosaic_gpu-serde pass registration
2025-01-24 06:35:54 -08:00
passes.cc
[mosaic] Migrated the serialization pass from codegen to pass_boilerplate.h
2024-12-12 14:19:36 -08:00
passes.h
[Mosaic GPU] Pass in TMA descriptors through kernel parameters
2024-07-02 09:30:52 -07:00
runtime.cc
[Mosaic GPU] Handle the swizzle attribute in the lowering of async_store and async_load
2025-01-27 05:18:16 -08:00
serde.cc
[mosaic_gpu] Added a serialization pass
2025-01-17 03:12:51 -08:00
serde.h
[mosaic_gpu] Added a serialization pass
2025-01-17 03:12:51 -08:00
target.cc
[Mosaic GPU] Query SM and PTX ISA dynamically using driver and LLVM
2024-10-24 01:46:29 -07:00
target.h
[Mosaic GPU] Query SM and PTX ISA dynamically using driver and LLVM
2024-10-24 01:46:29 -07:00