зеркало из https://github.com/microsoft/DeepSpeed.git
Rename files in fp_quantize op from quantize.* to fp_quantize.* (#5577)
Fixes #5535. Todo: need to test.
This commit is contained in:
Родитель
fd8051a69c
Коммит
bf66acdbae
|
@ -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):
|
||||
|
|
Загрузка…
Ссылка в новой задаче