Skip to content

GitLab

Unverified Commit 60e2fdc1 authored by Prabhat Roy's avatar Prabhat Roy Committed by GitHub
Browse files

Removed unnecessary if from setup.py (#4739) 

* Removed unnecessary if from setup.py

* Removed empty line
parent dc113995
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 2 deletions
setup.py
View file @ 60e2fdc1
...@@ -143,7 +143,7 @@ def get_extensions(): ...@@ -143,7 +143,7 @@ def get_extensions():
if torch.__version__ >= "1.5": if torch.__version__ >= "1.5":
from torch.utils.cpp_extension import ROCM_HOME from torch.utils.cpp_extension import ROCM_HOME
is_rocm_pytorch = True if ((torch.version.hip is not None) and (ROCM_HOME is not None)) else False is_rocm_pytorch = (torch.version.hip is not None) and (ROCM_HOME is not None)
if is_rocm_pytorch: if is_rocm_pytorch:
from torch.utils.hipify import hipify_python from torch.utils.hipify import hipify_python
...@@ -159,7 +159,6 @@ def get_extensions(): ...@@ -159,7 +159,6 @@ def get_extensions():
# Copy over additional files # Copy over additional files
for file in glob.glob(r"torchvision/csrc/ops/cuda/*.h"): for file in glob.glob(r"torchvision/csrc/ops/cuda/*.h"):
shutil.copy(file, "torchvision/csrc/ops/hip") shutil.copy(file, "torchvision/csrc/ops/hip")
else: else:
source_cuda = glob.glob(os.path.join(extensions_dir, "ops", "cuda", "*.cu")) source_cuda = glob.glob(os.path.join(extensions_dir, "ops", "cuda", "*.cu"))
......
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