From 92ffaefa455f09399cc018f9eb87b6e8bac55ec1 Mon Sep 17 00:00:00 2001
From: Xu Zhao <xzhao9@meta.com>
Date: Tue, 4 Jun 2024 10:00:04 -0400
Subject: [PATCH] Push the result

---
 .github/workflows/torchao.yml | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/.github/workflows/torchao.yml b/.github/workflows/torchao.yml
index c2bf850ea5..d5d6016aa1 100644
--- 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