Add nccl version print for cuda related smoke test (#1667)

atalman · web-flow · commit 04ef1bf031cc · 2024-01-08T11:53:56.000-05:00
diff --git a/test/smoke_test/smoke_test.py b/test/smoke_test/smoke_test.py
@@ -163,6 +163,7 @@ def smoke_test_cuda(package: str, runtime_error_check: str) -> None:
                 f"Wrong CUDA version. Loaded: {torch.version.cuda} Expected: {gpu_arch_ver}"
             )
         print(f"torch cuda: {torch.version.cuda}")
+        print(f"torch nccl version: {torch.cuda.nccl.version()}" )
         # todo add cudnn version validation
         print(f"torch cudnn: {torch.backends.cudnn.version()}")
         print(f"cuDNN enabled? {torch.backends.cudnn.enabled}")

Original file line number	Diff line number	Diff line change
`@@ -163,6 +163,7 @@ def smoke_test_cuda(package: str, runtime_error_check: str) -> None:`
`163`	`163`	`f"Wrong CUDA version. Loaded: {torch.version.cuda} Expected: {gpu_arch_ver}"`
`164`	`164`	`)`
`165`	`165`	`print(f"torch cuda: {torch.version.cuda}")`
	`166`	`+ print(f"torch nccl version: {torch.cuda.nccl.version()}" )`
`166`	`167`	`# todo add cudnn version validation`
`167`	`168`	`print(f"torch cudnn: {torch.backends.cudnn.version()}")`
`168`	`169`	`print(f"cuDNN enabled? {torch.backends.cudnn.enabled}")`