Chris Jones 38d52a19ef [mosaic_gpu] Force flush all cupti activity, then unsubscribe. ...

With default flushing, it is possible for events to be missed. We should only unsubscribe after we are finished with cupti.

PiperOrigin-RevId: 737939327

2025-03-18 03:35:03 -07: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] Force flush all cupti activity, then unsubscribe.

2025-03-18 03:35:03 -07: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