Skip to content

Commit

Permalink
More tests nov16 (#1608)
Browse files Browse the repository at this point in the history
* tests

* test

* test

* test
  • Loading branch information
atalman committed Nov 16, 2023
1 parent 91a24a7 commit e688fc7
Showing 1 changed file with 7 additions and 9 deletions.
16 changes: 7 additions & 9 deletions test/smoke_test/smoke_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -19,7 +19,6 @@
package_type = os.getenv("MATRIX_PACKAGE_TYPE")
target_os = os.getenv("TARGET_OS")
BASE_DIR = Path(__file__).parent.parent.parent
release_matrix = None

is_cuda_system = gpu_arch_type == "cuda"
NIGHTLY_ALLOWED_DELTA = 3
Expand All @@ -41,6 +40,7 @@
},
]


class Net(nn.Module):
def __init__(self):
super().__init__()
Expand Down Expand Up @@ -69,6 +69,12 @@ def read_release_matrix():
return load_json_from_basedir("release_matrix.json")

def check_version(package: str) -> None:

# if release_version is specified, override stable_version coming binary matrix
if(release_version):
release_matrix = read_release_matrix()
stable_version = release_matrix["torch"]

# only makes sense to check nightly package where dates are known
if channel == "nightly":
check_nightly_binaries_date(package)
Expand Down Expand Up @@ -282,14 +288,6 @@ def main() -> None:
)
options = parser.parse_args()
print(f"torch: {torch.__version__}")
print(f"torch expected: {release_version}")

# if release_version is specified, override stable_version coming binary matrix
if(release_version):
print(f"reading release_matrix for: {release_version}")
release_matrix = read_release_matrix()
stable_version = release_matrix["torch"]
print(f"new stable version : {stable_version}")

check_version(options.package)
smoke_test_conv2d()
Expand Down

0 comments on commit e688fc7

Please sign in to comment.