Skip to content

GitLab

Commit de61a992 authored by maxiao1's avatar maxiao1
Browse files

Merge branch 'v0.5.4_torch2.7' into 'v0.5.4_dev' 

sgl_kernel适配torch2.7

See merge request OpenDAS/sglang!26
parents bd63af06 48217ab6
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 0 deletions
sgl-kernel/csrc/attention/merge_attn_states.cu
View file @ de61a992
#include <ATen/cuda/CUDAContext.h> #include <ATen/cuda/CUDAContext.h>
#include <c10/cuda/CUDAGuard.h> #include <c10/cuda/CUDAGuard.h>
#ifdef __HIP_PLATFORM_AMD__
#include <hip/hip_bf16.h>
#endif
#include <algorithm> #include <algorithm>
#include <optional> #include <optional>
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment