EngineX-Mthreads/enginex-mthreads-vllm
4
0
Fork 0
Code Issues Pull Requests Actions Projects Releases Wiki Activity

refactor: 统一硬件相关头文件引用

Browse Source 
将分散在各文件中的CUDA/HIP/MUSA硬件相关头文件引用统一到vendors目录下的对应头文件中，提高代码可维护性。移除重复的头文件引用，优化构建配置。
This commit is contained in:
xiezhongtao
2026-01-20 10:14:31 +08:00
parent 5aef6c175a
commit 2bd9bd4cc2
98 changed files with 1757 additions and 1286 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
csrc/sparse/cutlass/sparse_compressor_c3x.cuh
View File

@@ -2,7 +2,7 @@
// clang-format will break include orders
// clang-format off
#include <cudaTypedefs.h>
#include "../../vendors/functions.h"
#if defined CUDA_VERSION && CUDA_VERSION >= 12020
#include "sparse_scaled_mm_c3x.cuh"

2
csrc/sparse/cutlass/sparse_scaled_mm_c3x.cu
View File

@@ -1,6 +1,6 @@
// clang-format will break include orders
// clang-format off
#include <cudaTypedefs.h>
#include "../../vendors/functions.h"
#if defined CUDA_VERSION && CUDA_VERSION >= 12020
#include "sparse_scaled_mm_c3x.cuh"

5
csrc/sparse/cutlass/sparse_scaled_mm_c3x.cuh
View File

@@ -2,11 +2,8 @@
// clang-format will break include orders
// clang-format off
#include <cudaTypedefs.h>
#include "../../vendors/functions.h"
#include <torch/all.h>
#include <ATen/cuda/CUDAContext.h>
#include "cuda_utils.h"

4
csrc/sparse/cutlass/sparse_scaled_mm_entry.cu
View File

@@ -1,7 +1,5 @@
#include <cudaTypedefs.h>
#include "../../vendors/functions.h"
#include <c10/cuda/CUDAGuard.h>
#include <torch/all.h>
#include "cutlass_extensions/common.hpp"