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

fix: remove glog/cuda_ops_api.h dependency from all xllm CUDA kernels

Browse Source 
cuda_ops_api.h includes glog/logging.h and ATen/DynamicLibrary.h
which are not available in corex standalone compilation.

All kernel .cu files only need device_utils.cuh (provides namespace,
XLLM_KERNEL_ATTR macro, CUB includes, type helpers).

Fixed files:
  norm.cu, activation.cu, rope.cu, block_copy.cu, reshape_paged_cache.cu
  moe/moe_combine.cu, moe/moe_compute_index.cu, moe/moe_fused_topk.cu
This commit is contained in:
claude
2026-08-14 10:30:10 +00:00
parent 51cb90b9ab
commit 0359103b9b
8 changed files with 9 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
ex_engine/xllm_kernels/cuda/moe/moe_combine.cu
View File

@@ -28,7 +28,7 @@ limitations under the License.
#include <c10/cuda/CUDAGuard.h>
#include "device_utils.cuh"
#include "kernels/cuda/cuda_ops_api.h"
#include "device_utils.cuh"
namespace xllm::kernel::cuda {

2
ex_engine/xllm_kernels/cuda/moe/moe_compute_index.cu
View File

@@ -27,7 +27,7 @@ limitations under the License.
#include <cub/block/block_scan.cuh>
#include "kernels/cuda/cuda_ops_api.h"
#include "device_utils.cuh"
namespace xllm::kernel::cuda {

2
ex_engine/xllm_kernels/cuda/moe/moe_fused_topk.cu
View File

@@ -15,7 +15,7 @@ limitations under the License.
#if defined(USE_DCU)
#include "kernels/dcu/dcu_ops_api.h"
#else
#include "kernels/cuda/cuda_ops_api.h"
#include "device_utils.cuh"
#endif
#include "moe_topk_sigmoid_kernels.cuh"
#include "moe_topk_softmax_kernels.cuh"