Peter Hawkins 90d8f37863 Rename pybind_extension to nanobind_extension. ...

We have no remaining uses of pybind11 outside a GPU custom call example.

PiperOrigin-RevId: 712608834

2025-01-06 11:53:44 -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

Rename pybind_extension to nanobind_extension.

2025-01-06 11:53:44 -08:00

custom_call.cc

[Mosaic GPU] Add end-to-end lowering example for a pointwise kernel using the dialect and layout inference.

2024-12-17 09:28:05 -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] 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