dylanyunlong/project_6
1
0
Fork 0
Code Issues Pull Requests Actions Projects Releases Wiki Activity

fix: MoE kernel include paths — device_utils.cuh + arch_condition.h

Browse Source 
Fixed xllm internal paths to our headers/ directory:
  kernels/cuda/device_utils.cuh → device_utils.cuh
  core/kernels/cuda/device_utils.cuh → device_utils.cuh
  core/kernels/cuda/arch_condition.h → arch_condition.h (copied)
This commit is contained in:
claude
2026-08-14 11:31:54 +00:00
parent df6a0f5d47
commit 31d3ee99bb
4 changed files with 112 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
ex_engine/xllm_kernels/cuda/moe/moe_topk.cuh
View File

@@ -35,14 +35,14 @@
#include <hipcub/hipcub.hpp>
#endif
#include "core/kernels/cuda/arch_condition.h"
#include "arch_condition.h"
#if defined(USE_DCU)
#include <hip/hip_bfloat16.h>
#include <hip/hip_fp16.h>
#endif
#include "core/kernels/cuda/device_utils.cuh"
#include "device_utils.cuh"
namespace xllm::kernel::cuda {
namespace reduce_topk {

2
ex_engine/xllm_kernels/cuda/moe/moe_topk_sigmoid_kernels.cuh
View File

@@ -26,7 +26,7 @@ limitations under the License.
#if !defined(USE_DCU) && !defined(USE_MACA)
#endif
#include "kernels/cuda/device_utils.cuh"
#include "device_utils.cuh"
namespace {

2
ex_engine/xllm_kernels/cuda/moe/moe_topk_softmax_kernels.cuh
View File

@@ -26,7 +26,7 @@ limitations under the License.
#if !defined(USE_DCU) && !defined(USE_MACA)
#endif
#include "kernels/cuda/device_utils.cuh"
#include "device_utils.cuh"
using cub_kvp = cub::KeyValuePair<int, float>;