github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-01 17:17:05 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix include path

Browse Source
This commit is contained in:
xs_courtesy
2024-03-13 11:21:06 +08:00
parent b699f54007
commit c1c45e9d8e
2 changed files with 2 additions and 2 deletions
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):