saymrwulf/pytorch
1
0
Fork 0
mirror of https://github.com/saymrwulf/pytorch.git synced 2026-05-14 20:57:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add -DHIP_ENABLE_WARP_SYNC_BUILTINS to cpp_extensions for torchvision

Browse source
This commit is contained in:
Prachi Gupta 2025-02-07 22:05:15 +00:00
parent b134bfade7
commit 75800c9f4b
1 changed files with 1 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
torch/utils/cpp_extension.py
View file

@ -280,6 +280,7 @@ COMMON_HIPCC_FLAGS = [
'-DCUDA_HAS_FP16=1',
'-D__HIP_NO_HALF_OPERATORS__=1',
'-D__HIP_NO_HALF_CONVERSIONS__=1',
'-DHIP_ENABLE_WARP_SYNC_BUILTINS=1'
]
JIT_EXTENSION_VERSIONER = ExtensionVersioner()