Add test ops validation for validation workflows (pytorch#1652)

atalman · Dec 28, 2023 · 9c8a8dc · 9c8a8dc
1 parent b16ac1f
commit 9c8a8dc
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/.github/scripts/validate_test_ops.sh b/.github/scripts/validate_test_ops.sh
@@ -11,6 +11,8 @@ retry git clone --depth 1 https://github.com/pytorch/pytorch.git
 retry git submodule update --init --recursive
 pushd pytorch
 
+pip install expecttest pyyaml jinja2
+
 # Run test_ops validation
 export CUDA_LAUNCH_BLOCKING=1
 python3 test/test_ops.py