saymrwulf/pytorch
1
0
Fork 0
mirror of https://github.com/saymrwulf/pytorch.git synced 2026-05-14 20:57:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

test_cuda

Browse source
This commit is contained in:
atalman 2025-02-09 13:16:05 -08:00
parent 44202a1019
commit 0972ac8013
1 changed files with 9 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
setup.py
View file

@ -402,15 +402,15 @@ def get_submodule_folders():
def checkout_nccl():
report(f'-- Checkout nccl: {get_cmake_cache_vars()["USE_CUDA"]}')
if get_cmake_cache_vars()["USE_CUDA"]:
cuda_version = os.getenv("CUDA_VERSION", "")
commit_hash = "80f6bda4378b99d99e82b4d76a633791cc45fef0"
if cuda_version.startswith("11.8"):
commit_hash = "ab2b89c4c339bd7f816fbc114a4b05d386b66290"
nccl_basedir = os.path.join(cwd, "nccl")
report(f"-- Calling nccl checkout: {commit_hash}")
subprocess.check_call(["git", "clone", "https://github.com/NVIDIA/nccl.git", "nccl"], cwd=cwd)
subprocess.check_call(["git", "checkout", commit_hash], cwd=nccl_basedir)
#if get_cmake_cache_vars()["USE_CUDA"]:
cuda_version = os.getenv("DESIRED_CUDA", "")
commit_hash = "80f6bda4378b99d99e82b4d76a633791cc45fef0"
if cuda_version.startswith("11.8"):
commit_hash = "ab2b89c4c339bd7f816fbc114a4b05d386b66290"
nccl_basedir = os.path.join(cwd, "nccl")
report(f"-- Calling nccl checkout: {commit_hash}")
subprocess.check_call(["git", "clone", "https://github.com/NVIDIA/nccl.git", "nccl"], cwd=cwd)
subprocess.check_call(["git", "checkout", commit_hash], cwd=nccl_basedir)
def check_submodules():
def check_for_files(folder, files):