Sergei Lebedev a14e6968bf [mosaic] Migrated the serialization pass from codegen to pass_boilerplate.h ...

This prepares teh generalization of the serialization pass to handle both
Mosaic TPU and GPU.

PiperOrigin-RevId: 705628923

2024-12-12 14:19:36 -08: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

[mosaic] Migrated the serialization pass from codegen to pass_boilerplate.h

2024-12-12 14:19:36 -08:00

custom_call.cc

[Mosaic GPU] Improve default kernel name and add option to customize

2024-12-02 22:22:11 -05: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] Add CUPTI profiler alongside events-based implementation

2024-12-09 14:31:20 -05: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

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