Contents Menu Expand Light mode Dark mode Auto light/dark mode
CUTLASS Python
Light Logo Dark Logo
CUTLASS Python
  • Home

Getting Started:

  • Installation
  • Getting Started
  • Contributing

Python Documentation:

  • CUTLASS Python API
    • CUTLASS
      • Emitters
      • Operations
      • Utilities

Examples and Tutorials:

  • Examples
    • Basic GEMM
    • Epilogue
    • PyTorch Extension

Reference:

  • Github
Back to top

Examples#

  • Basic GEMM
    • Declaring and running a GEMM
    • Changing operation modes
    • Running cached kernels
    • Running non-default GEMMs
    • Handling errors
  • Epilogue
    • Run a GEMM with an identity activation function
    • Run a GEMM with a ReLU element-wise activation function
    • Other element-wise activation functions
  • PyTorch Extension
    • Background on grouped GEMM
    • Declaring a grouped GEMM via the CUTLASS Python interface
    • Exporting the CUTLASS kernel to a PyTorch CUDA extension
Next
Example of using elementwise activation functions in the CUTLASS Python interface
Previous
Utilities
Copyright © 2023, NVIDIA
Made with Sphinx and @pradyunsg's Furo