diff --git a/.github/scripts/validate_test_ops.sh b/.github/scripts/validate_test_ops.sh index bd724394b..91ef3ffde 100644 --- a/.github/scripts/validate_test_ops.sh +++ b/.github/scripts/validate_test_ops.sh @@ -18,7 +18,7 @@ retry git clone ${BRANCH} --depth 1 https://github.com/pytorch/pytorch.git retry git submodule update --init --recursive pushd pytorch -pip install expecttest pyyaml jinja2 packaging +pip install expecttest numpy pyyaml jinja2 packaging xmlrunner hypothesis unittest-xml-reporting # Run test_ops validation export CUDA_LAUNCH_BLOCKING=1