Skip to content

Commit 9af448e

Browse files
committed
change the order of cuda version check and date check
1 parent f656e90 commit 9af448e

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

test/smoke_test/smoke_test.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -65,7 +65,7 @@ def smoke_test_cuda() -> None:
6565

6666

6767
if installation_str.find('nightly') != -1:
68-
# just print out cuda version, as version check were perform during import
68+
# just print out cuda version, as version check were already performed during import
6969
print(f"torchvision cuda: {torch.ops.torchvision._cuda_version()}")
7070
print(f"torchaudio cuda: {torch.ops.torchaudio.cuda_version()}")
7171
else:
@@ -152,9 +152,9 @@ def main() -> None:
152152
print(f"torch: {torch.__version__}")
153153
print(f"torchvision: {torchvision.__version__}")
154154
print(f"torchaudio: {torchaudio.__version__}")
155+
smoke_test_cuda()
155156
if installation_str.find('nightly') != -1:
156157
check_nightly_binaries_date()
157-
smoke_test_cuda()
158158
#smoke_test_conv2d()
159159
#smoke_test_torchaudio()
160160
#smoke_test_torchvision()

0 commit comments

Comments
 (0)