test ci-temp-test-model

Author: botbigeyes

.github/workflows/ci-temp-model-test.yaml

@@ -6,18 +6,21 @@ on:
   pull_request:
     branches: [ "main" ]
 
-# env:
-#   CUDA_VISIBLE_DEVICES: 7
-
 jobs:
   ci-temp-test-model:
     runs-on: ci-temp-test-model
     concurrency:
       group: ci-temp-test-model-${{ github.event.pull_request.number || github.ref }}
       cancel-in-progress: true
     steps:
+      - name: Checkout accelerator-integration-wg
+        uses: actions/checkout@v4
+     
       - name: Checkout flaggems
         uses: actions/checkout@v4
+        with:
+          repository: FlagOpen/FlagGems
+          path: flag_gems
 
       - name: Checkout benchmark
         uses: actions/checkout@v4
@@ -28,15 +31,16 @@ jobs:
       - name: Install and run benchmark
         shell: bash
         run: |
-          source tools/run_command.sh
-          run_command source "/root/anaconda3/etc/profile.d/conda.sh"
-          run_command conda init bash
-          run_command conda activate wz-torch-bench
-          run_command pip install -e .
-          run_command sed -i '/self\.worker\.run("import torch")/a\        self.worker.run(\
+          # source tools/run_command.sh
+          set -e
+          source "/root/anaconda3/etc/profile.d/conda.sh"
+          conda init bash
+          conda activate new-torchbenchmark
+          pip install -e flag_gems/
+          sed -i '/self\.worker\.run("import torch")/a\        self.worker.run(\
             """\
             import flag_gems\
             flag_gems.enable(record=False, once=True, path='"'"'benchmark/oplist.log'"'"')""")' benchmark/torchbenchmark/__init__.py
-          run_command python benchmark/install.py models hf_GPT2
-          run_command python benchmark/run_benchmark.py test_bench --accuracy --device cuda --test eval --output output.json --models hf_GPT2
-          run_command sed -i '/self\.worker\.run($/,/^[[:space:]]*flag_gems\.enable.*oplist\.log.*""")/d' benchmark/torchbenchmark/__init__.py
+          python benchmark/install.py models hf_GPT2
+          python benchmark/run_benchmark.py test_bench --accuracy --device cuda --test eval --output output.json --models hf_GPT2
+          sed -i '/self\.worker\.run($/,/^[[:space:]]*flag_gems\.enable.*oplist\.log.*""")/d' benchmark/torchbenchmark/__init__.py