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

Merge pull request #5452 from Courtesy-Xs/fix_include_path 

fix include path
pull/5454/head
傅剑寒 2024-03-13 11:26:41 +08:00 committed by GitHub
parent b699f54007 c1c45e9d8e
commit 6fd355a5a6
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
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 <cassert>
#include <vector> #include <vector>
#include "../common/micros.h" #include "../../common/micros.h"
namespace { namespace {

2
extensions/moe/moe_cuda.py
View File

@ -11,7 +11,7 @@ class MoeCudaExtension(_CudaExtension):
return ret return ret
def sources_files(self): 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 return ret
def cxx_flags(self): def cxx_flags(self):