From e5d4e6a8aa23f42e34d2c8aac12108cdc905500d Mon Sep 17 00:00:00 2001 From: chenyu Date: Thu, 16 May 2024 23:28:28 -0400 Subject: [PATCH] BEAM=2 in green CI for 100 TFLOPS (#4624) --- .github/workflows/benchmark.yml | 2 ++ 1 file changed, 2 insertions(+) diff --git a/.github/workflows/benchmark.yml b/.github/workflows/benchmark.yml index 088879671c..74478cfc91 100644 --- a/.github/workflows/benchmark.yml +++ b/.github/workflows/benchmark.yml @@ -120,6 +120,8 @@ jobs: run: | CUDA=1 HALF=1 DEBUG=2 python3 extra/gemm/simple_matmul.py | tee matmul.txt CUDA=1 BFLOAT16=1 DEBUG=2 python3 extra/gemm/simple_matmul.py | tee matmul_bfloat16.txt + - name: Run Tensor Core GEMM (CUDA) with BEAM + run: BEAM=2 CUDA=1 HALF=1 DEBUG=2 python3 extra/gemm/simple_matmul.py - name: Run Tensor Core GEMM (PTX) run: CUDA=1 PTX=1 HALF=1 DEBUG=2 python3 extra/gemm/simple_matmul.py | tee matmul_ptx.txt - name: Run Tensor Core GEMM (NV)