Skip to content

GitLab

Unverified Commit 20d90dfc authored by Paul Mulders's avatar Paul Mulders Committed by GitHub
Browse files

setup.py: fix ROCm build (#7573)

parent 078959f5
Hide whitespace changes
Inline Side-by-side
Showing with 7 additions and 1 deletion
setup.py
View file @ 20d90dfc
...@@ -328,9 +328,15 @@ def get_extensions(): ...@@ -328,9 +328,15 @@ def get_extensions():
image_src = ( image_src = (
glob.glob(os.path.join(image_path, "*.cpp")) glob.glob(os.path.join(image_path, "*.cpp"))
+ glob.glob(os.path.join(image_path, "cpu", "*.cpp")) + glob.glob(os.path.join(image_path, "cpu", "*.cpp"))
+ glob.glob(os.path.join(image_path, "cuda", "*.cpp"))
) )
if is_rocm_pytorch:
image_src += glob.glob(os.path.join(image_path, "hip", "*.cpp"))
# we need to exclude this in favor of the hipified source
image_src.remove(os.path.join(image_path, "image.cpp"))
else:
image_src += glob.glob(os.path.join(image_path, "cuda", "*.cpp"))
if use_png or use_jpeg: if use_png or use_jpeg:
ext_modules.append( ext_modules.append(
extension( extension(
......
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