Revert "Add some more validation checks for torch.linalg.eigh and tor…

…ch.compile (#1580)" (#1594) This reverts commit 4c7fa06.
pytorch · Nov 16, 2023 · b321562 · b321562
1 parent 4c7fa06
commit b321562
Show file tree

Hide file tree

Showing 3 changed files with 0 additions and 21 deletions.
diff --git a/check_binary.sh b/check_binary.sh
@@ -404,12 +404,6 @@ if [[ "$DESIRED_CUDA" != 'cpu' && "$DESIRED_CUDA" != 'cpu-cxx11-abi' && "$DESIRE
     echo "Test that linalg works"
     python -c "import torch;x=torch.rand(3,3,device='cuda');print(torch.linalg.svd(torch.mm(x.t(), x)))"
 
-    echo "Test that linalg.eigh works"
-    python -c "import torch;x=torch.rand(3,3,device='cuda');print(torch.linalg.eigh(torch.mm(x.t(), x)))"
-
-    echo "Checking that basic torch.compile works"
-    python ${TEST_CODE_DIR}/torch_compile_smoke.py
-
     popd
   fi # if libtorch
 fi # if cuda

diff --git a/test/smoke_test/smoke_test.py b/test/smoke_test/smoke_test.py
@@ -193,9 +193,6 @@ def smoke_test_linalg() -> None:
             A = torch.randn(20, 16, 50, 100, device="cuda").type(dtype)
             torch.linalg.svd(A)
 
-        A = torch.rand(3, 3, device="cuda")
-        L, Q = torch.linalg.eigh(torch.mm(A.t(), A))
-
 
 def smoke_test_compile() -> None:
     supported_dtypes = [torch.float16, torch.float32, torch.float64]

diff --git a/test_example_code/torch_compile_smoke.py b/test_example_code/torch_compile_smoke.py