Add test and benchmark for explicit dot GEMM

ROCm · Oct 11, 2024 · 0c16e7a · 0c16e7a
1 parent b36e072
commit 0c16e7a
Show file tree

Hide file tree

Showing 2 changed files with 439 additions and 17 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