Add basic numpy test (#1777)

* Add basic numpy tests * lint
pytorch · Apr 8, 2024 · ef4a28a · ef4a28a
1 parent 0f4e159
commit ef4a28a
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/test/smoke_test/smoke_test.py b/test/smoke_test/smoke_test.py
@@ -66,6 +66,11 @@ def load_json_from_basedir(filename: str):
 def read_release_matrix():
     return load_json_from_basedir("release_matrix.json")
 
+def test_numpy():
+    import numpy as np
+    x = np.arange(5)
+    torch.tensor(x)
+
 def check_version(package: str) -> None:
     release_version = os.getenv("RELEASE_VERSION")
     # if release_version is specified, use it to validate the packages
@@ -307,6 +312,7 @@ def main() -> None:
     check_version(options.package)
     smoke_test_conv2d()
     test_linalg()
+    test_numpy()
     if is_cuda_system:
         test_linalg("cuda")