Skip to content

Commit b321562

Browse files
authored
Revert "Add some more validation checks for torch.linalg.eigh and torch.compile (#1580)" (#1594)
This reverts commit 4c7fa06.
1 parent 4c7fa06 commit b321562

File tree

3 files changed

+0
-21
lines changed

3 files changed

+0
-21
lines changed

check_binary.sh

Lines changed: 0 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -404,12 +404,6 @@ if [[ "$DESIRED_CUDA" != 'cpu' && "$DESIRED_CUDA" != 'cpu-cxx11-abi' && "$DESIRE
404404
echo "Test that linalg works"
405405
python -c "import torch;x=torch.rand(3,3,device='cuda');print(torch.linalg.svd(torch.mm(x.t(), x)))"
406406

407-
echo "Test that linalg.eigh works"
408-
python -c "import torch;x=torch.rand(3,3,device='cuda');print(torch.linalg.eigh(torch.mm(x.t(), x)))"
409-
410-
echo "Checking that basic torch.compile works"
411-
python ${TEST_CODE_DIR}/torch_compile_smoke.py
412-
413407
popd
414408
fi # if libtorch
415409
fi # if cuda

test/smoke_test/smoke_test.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -193,9 +193,6 @@ def smoke_test_linalg() -> None:
193193
A = torch.randn(20, 16, 50, 100, device="cuda").type(dtype)
194194
torch.linalg.svd(A)
195195

196-
A = torch.rand(3, 3, device="cuda")
197-
L, Q = torch.linalg.eigh(torch.mm(A.t(), A))
198-
199196

200197
def smoke_test_compile() -> None:
201198
supported_dtypes = [torch.float16, torch.float32, torch.float64]

test_example_code/torch_compile_smoke.py

Lines changed: 0 additions & 12 deletions
This file was deleted.

0 commit comments

Comments
 (0)