Fix path to included and renamed fp_quantize.h

deepspeedai · May 28, 2024 · 1bb692a · 1bb692a
1 parent e32321c
commit 1bb692a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/csrc/fp_quantizer/fp_quantize.cpp b/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/fp_quantize.cu b/csrc/fp_quantizer/fp_quantize.cu
@@ -6,7 +6,7 @@
 #include <stdexcept>
 #include "context.h"
 #include "memory_access_utils.h"
-#include "quantize.h"
+#include "fp_quantize.h"
 #include "reduction_utils.h"
 
 #include <cuda.h>