Skip to content

Commit ef4a28a

Browse files
authored
Add basic numpy test (#1777)
* Add basic numpy tests * lint
1 parent 0f4e159 commit ef4a28a

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

test/smoke_test/smoke_test.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -66,6 +66,11 @@ def load_json_from_basedir(filename: str):
6666
def read_release_matrix():
6767
return load_json_from_basedir("release_matrix.json")
6868

69+
def test_numpy():
70+
import numpy as np
71+
x = np.arange(5)
72+
torch.tensor(x)
73+
6974
def check_version(package: str) -> None:
7075
release_version = os.getenv("RELEASE_VERSION")
7176
# if release_version is specified, use it to validate the packages
@@ -307,6 +312,7 @@ def main() -> None:
307312
check_version(options.package)
308313
smoke_test_conv2d()
309314
test_linalg()
315+
test_numpy()
310316
if is_cuda_system:
311317
test_linalg("cuda")
312318

0 commit comments

Comments
 (0)