Re-enable torchdynamo unit tests (#86658)

Pull Request resolved: #86658 Approved by: https://github.com/jansel
pytorch · Oct 12, 2022 · ce48df9 · ce48df9
1 parent 692b525
commit ce48df9
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 4 deletions.
diff --git a/.github/ci_commit_pins/torchdynamo.txt b/.github/ci_commit_pins/torchdynamo.txt
@@ -1 +1 @@
-3f92a095ae72ab276e1efe0466771996292544c5
+d4b62d69a2d5ae1823c1d6695237b44faee05acd
diff --git a/.jenkins/pytorch/common_utils.sh b/.jenkins/pytorch/common_utils.sh
@@ -123,6 +123,10 @@ function clone_pytorch_xla() {
   fi
 }
 
+function install_filelock() {
+  pip_install filelock
+}
+
 function install_torchdynamo() {
   local commit
   commit=$(get_pinned_commit torchdynamo)

diff --git a/.jenkins/pytorch/test.sh b/.jenkins/pytorch/test.sh
@@ -661,7 +661,7 @@ test_vec256() {
 
 test_dynamo() {
   pushd ../torchdynamo
-  pytest test
+  pytest test/dynamo
   popd
 }
 
@@ -703,8 +703,8 @@ elif [[ "${TEST_CONFIG}" == *dynamo* && "${SHARD_NUMBER}" == 2 && $NUM_TEST_SHAR
   install_torchvision
   checkout_install_torchdynamo
   test_dynamo_shard 2
-  # Skip running test in the dynamo repo to unblock the dynamo pin update
-  # test_dynamo
+  install_filelock
+  test_dynamo
 elif [[ "${SHARD_NUMBER}" == 1 && $NUM_TEST_SHARDS -gt 1 ]]; then
   test_without_numpy
   install_torchvision