mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-15 19:36:06 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/jaxlib/mosaic/gpu
History
jax authors f3b7c5cb9e Integrate LLVM at llvm/llvm-project@0230d63b4a 
Updates LLVM usage to match
[0230d63b4a8b](https://github.com/llvm/llvm-project/commit/0230d63b4a8b)

PiperOrigin-RevId: 738222096
2025-03-18 19:23:20 -07:00
..
integrations/c
[mosaic_gpu] Added a serialization pass
2025-01-17 03:12:51 -08:00
BUILD
[Mosaic GPU] Fuse slicing into s4 -> bf16 upcasts
2025-03-18 05:38:49 -07:00
custom_call.cc
Integrate LLVM at llvm/llvm-project@0230d63b4a
2025-03-18 19:23:20 -07: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] Force flush all cupti activity, then unsubscribe.
2025-03-18 03:35:03 -07:00
passes.cc
[Mosaic GPU] Fuse slicing into s4 -> bf16 upcasts
2025-03-18 05:38:49 -07: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