cutlass/examples/python

Examples of using the CUTLASS Python interface

• 00_basic_gemm

  Shows how declare, configure, compile, and run a CUTLASS GEMM using the Python interface

• 01_epilogue

  Shows how to fuse elementwise activation functions to GEMMs via the Python interface

• 02_pytorch_extension_grouped_gemm

  Shows how to declare, compile, and run a grouped GEMM operation via the Python interface, along with how the emitted kernel can be easily exported to a PyTorch CUDA extension.

• 03_basic_conv2d

  Shows how to declare, configure, compile, and run a CUTLASS Conv2d using the Python interface

• 04_epilogue_visitor

  Shows how to fuse elementwise activation functions to GEMMs via the Python Epilogue Visitor interface