Skip to content

GitLab

"...en/git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "8f8888a76ec16ea7afc2cc8e9be04bd8cccf6b37"
Unverified Commit 62f06964 authored by Hubert Lu's avatar Hubert Lu Committed by GitHub
Browse files

Update setup.py 


Co-authored-by: default avatarJeff Daily <jeff.daily@amd.com>
parent 4b15f641
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 5 deletions
setup.py
View file @ 62f06964
...@@ -34,11 +34,6 @@ def check_if_rocm_pytorch(): ...@@ -34,11 +34,6 @@ def check_if_rocm_pytorch():
IS_ROCM_PYTORCH = check_if_rocm_pytorch() IS_ROCM_PYTORCH = check_if_rocm_pytorch()
if IS_ROCM_PYTORCH:
rocm_include_dirs = ["/opt/rocm/include/hiprand", "/opt/rocm/include/rocrand"]
else:
rocm_include_dirs = []
if not torch.cuda.is_available() and not IS_ROCM_PYTORCH: if not torch.cuda.is_available() and not IS_ROCM_PYTORCH:
# https://github.com/NVIDIA/apex/issues/486 # https://github.com/NVIDIA/apex/issues/486
# Extension builds after https://github.com/pytorch/pytorch/pull/23408 attempt to query torch.cuda.get_device_capability(), # Extension builds after https://github.com/pytorch/pytorch/pull/23408 attempt to query torch.cuda.get_device_capability(),
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment