зеркало из https://github.com/microsoft/DeepSpeed.git
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:
Родитель
6112d08ca7
Коммит
f2d4845f07
|
@ -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 []
|
||||
|
|
Загрузка…
Ссылка в новой задаче