squall/cutlass
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity
e36912f961
cutlass/examples/python
History
ANIKET SHIVAM d572cc1aab
CUTLASS 3.1 (#915) 
Co-authored-by: Aniket Shivam <ashivam@nvidia.com>
2023-04-14 23:19:34 -04:00
..
00_basic_gemm.ipynb CUTLASS 3.1 (#915) 2023-04-14 23:19:34 -04:00
01_epilogue.ipynb CUTLASS 3.1 (#915) 2023-04-14 23:19:34 -04:00
02_pytorch_extension_grouped_gemm.ipynb CUTLASS 3.1 (#915) 2023-04-14 23:19:34 -04:00
README.md CUTLASS 3.1 (#915) 2023-04-14 23:19:34 -04:00

README.md

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.