Skip to content

Commit 5a361e1

Browse files
rraminenloadams
andauthored
[ROCm] Enable fp_quantizer on ROCm (#7027)
This change is required to successfully build fp_quantizer extension on ROCm. --------- Co-authored-by: Logan Adams <114770087+loadams@users.noreply.github.com>
1 parent 079de6b commit 5a361e1

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

op_builder/fp_quantizer.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -94,7 +94,10 @@ def sources(self):
9494
]
9595

9696
def extra_ldflags(self):
97-
return ['-lcurand']
97+
if not self.is_rocm_pytorch():
98+
return ['-lcurand']
99+
else:
100+
return []
98101

99102
def include_paths(self):
100103
return ['csrc/fp_quantizer/includes', 'csrc/includes']

0 commit comments

Comments
 (0)