github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-06-01 03:45:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #5452 from Courtesy-Xs/fix_include_path

Browse Source 
fix include path
This commit is contained in:
傅剑寒 2024-03-13 11:26:41 +08:00 committed by GitHub
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 <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):