cutlass/examples/40_cutlass_py/README.md

24 lines
903 B
Markdown
Raw Normal View History

2022-11-19 22:02:15 +08:00
# CUTLASS Python Interface Examples
This directory contains examples of using CUTLASS's Python interface. It consists of two types of examples:
* _Basic examples_: minimal examples that illustrate how to set up GEMMs, convolutions, and grouped GEMM operations
* [_Customizable examples_](customizable): examples that allow one to specify a variety of template parameters for the given kernel
>>>>>>> Add simplified examples
2022-11-19 22:02:15 +08:00
## Setting up the Python interface
Please follow the instructions [here](/tools/library/scripts/pycutlass/README.md#installation) to set up the Python API.
2022-11-19 22:02:15 +08:00
## Running examples
Each of the basic examples can be run as follows:
```shell
2022-11-19 22:02:15 +08:00
# Run the GEMM example
python gemm.py
2022-11-19 22:02:15 +08:00
# Run the Conv2d example
python conv2d.py
2022-11-19 22:02:15 +08:00
# Run the grouped GEMM example
python gemm_grouped.py
```
2022-11-19 22:02:15 +08:00
To run the customizable examples, refer to the README in the [customizable](customizable) directory.