|
19 | 19 | package_type = os.getenv("MATRIX_PACKAGE_TYPE")
|
20 | 20 | target_os = os.getenv("TARGET_OS")
|
21 | 21 | BASE_DIR = Path(__file__).parent.parent.parent
|
22 |
| -release_matrix = None |
23 | 22 |
|
24 | 23 | is_cuda_system = gpu_arch_type == "cuda"
|
25 | 24 | NIGHTLY_ALLOWED_DELTA = 3
|
|
41 | 40 | },
|
42 | 41 | ]
|
43 | 42 |
|
| 43 | + |
44 | 44 | class Net(nn.Module):
|
45 | 45 | def __init__(self):
|
46 | 46 | super().__init__()
|
@@ -69,6 +69,12 @@ def read_release_matrix():
|
69 | 69 | return load_json_from_basedir("release_matrix.json")
|
70 | 70 |
|
71 | 71 | 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 | + |
72 | 78 | # only makes sense to check nightly package where dates are known
|
73 | 79 | if channel == "nightly":
|
74 | 80 | check_nightly_binaries_date(package)
|
@@ -282,14 +288,6 @@ def main() -> None:
|
282 | 288 | )
|
283 | 289 | options = parser.parse_args()
|
284 | 290 | 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}") |
293 | 291 |
|
294 | 292 | check_version(options.package)
|
295 | 293 | smoke_test_conv2d()
|
|
0 commit comments