├── .gitignore ├── LICENSE ├── README.md └── problems ├── amd.yaml ├── amd ├── eval.py ├── fp8-mm │ ├── reference.py │ ├── submission.py │ ├── task.py │ ├── task.yml │ ├── template-hip.py │ └── template.py ├── identity │ ├── reference.py │ ├── submission.py │ ├── task.py │ ├── task.yml │ └── template.py ├── mla-decode │ ├── README.md │ ├── eval.py │ ├── reference.py │ ├── submission.py │ ├── task.py │ ├── task.yml │ └── utils.py ├── moe │ ├── reference.py │ ├── submission.py │ ├── task.py │ └── task.yml └── utils.py ├── amd_distributed.yaml ├── amd_distributed ├── ag-gemm │ ├── reference.py │ ├── submission.py │ ├── task.py │ └── task.yml ├── all2all │ ├── reference.py │ ├── submission.py │ ├── task.py │ └── task.yml ├── eval.py ├── gemm-rs │ ├── reference.py │ ├── submission.py │ ├── task.py │ └── task.yml ├── rocshmem_example.py └── utils.py ├── beta.yaml ├── bioml.yaml ├── bioml └── trimul │ ├── eval.py │ ├── reference.py │ ├── submission.py │ ├── task.py │ ├── task.yml │ └── utils.py ├── nvidia.yaml ├── nvidia ├── eval.py ├── eval_better_bench.py ├── nvfp4_gemm │ ├── eval.py │ ├── reference.py │ ├── submission.py │ ├── task.py │ ├── task.yml │ ├── template.py │ └── utils.py ├── nvfp4_gemv │ ├── reference.py │ ├── submission.py │ ├── task.py │ ├── task.yml │ ├── template.py │ └── template_cute.py └── utils.py ├── pmpp ├── conv2d_py │ ├── reference.py │ ├── solutions │ │ ├── correct │ │ │ └── ref.py │ │ └── wrong │ │ │ └── empty.py │ ├── submission.py │ ├── task.py │ └── task.yml ├── eval.py ├── grayscale_py │ ├── reference.py │ ├── solutions │ │ ├── correct │ │ │ └── ref.py │ │ └── wrong │ │ │ └── empty.py │ ├── submission.py │ ├── task.py │ └── task.yml ├── histogram_py │ ├── reference.py │ ├── solutions │ │ ├── correct │ │ │ └── ref.py │ │ └── wrong │ │ │ └── empty.py │ ├── submission.py │ ├── task.py │ └── task.yml ├── matmul_py │ ├── reference.py │ ├── solutions │ │ ├── correct │ │ │ └── ref.py │ │ └── wrong │ │ │ └── low-precision.py │ ├── submission.py │ ├── task.py │ └── task.yml ├── prefixsum_py │ ├── reference.py │ ├── solutions │ │ ├── correct │ │ │ └── ref.py │ │ └── wrong │ │ │ └── empty.py │ ├── submission.py │ ├── task.py │ └── task.yml ├── sort_py │ ├── reference.py │ ├── solutions │ │ ├── correct │ │ │ └── ref.py │ │ └── wrong │ │ │ └── empty.py │ ├── submission.py │ ├── task.py │ └── task.yml ├── template.py ├── utils.py ├── vectoradd_py │ ├── reference.py │ ├── solutions │ │ └── correct │ │ │ ├── submission_cuda_inline.py │ │ │ └── submission_triton.py │ ├── task.py │ └── task.yml └── vectorsum_py │ ├── reference.py │ ├── solutions │ ├── correct │ │ └── pytorch.py │ └── wrong │ │ └── cheat.py │ ├── submission.py │ ├── task.py │ └── task.yml ├── pmpp_v2.yaml └── pmpp_v2 ├── conv2d_py ├── reference.py ├── solutions │ ├── correct │ │ └── ref.py │ └── wrong │ │ └── empty.py ├── submission.py ├── task.py └── task.yml ├── eval.py ├── grayscale_py ├── reference.py ├── solutions │ ├── correct │ │ └── ref.py │ └── wrong │ │ └── empty.py ├── submission.py ├── task.py └── task.yml ├── histogram_py ├── reference.py ├── solutions │ ├── correct │ │ └── ref.py │ └── wrong │ │ └── empty.py ├── submission.py ├── task.py └── task.yml ├── matmul_py ├── reference.py ├── solutions │ ├── correct │ │ └── ref.py │ └── wrong │ │ └── low-precision.py ├── submission.py ├── task.py └── task.yml ├── prefixsum_py ├── reference.py ├── solutions │ ├── correct │ │ └── ref.py │ └── wrong │ │ └── empty.py ├── submission.py ├── task.py └── task.yml ├── sort_py ├── reference.py ├── solutions │ ├── correct │ │ └── ref.py │ └── wrong │ │ └── empty.py ├── submission.py ├── task.py └── task.yml ├── template.py ├── utils.py ├── vectoradd_py ├── reference.py ├── solutions │ └── correct │ │ ├── submission_cuda_inline.py │ │ └── submission_triton.py ├── task.py └── task.yml └── vectorsum_py ├── reference.py ├── solutions ├── correct │ └── pytorch.py └── wrong │ └── cheat.py ├── submission.py ├── task.py └── task.yml /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/README.md -------------------------------------------------------------------------------- /problems/amd.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/amd.yaml -------------------------------------------------------------------------------- /problems/amd/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/amd/eval.py -------------------------------------------------------------------------------- /problems/amd/fp8-mm/reference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/amd/fp8-mm/reference.py -------------------------------------------------------------------------------- /problems/amd/fp8-mm/submission.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/amd/fp8-mm/submission.py -------------------------------------------------------------------------------- /problems/amd/fp8-mm/task.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/amd/fp8-mm/task.py -------------------------------------------------------------------------------- /problems/amd/fp8-mm/task.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/amd/fp8-mm/task.yml -------------------------------------------------------------------------------- /problems/amd/fp8-mm/template-hip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/amd/fp8-mm/template-hip.py -------------------------------------------------------------------------------- /problems/amd/fp8-mm/template.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/amd/fp8-mm/template.py -------------------------------------------------------------------------------- /problems/amd/identity/reference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/amd/identity/reference.py -------------------------------------------------------------------------------- /problems/amd/identity/submission.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/amd/identity/submission.py -------------------------------------------------------------------------------- /problems/amd/identity/task.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/amd/identity/task.py -------------------------------------------------------------------------------- /problems/amd/identity/task.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/amd/identity/task.yml -------------------------------------------------------------------------------- /problems/amd/identity/template.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/amd/identity/template.py -------------------------------------------------------------------------------- /problems/amd/mla-decode/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/amd/mla-decode/README.md -------------------------------------------------------------------------------- /problems/amd/mla-decode/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/amd/mla-decode/eval.py -------------------------------------------------------------------------------- /problems/amd/mla-decode/reference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/amd/mla-decode/reference.py -------------------------------------------------------------------------------- /problems/amd/mla-decode/submission.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/amd/mla-decode/submission.py -------------------------------------------------------------------------------- /problems/amd/mla-decode/task.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/amd/mla-decode/task.py -------------------------------------------------------------------------------- /problems/amd/mla-decode/task.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/amd/mla-decode/task.yml -------------------------------------------------------------------------------- /problems/amd/mla-decode/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/amd/mla-decode/utils.py -------------------------------------------------------------------------------- /problems/amd/moe/reference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/amd/moe/reference.py -------------------------------------------------------------------------------- /problems/amd/moe/submission.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/amd/moe/submission.py -------------------------------------------------------------------------------- /problems/amd/moe/task.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/amd/moe/task.py -------------------------------------------------------------------------------- /problems/amd/moe/task.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/amd/moe/task.yml -------------------------------------------------------------------------------- /problems/amd/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/amd/utils.py -------------------------------------------------------------------------------- /problems/amd_distributed.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/amd_distributed.yaml -------------------------------------------------------------------------------- /problems/amd_distributed/ag-gemm/reference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/amd_distributed/ag-gemm/reference.py -------------------------------------------------------------------------------- /problems/amd_distributed/ag-gemm/submission.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/amd_distributed/ag-gemm/submission.py -------------------------------------------------------------------------------- /problems/amd_distributed/ag-gemm/task.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/amd_distributed/ag-gemm/task.py -------------------------------------------------------------------------------- /problems/amd_distributed/ag-gemm/task.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/amd_distributed/ag-gemm/task.yml -------------------------------------------------------------------------------- /problems/amd_distributed/all2all/reference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/amd_distributed/all2all/reference.py -------------------------------------------------------------------------------- /problems/amd_distributed/all2all/submission.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/amd_distributed/all2all/submission.py -------------------------------------------------------------------------------- /problems/amd_distributed/all2all/task.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/amd_distributed/all2all/task.py -------------------------------------------------------------------------------- /problems/amd_distributed/all2all/task.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/amd_distributed/all2all/task.yml -------------------------------------------------------------------------------- /problems/amd_distributed/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/amd_distributed/eval.py -------------------------------------------------------------------------------- /problems/amd_distributed/gemm-rs/reference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/amd_distributed/gemm-rs/reference.py -------------------------------------------------------------------------------- /problems/amd_distributed/gemm-rs/submission.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/amd_distributed/gemm-rs/submission.py -------------------------------------------------------------------------------- /problems/amd_distributed/gemm-rs/task.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/amd_distributed/gemm-rs/task.py -------------------------------------------------------------------------------- /problems/amd_distributed/gemm-rs/task.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/amd_distributed/gemm-rs/task.yml -------------------------------------------------------------------------------- /problems/amd_distributed/rocshmem_example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/amd_distributed/rocshmem_example.py -------------------------------------------------------------------------------- /problems/amd_distributed/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/amd_distributed/utils.py -------------------------------------------------------------------------------- /problems/beta.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/beta.yaml -------------------------------------------------------------------------------- /problems/bioml.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/bioml.yaml -------------------------------------------------------------------------------- /problems/bioml/trimul/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/bioml/trimul/eval.py -------------------------------------------------------------------------------- /problems/bioml/trimul/reference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/bioml/trimul/reference.py -------------------------------------------------------------------------------- /problems/bioml/trimul/submission.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/bioml/trimul/submission.py -------------------------------------------------------------------------------- /problems/bioml/trimul/task.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/bioml/trimul/task.py -------------------------------------------------------------------------------- /problems/bioml/trimul/task.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/bioml/trimul/task.yml -------------------------------------------------------------------------------- /problems/bioml/trimul/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/bioml/trimul/utils.py -------------------------------------------------------------------------------- /problems/nvidia.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/nvidia.yaml -------------------------------------------------------------------------------- /problems/nvidia/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/nvidia/eval.py -------------------------------------------------------------------------------- /problems/nvidia/eval_better_bench.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/nvidia/eval_better_bench.py -------------------------------------------------------------------------------- /problems/nvidia/nvfp4_gemm/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/nvidia/nvfp4_gemm/eval.py -------------------------------------------------------------------------------- /problems/nvidia/nvfp4_gemm/reference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/nvidia/nvfp4_gemm/reference.py -------------------------------------------------------------------------------- /problems/nvidia/nvfp4_gemm/submission.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/nvidia/nvfp4_gemm/submission.py -------------------------------------------------------------------------------- /problems/nvidia/nvfp4_gemm/task.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/nvidia/nvfp4_gemm/task.py -------------------------------------------------------------------------------- /problems/nvidia/nvfp4_gemm/task.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/nvidia/nvfp4_gemm/task.yml -------------------------------------------------------------------------------- /problems/nvidia/nvfp4_gemm/template.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/nvidia/nvfp4_gemm/template.py -------------------------------------------------------------------------------- /problems/nvidia/nvfp4_gemm/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/nvidia/nvfp4_gemm/utils.py -------------------------------------------------------------------------------- /problems/nvidia/nvfp4_gemv/reference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/nvidia/nvfp4_gemv/reference.py -------------------------------------------------------------------------------- /problems/nvidia/nvfp4_gemv/submission.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/nvidia/nvfp4_gemv/submission.py -------------------------------------------------------------------------------- /problems/nvidia/nvfp4_gemv/task.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/nvidia/nvfp4_gemv/task.py -------------------------------------------------------------------------------- /problems/nvidia/nvfp4_gemv/task.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/nvidia/nvfp4_gemv/task.yml -------------------------------------------------------------------------------- /problems/nvidia/nvfp4_gemv/template.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/nvidia/nvfp4_gemv/template.py -------------------------------------------------------------------------------- /problems/nvidia/nvfp4_gemv/template_cute.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/nvidia/nvfp4_gemv/template_cute.py -------------------------------------------------------------------------------- /problems/nvidia/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/nvidia/utils.py -------------------------------------------------------------------------------- /problems/pmpp/conv2d_py/reference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/pmpp/conv2d_py/reference.py -------------------------------------------------------------------------------- /problems/pmpp/conv2d_py/solutions/correct/ref.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/pmpp/conv2d_py/solutions/correct/ref.py -------------------------------------------------------------------------------- /problems/pmpp/conv2d_py/solutions/wrong/empty.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/pmpp/conv2d_py/solutions/wrong/empty.py -------------------------------------------------------------------------------- /problems/pmpp/conv2d_py/submission.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/pmpp/conv2d_py/submission.py -------------------------------------------------------------------------------- /problems/pmpp/conv2d_py/task.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/pmpp/conv2d_py/task.py -------------------------------------------------------------------------------- /problems/pmpp/conv2d_py/task.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/pmpp/conv2d_py/task.yml -------------------------------------------------------------------------------- /problems/pmpp/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/pmpp/eval.py -------------------------------------------------------------------------------- /problems/pmpp/grayscale_py/reference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/pmpp/grayscale_py/reference.py -------------------------------------------------------------------------------- /problems/pmpp/grayscale_py/solutions/correct/ref.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/pmpp/grayscale_py/solutions/correct/ref.py -------------------------------------------------------------------------------- /problems/pmpp/grayscale_py/solutions/wrong/empty.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/pmpp/grayscale_py/solutions/wrong/empty.py -------------------------------------------------------------------------------- /problems/pmpp/grayscale_py/submission.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/pmpp/grayscale_py/submission.py -------------------------------------------------------------------------------- /problems/pmpp/grayscale_py/task.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/pmpp/grayscale_py/task.py -------------------------------------------------------------------------------- /problems/pmpp/grayscale_py/task.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/pmpp/grayscale_py/task.yml -------------------------------------------------------------------------------- /problems/pmpp/histogram_py/reference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/pmpp/histogram_py/reference.py -------------------------------------------------------------------------------- /problems/pmpp/histogram_py/solutions/correct/ref.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/pmpp/histogram_py/solutions/correct/ref.py -------------------------------------------------------------------------------- /problems/pmpp/histogram_py/solutions/wrong/empty.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/pmpp/histogram_py/solutions/wrong/empty.py -------------------------------------------------------------------------------- /problems/pmpp/histogram_py/submission.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/pmpp/histogram_py/submission.py -------------------------------------------------------------------------------- /problems/pmpp/histogram_py/task.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/pmpp/histogram_py/task.py -------------------------------------------------------------------------------- /problems/pmpp/histogram_py/task.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/pmpp/histogram_py/task.yml -------------------------------------------------------------------------------- /problems/pmpp/matmul_py/reference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/pmpp/matmul_py/reference.py -------------------------------------------------------------------------------- /problems/pmpp/matmul_py/solutions/correct/ref.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/pmpp/matmul_py/solutions/correct/ref.py -------------------------------------------------------------------------------- /problems/pmpp/matmul_py/solutions/wrong/low-precision.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/pmpp/matmul_py/solutions/wrong/low-precision.py -------------------------------------------------------------------------------- /problems/pmpp/matmul_py/submission.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/pmpp/matmul_py/submission.py -------------------------------------------------------------------------------- /problems/pmpp/matmul_py/task.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/pmpp/matmul_py/task.py -------------------------------------------------------------------------------- /problems/pmpp/matmul_py/task.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/pmpp/matmul_py/task.yml -------------------------------------------------------------------------------- /problems/pmpp/prefixsum_py/reference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/pmpp/prefixsum_py/reference.py -------------------------------------------------------------------------------- /problems/pmpp/prefixsum_py/solutions/correct/ref.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/pmpp/prefixsum_py/solutions/correct/ref.py -------------------------------------------------------------------------------- /problems/pmpp/prefixsum_py/solutions/wrong/empty.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/pmpp/prefixsum_py/solutions/wrong/empty.py -------------------------------------------------------------------------------- /problems/pmpp/prefixsum_py/submission.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/pmpp/prefixsum_py/submission.py -------------------------------------------------------------------------------- /problems/pmpp/prefixsum_py/task.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/pmpp/prefixsum_py/task.py -------------------------------------------------------------------------------- /problems/pmpp/prefixsum_py/task.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/pmpp/prefixsum_py/task.yml -------------------------------------------------------------------------------- /problems/pmpp/sort_py/reference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/pmpp/sort_py/reference.py -------------------------------------------------------------------------------- /problems/pmpp/sort_py/solutions/correct/ref.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/pmpp/sort_py/solutions/correct/ref.py -------------------------------------------------------------------------------- /problems/pmpp/sort_py/solutions/wrong/empty.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/pmpp/sort_py/solutions/wrong/empty.py -------------------------------------------------------------------------------- /problems/pmpp/sort_py/submission.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/pmpp/sort_py/submission.py -------------------------------------------------------------------------------- /problems/pmpp/sort_py/task.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/pmpp/sort_py/task.py -------------------------------------------------------------------------------- /problems/pmpp/sort_py/task.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/pmpp/sort_py/task.yml -------------------------------------------------------------------------------- /problems/pmpp/template.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/pmpp/template.py -------------------------------------------------------------------------------- /problems/pmpp/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/pmpp/utils.py -------------------------------------------------------------------------------- /problems/pmpp/vectoradd_py/reference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/pmpp/vectoradd_py/reference.py -------------------------------------------------------------------------------- /problems/pmpp/vectoradd_py/solutions/correct/submission_cuda_inline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/pmpp/vectoradd_py/solutions/correct/submission_cuda_inline.py -------------------------------------------------------------------------------- /problems/pmpp/vectoradd_py/solutions/correct/submission_triton.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/pmpp/vectoradd_py/solutions/correct/submission_triton.py -------------------------------------------------------------------------------- /problems/pmpp/vectoradd_py/task.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/pmpp/vectoradd_py/task.py -------------------------------------------------------------------------------- /problems/pmpp/vectoradd_py/task.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/pmpp/vectoradd_py/task.yml -------------------------------------------------------------------------------- /problems/pmpp/vectorsum_py/reference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/pmpp/vectorsum_py/reference.py -------------------------------------------------------------------------------- /problems/pmpp/vectorsum_py/solutions/correct/pytorch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/pmpp/vectorsum_py/solutions/correct/pytorch.py -------------------------------------------------------------------------------- /problems/pmpp/vectorsum_py/solutions/wrong/cheat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/pmpp/vectorsum_py/solutions/wrong/cheat.py -------------------------------------------------------------------------------- /problems/pmpp/vectorsum_py/submission.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/pmpp/vectorsum_py/submission.py -------------------------------------------------------------------------------- /problems/pmpp/vectorsum_py/task.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/pmpp/vectorsum_py/task.py -------------------------------------------------------------------------------- /problems/pmpp/vectorsum_py/task.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/pmpp/vectorsum_py/task.yml -------------------------------------------------------------------------------- /problems/pmpp_v2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/pmpp_v2.yaml -------------------------------------------------------------------------------- /problems/pmpp_v2/conv2d_py/reference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/pmpp_v2/conv2d_py/reference.py -------------------------------------------------------------------------------- /problems/pmpp_v2/conv2d_py/solutions/correct/ref.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/pmpp_v2/conv2d_py/solutions/correct/ref.py -------------------------------------------------------------------------------- /problems/pmpp_v2/conv2d_py/solutions/wrong/empty.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/pmpp_v2/conv2d_py/solutions/wrong/empty.py -------------------------------------------------------------------------------- /problems/pmpp_v2/conv2d_py/submission.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/pmpp_v2/conv2d_py/submission.py -------------------------------------------------------------------------------- /problems/pmpp_v2/conv2d_py/task.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/pmpp_v2/conv2d_py/task.py -------------------------------------------------------------------------------- /problems/pmpp_v2/conv2d_py/task.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/pmpp_v2/conv2d_py/task.yml -------------------------------------------------------------------------------- /problems/pmpp_v2/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/pmpp_v2/eval.py -------------------------------------------------------------------------------- /problems/pmpp_v2/grayscale_py/reference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/pmpp_v2/grayscale_py/reference.py -------------------------------------------------------------------------------- /problems/pmpp_v2/grayscale_py/solutions/correct/ref.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/pmpp_v2/grayscale_py/solutions/correct/ref.py -------------------------------------------------------------------------------- /problems/pmpp_v2/grayscale_py/solutions/wrong/empty.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/pmpp_v2/grayscale_py/solutions/wrong/empty.py -------------------------------------------------------------------------------- /problems/pmpp_v2/grayscale_py/submission.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/pmpp_v2/grayscale_py/submission.py -------------------------------------------------------------------------------- /problems/pmpp_v2/grayscale_py/task.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/pmpp_v2/grayscale_py/task.py -------------------------------------------------------------------------------- /problems/pmpp_v2/grayscale_py/task.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/pmpp_v2/grayscale_py/task.yml -------------------------------------------------------------------------------- /problems/pmpp_v2/histogram_py/reference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/pmpp_v2/histogram_py/reference.py -------------------------------------------------------------------------------- /problems/pmpp_v2/histogram_py/solutions/correct/ref.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/pmpp_v2/histogram_py/solutions/correct/ref.py -------------------------------------------------------------------------------- /problems/pmpp_v2/histogram_py/solutions/wrong/empty.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/pmpp_v2/histogram_py/solutions/wrong/empty.py -------------------------------------------------------------------------------- /problems/pmpp_v2/histogram_py/submission.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/pmpp_v2/histogram_py/submission.py -------------------------------------------------------------------------------- /problems/pmpp_v2/histogram_py/task.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/pmpp_v2/histogram_py/task.py -------------------------------------------------------------------------------- /problems/pmpp_v2/histogram_py/task.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/pmpp_v2/histogram_py/task.yml -------------------------------------------------------------------------------- /problems/pmpp_v2/matmul_py/reference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/pmpp_v2/matmul_py/reference.py -------------------------------------------------------------------------------- /problems/pmpp_v2/matmul_py/solutions/correct/ref.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/pmpp_v2/matmul_py/solutions/correct/ref.py -------------------------------------------------------------------------------- /problems/pmpp_v2/matmul_py/solutions/wrong/low-precision.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/pmpp_v2/matmul_py/solutions/wrong/low-precision.py -------------------------------------------------------------------------------- /problems/pmpp_v2/matmul_py/submission.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/pmpp_v2/matmul_py/submission.py -------------------------------------------------------------------------------- /problems/pmpp_v2/matmul_py/task.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/pmpp_v2/matmul_py/task.py -------------------------------------------------------------------------------- /problems/pmpp_v2/matmul_py/task.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/pmpp_v2/matmul_py/task.yml -------------------------------------------------------------------------------- /problems/pmpp_v2/prefixsum_py/reference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/pmpp_v2/prefixsum_py/reference.py -------------------------------------------------------------------------------- /problems/pmpp_v2/prefixsum_py/solutions/correct/ref.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/pmpp_v2/prefixsum_py/solutions/correct/ref.py -------------------------------------------------------------------------------- /problems/pmpp_v2/prefixsum_py/solutions/wrong/empty.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/pmpp_v2/prefixsum_py/solutions/wrong/empty.py -------------------------------------------------------------------------------- /problems/pmpp_v2/prefixsum_py/submission.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/pmpp_v2/prefixsum_py/submission.py -------------------------------------------------------------------------------- /problems/pmpp_v2/prefixsum_py/task.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/pmpp_v2/prefixsum_py/task.py -------------------------------------------------------------------------------- /problems/pmpp_v2/prefixsum_py/task.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/pmpp_v2/prefixsum_py/task.yml -------------------------------------------------------------------------------- /problems/pmpp_v2/sort_py/reference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/pmpp_v2/sort_py/reference.py -------------------------------------------------------------------------------- /problems/pmpp_v2/sort_py/solutions/correct/ref.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/pmpp_v2/sort_py/solutions/correct/ref.py -------------------------------------------------------------------------------- /problems/pmpp_v2/sort_py/solutions/wrong/empty.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/pmpp_v2/sort_py/solutions/wrong/empty.py -------------------------------------------------------------------------------- /problems/pmpp_v2/sort_py/submission.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/pmpp_v2/sort_py/submission.py -------------------------------------------------------------------------------- /problems/pmpp_v2/sort_py/task.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/pmpp_v2/sort_py/task.py -------------------------------------------------------------------------------- /problems/pmpp_v2/sort_py/task.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/pmpp_v2/sort_py/task.yml -------------------------------------------------------------------------------- /problems/pmpp_v2/template.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/pmpp_v2/template.py -------------------------------------------------------------------------------- /problems/pmpp_v2/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/pmpp_v2/utils.py -------------------------------------------------------------------------------- /problems/pmpp_v2/vectoradd_py/reference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/pmpp_v2/vectoradd_py/reference.py -------------------------------------------------------------------------------- /problems/pmpp_v2/vectoradd_py/solutions/correct/submission_cuda_inline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/pmpp_v2/vectoradd_py/solutions/correct/submission_cuda_inline.py -------------------------------------------------------------------------------- /problems/pmpp_v2/vectoradd_py/solutions/correct/submission_triton.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/pmpp_v2/vectoradd_py/solutions/correct/submission_triton.py -------------------------------------------------------------------------------- /problems/pmpp_v2/vectoradd_py/task.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/pmpp_v2/vectoradd_py/task.py -------------------------------------------------------------------------------- /problems/pmpp_v2/vectoradd_py/task.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/pmpp_v2/vectoradd_py/task.yml -------------------------------------------------------------------------------- /problems/pmpp_v2/vectorsum_py/reference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/pmpp_v2/vectorsum_py/reference.py -------------------------------------------------------------------------------- /problems/pmpp_v2/vectorsum_py/solutions/correct/pytorch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/pmpp_v2/vectorsum_py/solutions/correct/pytorch.py -------------------------------------------------------------------------------- /problems/pmpp_v2/vectorsum_py/solutions/wrong/cheat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/pmpp_v2/vectorsum_py/solutions/wrong/cheat.py -------------------------------------------------------------------------------- /problems/pmpp_v2/vectorsum_py/submission.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/pmpp_v2/vectorsum_py/submission.py -------------------------------------------------------------------------------- /problems/pmpp_v2/vectorsum_py/task.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/pmpp_v2/vectorsum_py/task.py -------------------------------------------------------------------------------- /problems/pmpp_v2/vectorsum_py/task.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gpu-mode/reference-kernels/HEAD/problems/pmpp_v2/vectorsum_py/task.yml --------------------------------------------------------------------------------