From eb0b74e55985b9ac932400a9fc7f41b33b96dc55 Mon Sep 17 00:00:00 2001 From: Ross Wheeler Date: Thu, 27 Jun 2024 00:30:38 -0700 Subject: [PATCH] Add H100 and A100 ptx/sass files --- .github/workflows/ci_tests.yml | 42 +++++++++++++++++++++++++++++++++- 1 file changed, 41 insertions(+), 1 deletion(-) diff --git a/.github/workflows/ci_tests.yml b/.github/workflows/ci_tests.yml index 6b17917..58412e0 100644 --- a/.github/workflows/ci_tests.yml +++ b/.github/workflows/ci_tests.yml @@ -52,9 +52,49 @@ jobs: cp *.sass ptx_sass_logs/ ls ptx_sass_logs/ + - name: Generate ptx/sass files for A100 and upload them to persistent storage + run: | + mkdir -p dev/cuda/ptx_sass_logs_A100 + make train_gpt2cu GPU_COMPUTE_CAPABILITY=80 + cuobjdump --dump-ptx train_gpt2cu > dev/cuda/train_gpt2cu.ptx + cuobjdump --dump-sass train_gpt2cu > dev/cuda/train_gpt2cu.sass + cd dev/cuda + make -j GPU_COMPUTE_CAPABILITY=80 all_ptx + make -j GPU_COMPUTE_CAPABILITY=80 all_sass + cp *.ptx ptx_sass_logs_A100/ + cp *.sass ptx_sass_logs_A100/ + ls ptx_sass_logs_A100/ + + - name: Generate ptx/sass files for H100 and upload them to persistent storage + run: | + mkdir -p dev/cuda/ptx_sass_logs_H100 + make train_gpt2cu GPU_COMPUTE_CAPABILITY=90 + cuobjdump --dump-ptx train_gpt2cu > dev/cuda/train_gpt2cu.ptx + cuobjdump --dump-sass train_gpt2cu > dev/cuda/train_gpt2cu.sass + cd dev/cuda + make -j GPU_COMPUTE_CAPABILITY=90 all_ptx + make -j GPU_COMPUTE_CAPABILITY=90 all_sass + cp *.ptx ptx_sass_logs_H100/ + cp *.sass ptx_sass_logs_H100/ + ls ptx_sass_logs_H100/ + - name: Upload ptx/sass files uses: actions/upload-artifact@v4 with: name: ptx_sass_files path: dev/cuda/ptx_sass_logs/ - retention-days: 30 # days to retain \ No newline at end of file + retention-days: 30 # days to retain + + - name: Upload ptx/sass files for A100 + uses: actions/upload-artifact@v4 + with: + name: ptx_sass_files_A100 + path: dev/cuda/ptx_sass_logs_A100/ + retention-days: 30 # days to retain + + - name: Upload ptx/sass files for H100 + uses: actions/upload-artifact@v4 + with: + name: ptx_sass_files_H100 + path: dev/cuda/ptx_sass_logs_H100/ + retention-days: 30 # days to retain \ No newline at end of file