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 коммит произвёл GitHub
Родитель 6112d08ca7
Коммит f2d4845f07
Не найден ключ, соответствующий данной подписи
Идентификатор ключа GPG: 4AEE18F83AFDEB23
1 изменённых файлов: 4 добавлений и 1 удалений

Просмотреть файл

@ -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 []