Install IPEX when running benchmarks (#2105)

Signed-off-by: Whitney Tsang <whitney.tsang@intel.com>
intel · Sep 4, 2024 · d7e5226 · d7e5226
1 parent 1d5cc5e
commit d7e5226
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/scripts/test-triton.sh b/scripts/test-triton.sh
@@ -317,6 +317,9 @@ test_triton() {
   if [ "$TEST_MICRO_BENCHMARKS" = true ]; then
     run_microbench_tests
   fi
+  if [ "$TEST_BENCHMARK_SOFTMAX" = true ] || [ "$TEST_BENCHMARK_GEMM" = true ] || [ "$TEST_BENCHMARK_ATTENTION" = true ]; then
+    $SKIP_DEPS || $SCRIPTS_DIR/compile-pytorch-ipex.sh --pinned --source --pytorch --ipex
+  fi
   if [ "$TEST_BENCHMARK_SOFTMAX" = true ]; then
     run_benchmark_softmax
   fi