Add test and benchmark for explicit dot GEMM

ROCm · Oct 1, 2024 · 14a7e75 · 14a7e75
1 parent e13fc4c
commit 14a7e75
Show file tree

Hide file tree

Showing 2 changed files with 442 additions and 21 deletions.
diff --git a/.github/workflows/amd_perf_kernel_Integration_tests.yml b/.github/workflows/amd_perf_kernel_Integration_tests.yml
@@ -129,9 +129,11 @@ jobs:
           pytest -vvvv ./python/perf-kernels/softmax.py
           pytest -vvv ./python/perf-kernels/rmsnorm.py
           pytest -vvv ./python/perf-kernels/layernorm.py
+          pytest -vvv ./python/perf-kernels/multreduce_matmul_kernel.py
       - name: Run Perf Kernels Benchmark
         run: |
           python ./python/perf-kernels/flash-attention.py
           python ./python/perf-kernels/softmax.py
           python ./python/perf-kernels/rmsnorm.py
           python ./python/perf-kernels/layernorm.py
+          python ./python/perf-kernels/multreduce_matmul_kernel.py bench