├── .clang-format ├── LICENSE ├── Makefile ├── README.md ├── benchmark.py ├── benchmark.sh ├── denoise-h100.sh ├── gemm.cu ├── images ├── perf.png ├── pingpong.png ├── stmatrix_layout_funny.png ├── stmatrix_script.png ├── tma_swizzle.png ├── tma_swizzle_old.png └── wgmma_layout.png ├── l2.py ├── layout.py ├── main.cu ├── maxreg.cu ├── op.cpp ├── pingpong.cu ├── setup.py ├── stmatrix.cu ├── stmatrix.py └── test.py /.clang-format: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bertmaher/simplegemm/HEAD/.clang-format -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bertmaher/simplegemm/HEAD/LICENSE -------------------------------------------------------------------------------- /Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bertmaher/simplegemm/HEAD/Makefile -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bertmaher/simplegemm/HEAD/README.md -------------------------------------------------------------------------------- /benchmark.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bertmaher/simplegemm/HEAD/benchmark.py -------------------------------------------------------------------------------- /benchmark.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bertmaher/simplegemm/HEAD/benchmark.sh -------------------------------------------------------------------------------- /denoise-h100.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bertmaher/simplegemm/HEAD/denoise-h100.sh -------------------------------------------------------------------------------- /gemm.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bertmaher/simplegemm/HEAD/gemm.cu -------------------------------------------------------------------------------- /images/perf.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bertmaher/simplegemm/HEAD/images/perf.png -------------------------------------------------------------------------------- /images/pingpong.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bertmaher/simplegemm/HEAD/images/pingpong.png -------------------------------------------------------------------------------- /images/stmatrix_layout_funny.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bertmaher/simplegemm/HEAD/images/stmatrix_layout_funny.png -------------------------------------------------------------------------------- /images/stmatrix_script.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bertmaher/simplegemm/HEAD/images/stmatrix_script.png -------------------------------------------------------------------------------- /images/tma_swizzle.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bertmaher/simplegemm/HEAD/images/tma_swizzle.png -------------------------------------------------------------------------------- /images/tma_swizzle_old.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bertmaher/simplegemm/HEAD/images/tma_swizzle_old.png -------------------------------------------------------------------------------- /images/wgmma_layout.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bertmaher/simplegemm/HEAD/images/wgmma_layout.png -------------------------------------------------------------------------------- /l2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bertmaher/simplegemm/HEAD/l2.py -------------------------------------------------------------------------------- /layout.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bertmaher/simplegemm/HEAD/layout.py -------------------------------------------------------------------------------- /main.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bertmaher/simplegemm/HEAD/main.cu -------------------------------------------------------------------------------- /maxreg.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bertmaher/simplegemm/HEAD/maxreg.cu -------------------------------------------------------------------------------- /op.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bertmaher/simplegemm/HEAD/op.cpp -------------------------------------------------------------------------------- /pingpong.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bertmaher/simplegemm/HEAD/pingpong.cu -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bertmaher/simplegemm/HEAD/setup.py -------------------------------------------------------------------------------- /stmatrix.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bertmaher/simplegemm/HEAD/stmatrix.cu -------------------------------------------------------------------------------- /stmatrix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bertmaher/simplegemm/HEAD/stmatrix.py -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bertmaher/simplegemm/HEAD/test.py --------------------------------------------------------------------------------