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

fix: activation.cu torch/extension.h + LOG(FATAL)→TORCH_CHECK, reshape_paged_cache.cu torch header

Browse Source 
xllm_norm.so: ✓ COMPILED AND LOADED (rms_norm, fused_add_rms_norm)

activation.cu fixes:
  - Add #include <torch/extension.h> (torch::Tensor not visible from torch/cuda.h alone)
  - Replace LOG(FATAL) with TORCH_CHECK (no glog)

reshape_paged_cache.cu:
  - Add #include <torch/extension.h>
This commit is contained in:
claude
2026-08-14 10:52:54 +00:00
parent 093bfb380f
commit a206fc1d43
2 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
ex_engine/xllm_kernels/cuda/reshape_paged_cache.cu
View File

@@ -14,6 +14,7 @@ limitations under the License.
==============================================================================*/
#include <c10/cuda/CUDAStream.h>
#include <torch/extension.h>
#include "device_utils.cuh"