Chris Jones de8fd3b00d [mosaic:gpu] Fix MLIR canonicalization pass region-simplify option. ...

`region-simplify` now has `normal` and `aggressive` modes (using `normal` for now).

PiperOrigin-RevId: 644724434

2024-06-19 06:02:11 -07:00

..

integrations/c

Switch Mosaic GPU to a custom pass pipeline and improve the lowering of GPU launch

2024-04-24 03:27:45 -07:00

BUILD

Use constant memory to pass in TMA descriptors to the kernel

2024-06-17 05:31:26 -07:00

custom_call.cc

[mosaic:gpu] Fix MLIR canonicalization pass region-simplify option.

2024-06-19 06:02:11 -07:00

launch_lowering.cc

Use constant memory to pass in TMA descriptors to the kernel

2024-06-17 05:31:26 -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] Integrate the ExecutionEngine with the jaxlib GPU plugin

2024-05-30 01:46:23 -07:00

pass_boilerplate.h

[Mosaic GPU] Load LLVM lowering interfaces for all dialects

2024-06-10 05:55:01 -07:00

passes.cc

Integrate LLVM at llvm/llvm-project@8c5d9c79b9

2024-06-11 12:24:43 -07:00

passes.h

[Mosaic GPU] Load LLVM lowering interfaces for all dialects

2024-06-10 05:55:01 -07:00

runtime.cc

[Mosaic GPU] Stop using the MLIR CUDA runtime

2024-04-29 08:04:51 -07:00