Skip to content

GitLab

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():
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:
# 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(),
......
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