test inf2 driver install on benchmark

pytorch · Jan 10, 2024 · e52a945 · e52a945
1 parent ff1b858
commit e52a945
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 6 deletions.
diff --git a/.github/workflows/benchmark_nightly.yml b/.github/workflows/benchmark_nightly.yml
@@ -1,16 +1,14 @@
 name: Benchmark torchserve nightly
 
 on:
-  # run every day at 2:15am
-  schedule:
-    - cron:  '15 02 * * *'
+  push
 
 jobs:
   nightly:
     strategy:
       fail-fast: false
       matrix:
-        hardware: [cpu, gpu, inf2]
+        hardware: [inf2]
     runs-on:
       - self-hosted
       - ${{ matrix.hardware }}

diff --git a/benchmarks/benchmark_config_neuronx.yaml b/benchmarks/benchmark_config_neuronx.yaml
@@ -3,8 +3,8 @@
 #  - nightly: "2022.3.16"
 #  - release: "0.5.3"
 # Nightly build will be installed if "ts_version" is not specifiged
-#ts_version:
-#    branch: &ts_version "master"
+ts_version:
+    branch: &ts_version "inf2-driver-install"
 
 # a list of model configure yaml files defined in benchmarks/models_config
 # or a list of model configure yaml files with full path