From 23639bc93a72ac174fb347e41afca8d931e2829f Mon Sep 17 00:00:00 2001 From: atalman Date: Mon, 8 Apr 2024 09:15:11 -0700 Subject: [PATCH 1/2] Add basic numpy tests --- test/smoke_test/smoke_test.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/test/smoke_test/smoke_test.py b/test/smoke_test/smoke_test.py index cf78eb9e1..2e8a31957 100644 --- 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) + t = 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") From eb263f511c9d8d2c9572b944745a8c27e0cc659c Mon Sep 17 00:00:00 2001 From: atalman Date: Mon, 8 Apr 2024 09:21:08 -0700 Subject: [PATCH 2/2] lint --- test/smoke_test/smoke_test.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/test/smoke_test/smoke_test.py b/test/smoke_test/smoke_test.py index 2e8a31957..7d4d0f862 100644 --- a/test/smoke_test/smoke_test.py +++ b/test/smoke_test/smoke_test.py @@ -69,7 +69,7 @@ def read_release_matrix(): def test_numpy(): import numpy as np x = np.arange(5) - t = torch.tensor(x) + torch.tensor(x) def check_version(package: str) -> None: release_version = os.getenv("RELEASE_VERSION")