├── .gitignore ├── 01_gpu_basics.ipynb ├── 02_vector_addition.ipynb ├── 03_small_matrix_multiplication.ipynb ├── 04_softmax_fwd_bwd.ipynb ├── 05_block_matmul.ipynb ├── 06_matmul_fwd_bwd.ipynb ├── README.md ├── images ├── comp_graph.png ├── gpu_threads.png ├── grid-of-clusters.png ├── grid-of-thread-blocks.png ├── grouped_vs_row_major_ordering.png └── memory-hierarchy.png ├── poetry.lock └── pyproject.toml /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VikParuchuri/triton_tutorial/HEAD/.gitignore -------------------------------------------------------------------------------- /01_gpu_basics.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VikParuchuri/triton_tutorial/HEAD/01_gpu_basics.ipynb -------------------------------------------------------------------------------- /02_vector_addition.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VikParuchuri/triton_tutorial/HEAD/02_vector_addition.ipynb -------------------------------------------------------------------------------- /03_small_matrix_multiplication.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VikParuchuri/triton_tutorial/HEAD/03_small_matrix_multiplication.ipynb -------------------------------------------------------------------------------- /04_softmax_fwd_bwd.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VikParuchuri/triton_tutorial/HEAD/04_softmax_fwd_bwd.ipynb -------------------------------------------------------------------------------- /05_block_matmul.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VikParuchuri/triton_tutorial/HEAD/05_block_matmul.ipynb -------------------------------------------------------------------------------- /06_matmul_fwd_bwd.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VikParuchuri/triton_tutorial/HEAD/06_matmul_fwd_bwd.ipynb -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VikParuchuri/triton_tutorial/HEAD/README.md -------------------------------------------------------------------------------- /images/comp_graph.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VikParuchuri/triton_tutorial/HEAD/images/comp_graph.png -------------------------------------------------------------------------------- /images/gpu_threads.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VikParuchuri/triton_tutorial/HEAD/images/gpu_threads.png -------------------------------------------------------------------------------- /images/grid-of-clusters.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VikParuchuri/triton_tutorial/HEAD/images/grid-of-clusters.png -------------------------------------------------------------------------------- /images/grid-of-thread-blocks.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VikParuchuri/triton_tutorial/HEAD/images/grid-of-thread-blocks.png -------------------------------------------------------------------------------- /images/grouped_vs_row_major_ordering.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VikParuchuri/triton_tutorial/HEAD/images/grouped_vs_row_major_ordering.png -------------------------------------------------------------------------------- /images/memory-hierarchy.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VikParuchuri/triton_tutorial/HEAD/images/memory-hierarchy.png -------------------------------------------------------------------------------- /poetry.lock: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VikParuchuri/triton_tutorial/HEAD/poetry.lock -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VikParuchuri/triton_tutorial/HEAD/pyproject.toml --------------------------------------------------------------------------------