fix custom_allreduce namespace (#6039)

This commit is contained in:
Xiaoyu Zhang
2025-05-07 10:13:06 +08:00
committed by GitHub
parent 8a828666a3
commit d25398cbc8
4 changed files with 24 additions and 24 deletions

View File

@@ -15,7 +15,7 @@
#include "utils.h"
namespace vllm {
namespace sglang {
constexpr int kMaxBlocks = 36;
// Counter may overflow, but it's fine since unsigned int overflow is
@@ -483,7 +483,7 @@ class CustomAllreduce {
/**
* To inspect PTX/SASS, copy paste this header file to compiler explorer and add
a template instantiation:
* template void vllm::CustomAllreduce::allreduce<half>(cudaStream_t, half *,
* template void sglang::CustomAllreduce::allreduce<half>(cudaStream_t, half *,
half *, int, int, int);
*/
} // namespace vllm
} // namespace sglang