EngineX-Hygon/sglang
5
0
Fork 0
Code Issues Pull Requests Actions 7 Projects Releases Wiki Activity

change sgl_kernel WARP_SIZE to 64

Browse Source
This commit is contained in:
maxiao1
2025-11-03 10:17:53 +08:00
parent 8fc552638f
commit 75cd34d172
5 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
sgl-kernel/csrc/kvcacheio/transfer.cu
View File

@@ -5,7 +5,7 @@
#include <cstdint>
#ifndef USE_ROCM
#define WARP_SIZE 32
#define WARP_SIZE 64
#include "pytorch_extension_utils.h"
#else
#include "pytorch_extension_utils_rocm.h"