Rename files in fp_quantize op from quantize.* to fp_quantize.* (#5577)

Fixes #5535.

Todo: need to test.
This commit is contained in:
Logan Adams 2024-05-28 13:01:48 -07:00 коммит произвёл GitHub
Родитель fd8051a69c
Коммит bf66acdbae
Не найден ключ, соответствующий данной подписи
Идентификатор ключа GPG: B5690EEEBB952194
4 изменённых файлов: 4 добавлений и 4 удалений

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

@ -3,7 +3,7 @@
// DeepSpeed Team
#include "quantize.h"
#include "fp_quantize.h"
#include <c10/cuda/CUDAStream.h>
#include <torch/extension.h>

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

@ -5,8 +5,8 @@
#include <stdexcept>
#include "context.h"
#include "fp_quantize.h"
#include "memory_access_utils.h"
#include "quantize.h"
#include "reduction_utils.h"
#include <cuda.h>

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

@ -52,8 +52,8 @@ class FPQuantizerBuilder(CUDAOpBuilder):
def sources(self):
return [
"csrc/fp_quantizer/quantize.cu",
"csrc/fp_quantizer/quantize.cpp",
"csrc/fp_quantizer/fp_quantize.cu",
"csrc/fp_quantizer/fp_quantize.cpp",
]
def extra_ldflags(self):