Skip to content

Commit

Permalink
Add test and benchmark for explicit dot GEMM
Browse files Browse the repository at this point in the history
  • Loading branch information
brunomazzottiamd committed Oct 11, 2024
1 parent b36e072 commit 0c16e7a
Show file tree
Hide file tree
Showing 2 changed files with 439 additions and 17 deletions.
2 changes: 2 additions & 0 deletions .github/workflows/amd_perf_kernel_Integration_tests.yml
Original file line number Diff line number Diff line change
Expand Up @@ -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
Loading

0 comments on commit 0c16e7a

Please sign in to comment.