Push the result

pytorch · Jun 4, 2024 · 92ffaef · 92ffaef
1 parent 1535b70
commit 92ffaef
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/.github/workflows/torchao.yml b/.github/workflows/torchao.yml
@@ -29,6 +29,7 @@ jobs:
           sudo nvidia-smi -pm 1
           sudo nvidia-smi -ac 1215,1410
           nvidia-smi
+          sudo ldconfig
       - name: Clone and setup conda env
         run: |
           CONDA_ENV=${BASE_CONDA_ENV} . "${SETUP_SCRIPT}"
@@ -43,7 +44,7 @@ jobs:
           if [ -d .userbenchmark ]; then rm -Rf .userbenchmark; fi
           # Install torchao
           echo "Installing torchao"
-          python install.py torchao
+          python install.py --userbenchmark torchao
           echo "Running the torchao userbenchmark"
           python run_benchmark.py torchao --ci
           cp -r ./.userbenchmark/torchao ../benchmark-output