Rename files in fp_quantize op from quantize.* to fp_quantize.* (micr…

…osoft#5577) Fixes microsoft#5535. Todo: need to test.
Snowflake-Labs · Jun 10, 2024 · 5e5c8a7 · 5e5c8a7
1 parent c203830
commit 5e5c8a7
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git a/csrc/fp_quantizer/quantize.cpp → csrc/fp_quantizer/fp_quantize.cpp b/csrc/fp_quantizer/quantize.cpp → csrc/fp_quantizer/fp_quantize.cpp
@@ -3,7 +3,7 @@
 
 // DeepSpeed Team
 
-#include "quantize.h"
+#include "fp_quantize.h"
 
 #include <c10/cuda/CUDAStream.h>
 #include <torch/extension.h>

diff --git a/csrc/fp_quantizer/quantize.cu → csrc/fp_quantizer/fp_quantize.cu b/csrc/fp_quantizer/quantize.cu → csrc/fp_quantizer/fp_quantize.cu
@@ -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>

diff --git a/csrc/fp_quantizer/includes/quantize.h → csrc/fp_quantizer/includes/fp_quantize.h b/csrc/fp_quantizer/includes/quantize.h → csrc/fp_quantizer/includes/fp_quantize.h
diff --git a/op_builder/fp_quantizer.py b/op_builder/fp_quantizer.py
@@ -52,8 +52,8 @@ def filter_ccs(self, ccs):
 
     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):