├── .pre-commit-config.yaml ├── README.md ├── compile └── pickable.py ├── ops ├── argmax.py ├── binary-search.py ├── convert-shuffle.py ├── float64-log.py ├── full.py ├── gather.py ├── gnn.py ├── indirect.py ├── int64-mod.py ├── load-2d.py ├── permute1d.py ├── permute2d-col.py ├── permute2d-row.py ├── scan-2d.py ├── sum-2d.py └── transpose-2d.py ├── proton ├── exp-aggr-profiles.py ├── golden.hatchet ├── overview.ipynb ├── triton-1.hatchet └── triton-2.hatchet └── pyproject.toml /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Deep-Learning-Profiling-Tools/triton-samples/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- 1 | # triton-samples -------------------------------------------------------------------------------- /compile/pickable.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Deep-Learning-Profiling-Tools/triton-samples/HEAD/compile/pickable.py -------------------------------------------------------------------------------- /ops/argmax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Deep-Learning-Profiling-Tools/triton-samples/HEAD/ops/argmax.py -------------------------------------------------------------------------------- /ops/binary-search.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Deep-Learning-Profiling-Tools/triton-samples/HEAD/ops/binary-search.py -------------------------------------------------------------------------------- /ops/convert-shuffle.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Deep-Learning-Profiling-Tools/triton-samples/HEAD/ops/convert-shuffle.py -------------------------------------------------------------------------------- /ops/float64-log.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Deep-Learning-Profiling-Tools/triton-samples/HEAD/ops/float64-log.py -------------------------------------------------------------------------------- /ops/full.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Deep-Learning-Profiling-Tools/triton-samples/HEAD/ops/full.py -------------------------------------------------------------------------------- /ops/gather.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Deep-Learning-Profiling-Tools/triton-samples/HEAD/ops/gather.py -------------------------------------------------------------------------------- /ops/gnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Deep-Learning-Profiling-Tools/triton-samples/HEAD/ops/gnn.py -------------------------------------------------------------------------------- /ops/indirect.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Deep-Learning-Profiling-Tools/triton-samples/HEAD/ops/indirect.py -------------------------------------------------------------------------------- /ops/int64-mod.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Deep-Learning-Profiling-Tools/triton-samples/HEAD/ops/int64-mod.py -------------------------------------------------------------------------------- /ops/load-2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Deep-Learning-Profiling-Tools/triton-samples/HEAD/ops/load-2d.py -------------------------------------------------------------------------------- /ops/permute1d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Deep-Learning-Profiling-Tools/triton-samples/HEAD/ops/permute1d.py -------------------------------------------------------------------------------- /ops/permute2d-col.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Deep-Learning-Profiling-Tools/triton-samples/HEAD/ops/permute2d-col.py -------------------------------------------------------------------------------- /ops/permute2d-row.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Deep-Learning-Profiling-Tools/triton-samples/HEAD/ops/permute2d-row.py -------------------------------------------------------------------------------- /ops/scan-2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Deep-Learning-Profiling-Tools/triton-samples/HEAD/ops/scan-2d.py -------------------------------------------------------------------------------- /ops/sum-2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Deep-Learning-Profiling-Tools/triton-samples/HEAD/ops/sum-2d.py -------------------------------------------------------------------------------- /ops/transpose-2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Deep-Learning-Profiling-Tools/triton-samples/HEAD/ops/transpose-2d.py -------------------------------------------------------------------------------- /proton/exp-aggr-profiles.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Deep-Learning-Profiling-Tools/triton-samples/HEAD/proton/exp-aggr-profiles.py -------------------------------------------------------------------------------- /proton/golden.hatchet: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Deep-Learning-Profiling-Tools/triton-samples/HEAD/proton/golden.hatchet -------------------------------------------------------------------------------- /proton/overview.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Deep-Learning-Profiling-Tools/triton-samples/HEAD/proton/overview.ipynb -------------------------------------------------------------------------------- /proton/triton-1.hatchet: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Deep-Learning-Profiling-Tools/triton-samples/HEAD/proton/triton-1.hatchet -------------------------------------------------------------------------------- /proton/triton-2.hatchet: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Deep-Learning-Profiling-Tools/triton-samples/HEAD/proton/triton-2.hatchet -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Deep-Learning-Profiling-Tools/triton-samples/HEAD/pyproject.toml --------------------------------------------------------------------------------