Skip to content

GitLab

Commit 48217ab6 authored by maxiao's avatar maxiao
Browse files

sgl_kernel适配torch2.7

parent bd63af06
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 0 deletions
sgl-kernel/csrc/attention/merge_attn_states.cu
View file @ 48217ab6
#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