Skip to content

Commit e688fc7

Browse files
authored
More tests nov16 (#1608)
* tests * test * test * test
1 parent 91a24a7 commit e688fc7

File tree

1 file changed

+7
-9
lines changed

1 file changed

+7
-9
lines changed

test/smoke_test/smoke_test.py

Lines changed: 7 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,6 @@
1919
package_type = os.getenv("MATRIX_PACKAGE_TYPE")
2020
target_os = os.getenv("TARGET_OS")
2121
BASE_DIR = Path(__file__).parent.parent.parent
22-
release_matrix = None
2322

2423
is_cuda_system = gpu_arch_type == "cuda"
2524
NIGHTLY_ALLOWED_DELTA = 3
@@ -41,6 +40,7 @@
4140
},
4241
]
4342

43+
4444
class Net(nn.Module):
4545
def __init__(self):
4646
super().__init__()
@@ -69,6 +69,12 @@ def read_release_matrix():
6969
return load_json_from_basedir("release_matrix.json")
7070

7171
def check_version(package: str) -> None:
72+
73+
# if release_version is specified, override stable_version coming binary matrix
74+
if(release_version):
75+
release_matrix = read_release_matrix()
76+
stable_version = release_matrix["torch"]
77+
7278
# only makes sense to check nightly package where dates are known
7379
if channel == "nightly":
7480
check_nightly_binaries_date(package)
@@ -282,14 +288,6 @@ def main() -> None:
282288
)
283289
options = parser.parse_args()
284290
print(f"torch: {torch.__version__}")
285-
print(f"torch expected: {release_version}")
286-
287-
# if release_version is specified, override stable_version coming binary matrix
288-
if(release_version):
289-
print(f"reading release_matrix for: {release_version}")
290-
release_matrix = read_release_matrix()
291-
stable_version = release_matrix["torch"]
292-
print(f"new stable version : {stable_version}")
293291

294292
check_version(options.package)
295293
smoke_test_conv2d()

0 commit comments

Comments
 (0)