Fix breakage caused by PyTorch PR #43528

jithunnair-amd · facebook-github-bot · commit 429f33e1e5cc · 2020-09-28T14:14:28.000-07:00
Summary: Fixes facebookresearch#2068 cc hgaspar Pull Request resolved: facebookresearch#2069 Reviewed By: alexander-kirillov Differential Revision: D23949698 Pulled By: ppwwyyxx fbshipit-source-id: 8d8ae859828058a3cb536eb825ecbf2921a9f7e9
diff --git a/setup.py b/setup.py
@@ -87,6 +87,11 @@ def get_extensions():
         )
 
     sources = [main_source] + sources
+    sources = [
+        s
+        for s in sources
+        if not is_rocm_pytorch or torch_ver < [1, 7] or not s.endswith("hip/vision.cpp")
+    ]
 
     extension = CppExtension
 

Original file line number	Diff line number	Diff line change
`@@ -87,6 +87,11 @@ def get_extensions():`
`87`	`87`	`)`
`88`	`88`
`89`	`89`	`sources = [main_source] + sources`
	`90`	`+ sources = [`
	`91`	`+ s`
	`92`	`+ for s in sources`
	`93`	`+ if not is_rocm_pytorch or torch_ver < [1, 7] or not s.endswith("hip/vision.cpp")`
	`94`	`+ ]`
`90`	`95`
`91`	`96`	`extension = CppExtension`
`92`	`97`