mirror of
https://github.com/deepspeedai/DeepSpeed.git
synced 2025-10-20 15:33:51 +08:00
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:
@ -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 []
|
||||
|
Reference in New Issue
Block a user