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
committed by GitHub
parent fd8051a69c
commit bf66acdbae
4 changed files with 4 additions and 4 deletions

View File

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

View File

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

View File

@ -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):