Fix breakage caused by PyTorch PR
Summary:
Fixes
cc hgaspar
Pull Request resolved:
@@ -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
