Add 2.1.2 release config

pytorch · Nov 27, 2023 · 4a987f3 · 4a987f3
1 parent 50e4012
commit 4a987f3
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/.github/workflows/userbenchmark-t4-metal.yml b/.github/workflows/userbenchmark-t4-metal.yml
@@ -86,6 +86,7 @@ jobs:
       - name: Upload result jsons to Scribe
         run: |
           . "${SETUP_SCRIPT}" && conda activate "${CONDA_ENV_NAME}"
+          conda install -y six
           pushd benchmark
           RESULTS=($(find ${PWD}/../benchmark-output -name "metrics-*.json" -maxdepth 2 | sort -r))
           echo "Uploading result jsons: ${RESULTS}"

diff --git a/userbenchmark/release-test/configs/2.1.12.yaml b/userbenchmark/release-test/configs/2.1.12.yaml
@@ -0,0 +1,8 @@
+cuda:
+  - version: 12.1
+    magma_version: magma-cuda121
+pytorch:
+  - version: 2.1.2
+    conda_channel: pytorch-test
+  - version: 2.1.1
+    conda_channel: pytorch