diff --git a/extensions/csrc/cuda/pybind/layer_norm.cpp b/extensions/csrc/cuda/pybind/layer_norm.cpp index 3439e5e71..b1f7c2543 100644 --- 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 index 722daae33..7a4744d4d 100644 --- a/extensions/moe/moe_cuda.py +++ b/extensions/moe/moe_cuda.py @@ -11,7 +11,7 @@ class MoeCudaExtension(_CudaExtension): 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):