Enable quantizer op on ROCm (#4114)

This PR is to enable quantizer extension on ROCm

cc: @jithunnair-amd

---------

Co-authored-by: Logan Adams <114770087+loadams@users.noreply.github.com>
This commit is contained in:
Ramya Ramineni
2024-01-10 17:48:54 -06:00
committed by GitHub
parent 6112d08ca7
commit f2d4845f07

View File

@ -32,4 +32,7 @@ class QuantizerBuilder(CUDAOpBuilder):
return ['csrc/includes']
def extra_ldflags(self):
return ['-lcurand']
if not self.is_rocm_pytorch():
return ['-lcurand']
else:
return []