From 056808723ee135926cacc20a95bde8d0757724c8 Mon Sep 17 00:00:00 2001 From: Alan Kao Date: Mon, 17 Aug 2026 17:22:58 +0800 Subject: [PATCH] [AMD] Guard ROCm 7.0 build from using hipMemcpyBatchAsync (#35128) --- python/sglang/kernels/aot/csrc/kvcacheio/transfer.cu | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/python/sglang/kernels/aot/csrc/kvcacheio/transfer.cu b/python/sglang/kernels/aot/csrc/kvcacheio/transfer.cu index bca392314..6b776a202 100644 --- a/python/sglang/kernels/aot/csrc/kvcacheio/transfer.cu +++ b/python/sglang/kernels/aot/csrc/kvcacheio/transfer.cu @@ -995,9 +995,9 @@ inline void transfer_kv_page_first_direct_impl( }; #if defined(USE_ROCM) || !defined(CUDA_VERSION) || CUDA_VERSION < 12080 -#if defined(USE_ROCM) - // Opt-in HIP batch copy path (mirrors cudaMemcpyBatchAsync); disabled by - // default, falls back to per-page copy below. +#if defined(USE_ROCM) && defined(HIP_VERSION) && HIP_VERSION >= 70200000 + // Opt-in HIP batch copy path (mirrors cudaMemcpyBatchAsync); requires + // ROCm >= 7.2. Disabled by default, falls back to per-page copy below. constexpr bool kEnableHipBatch = false; if (kEnableHipBatch) { std::vector b_srcs, b_dsts;