├── .gitignore ├── .gitmodules ├── .pre-commit-config.yaml ├── LICENSE ├── README.md ├── cutlass_viz ├── __init__.py ├── __main__.py ├── core.py └── layout.py ├── examples ├── layout_algebra.py ├── mma_sm70_f16.cc ├── mma_sm89_fp8.cc ├── mma_sm90_fp8.cc ├── swizzle_layout.cc └── tiled_copy.cc └── setup.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/flashinfer-ai/cutlass-viz/HEAD/.gitignore -------------------------------------------------------------------------------- /.gitmodules: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/flashinfer-ai/cutlass-viz/HEAD/.gitmodules -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/flashinfer-ai/cutlass-viz/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/flashinfer-ai/cutlass-viz/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/flashinfer-ai/cutlass-viz/HEAD/README.md -------------------------------------------------------------------------------- /cutlass_viz/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /cutlass_viz/__main__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/flashinfer-ai/cutlass-viz/HEAD/cutlass_viz/__main__.py -------------------------------------------------------------------------------- /cutlass_viz/core.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/flashinfer-ai/cutlass-viz/HEAD/cutlass_viz/core.py -------------------------------------------------------------------------------- /cutlass_viz/layout.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/flashinfer-ai/cutlass-viz/HEAD/cutlass_viz/layout.py -------------------------------------------------------------------------------- /examples/layout_algebra.py: -------------------------------------------------------------------------------- 1 | from pycute import * 2 | 3 | print(complement(Layout(4, 1), 24)) 4 | -------------------------------------------------------------------------------- /examples/mma_sm70_f16.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/flashinfer-ai/cutlass-viz/HEAD/examples/mma_sm70_f16.cc -------------------------------------------------------------------------------- /examples/mma_sm89_fp8.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/flashinfer-ai/cutlass-viz/HEAD/examples/mma_sm89_fp8.cc -------------------------------------------------------------------------------- /examples/mma_sm90_fp8.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/flashinfer-ai/cutlass-viz/HEAD/examples/mma_sm90_fp8.cc -------------------------------------------------------------------------------- /examples/swizzle_layout.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/flashinfer-ai/cutlass-viz/HEAD/examples/swizzle_layout.cc -------------------------------------------------------------------------------- /examples/tiled_copy.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/flashinfer-ai/cutlass-viz/HEAD/examples/tiled_copy.cc -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/flashinfer-ai/cutlass-viz/HEAD/setup.py --------------------------------------------------------------------------------