github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

fix include path

pull/5452/head
xs_courtesy 2024-03-13 11:21:06 +08:00
parent b699f54007
commit c1c45e9d8e
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
extensions/csrc/cuda/pybind/layer_norm.cpp
View File

@ -7,7 +7,7 @@
#include <cassert>
#include <vector>
#include "../common/micros.h"
#include "../../common/micros.h"
namespace {

2
extensions/moe/moe_cuda.py
View File

@ -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):