Skip to content

Commit 20b4b4f

Browse files
committed
Additional smoke tests
Remove release blocking changes
1 parent a86e0bc commit 20b4b4f

File tree

2 files changed

+4
-8
lines changed

2 files changed

+4
-8
lines changed

.github/workflows/validate-nightly-binaries.yml

+2-2
Original file line numberDiff line numberDiff line change
@@ -30,6 +30,6 @@ jobs:
3030
nightly:
3131
uses: ./.github/workflows/validate-binaries.yml
3232
with:
33-
channel: test
33+
channel: nightly
3434
os: all
35-
limit-win-builds: disable
35+
limit-win-builds: enable

test/smoke_test/smoke_test.py

+2-6
Original file line numberDiff line numberDiff line change
@@ -112,11 +112,7 @@ def smoke_test_cuda(package: str) -> None:
112112
if (sys.platform == "linux" or sys.platform == "linux2") and sys.version_info < (3, 11, 0):
113113
smoke_test_compile()
114114

115-
# This check has to be run last, since its messing up CUDA runtime.
116-
# Restrict only to conda builds since Wheel seems to crash with
117-
# segmentation fault and don't recover
118-
if(package_type == 'conda'):
119-
test_cuda_runtime_errors_captured()
115+
test_cuda_runtime_errors_captured()
120116

121117

122118
def smoke_test_conv2d() -> None:
@@ -169,7 +165,7 @@ def smoke_test_linalg() -> None:
169165
torch.linalg.svd(A)
170166

171167
def smoke_test_compile() -> None:
172-
supported_dtypes = [torch.float32]
168+
supported_dtypes = [torch.float16, torch.float32, torch.float64]
173169
def foo(x: torch.Tensor) -> torch.Tensor:
174170
return torch.sin(x) + torch.cos(x)
175171
for dtype in supported_dtypes:

0 commit comments

Comments
 (0)