Additional smoke tests

Remove release blocking changes
pytorch · Mar 7, 2023 · 20b4b4f · 20b4b4f
1 parent a86e0bc
commit 20b4b4f
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 8 deletions.
diff --git a/.github/workflows/validate-nightly-binaries.yml b/.github/workflows/validate-nightly-binaries.yml
@@ -30,6 +30,6 @@ jobs:
   nightly:
     uses: ./.github/workflows/validate-binaries.yml
     with:
-      channel: test
+      channel: nightly
       os: all
-      limit-win-builds: disable
+      limit-win-builds: enable
diff --git a/test/smoke_test/smoke_test.py b/test/smoke_test/smoke_test.py
@@ -112,11 +112,7 @@ def smoke_test_cuda(package: str) -> None:
         if (sys.platform == "linux" or sys.platform == "linux2") and sys.version_info < (3, 11, 0):
             smoke_test_compile()
 
-        # This check has to be run last, since its messing up CUDA runtime.
-        # Restrict only to conda builds since Wheel seems to crash with
-        # segmentation fault and don't recover
-        if(package_type == 'conda'):
-            test_cuda_runtime_errors_captured()
+        test_cuda_runtime_errors_captured()
 
 
 def smoke_test_conv2d() -> None:
@@ -169,7 +165,7 @@ def smoke_test_linalg() -> None:
             torch.linalg.svd(A)
 
 def smoke_test_compile() -> None:
-    supported_dtypes = [torch.float32]
+    supported_dtypes = [torch.float16, torch.float32, torch.float64]
     def foo(x: torch.Tensor) -> torch.Tensor:
         return torch.sin(x) + torch.cos(x)
     for dtype in supported_dtypes: