Sergei Lebedev d34c40f6b6 [mosaic_gpu] Added a serialization pass ...

The pass adds versioning to the Mosaic GPU IR in the lowered custom calls
and can apply forward/backward migration rules. Currently, no rules are
necessary since we are at version 1.

PiperOrigin-RevId: 716596848

2025-01-17 03:12:51 -08:00

..

integrations/c

[mosaic_gpu] Added a serialization pass

2025-01-17 03:12:51 -08:00

BUILD

[mosaic_gpu] Added a serialization pass

2025-01-17 03:12:51 -08:00

custom_call.cc

[mosaic_gpu] Added a serialization pass

2025-01-17 03:12:51 -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] Added a serialization pass

2025-01-17 03:12:51 -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

Added another boxDim check to mosaic_gpu_init_tma_desc

2024-08-07 03:16:54 -07: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