fix test

pytorch · Oct 18, 2024 · 85254b8 · 85254b8
1 parent 1278bb7
commit 85254b8
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/.github/scripts/run_tests.sh b/.github/scripts/run_tests.sh
@@ -44,8 +44,9 @@ function run_torch_xla_cpp_tests() {
                "test_aten_xla_tensor_2"
                "test_aten_xla_tensor_3"
                "test_aten_xla_tensor_4"
-               "pjrt_computation_client_test"
-               "ifrt_computation_client_test")
+               "pjrt_computation_client_test")
+               # Disable IFRT test as it currently crashes
+               #"ifrt_computation_client_test")
   test_names2=("test_aten_xla_tensor_5"
                "test_aten_xla_tensor_6"
                "test_ir"

diff --git a/BUILD b/BUILD
@@ -78,7 +78,6 @@ test_suite(
         "//test/cpp:test_tensor",
         "//test/cpp:test_xla_sharding",
         "//torch_xla/csrc/runtime:pjrt_computation_client_test",
-        # Disable IFRT test as it currently crashes
-        # "//torch_xla/csrc/runtime:ifrt_computation_client_test",
+        "//torch_xla/csrc/runtime:ifrt_computation_client_test",
     ],
 )