Merge pull request hpcaitech#5452 from Courtesy-Xs/fix_include_path

fix include path
SunflowerAries · Mar 13, 2024 · 6fd355a · 6fd355a
2 parents b699f54 + c1c45e9
commit 6fd355a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/extensions/csrc/cuda/pybind/layer_norm.cpp b/extensions/csrc/cuda/pybind/layer_norm.cpp
@@ -7,7 +7,7 @@
 #include <cassert>
 #include <vector>
 
-#include "../common/micros.h"
+#include "../../common/micros.h"
 
 namespace {
 

diff --git a/extensions/moe/moe_cuda.py b/extensions/moe/moe_cuda.py
@@ -11,7 +11,7 @@ def include_dirs(self):
         return ret
 
     def sources_files(self):
-        ret = [self.csrc_abs_path(fname) for fname in ["cuda/moe.cpp", "cuda/moe_kernel.cu"]]
+        ret = [self.csrc_abs_path(fname) for fname in ["cuda/pybind/moe.cpp", "cuda/moe_kernel.cu"]]
         return ret
 
     def cxx_flags(self):