[Fix] Try to fix nvcc compilation error (#21246)

This commit is contained in:
DarkSharpness
2026-03-26 10:59:36 +08:00
committed by GitHub
parent a34e9ed64a
commit bb29893689
7 changed files with 75 additions and 9 deletions
+2
View File
@@ -131,6 +131,8 @@ def get_custom_all_reduce_cls() -> type[CustomAllReduceObj]:
@tvm_ffi.register_object("sgl.CustomAllReduce")
class CustomAllReduceObjReal(tvm_ffi.Object):
__slots__ = ("__dict__",)
def __init__(
self,
rank: int,
@@ -194,8 +194,8 @@ struct CustomAllReduceBase : public tvm::ffi::Object {
for (const auto j : irange(new_registered_count)) {
/// NOTE: structural binding will cause intern compiler error...
const auto elem = array[j];
const auto offset = get<0>(elem);
const auto ipc_handle = get<1>(elem);
const auto offset = elem.get<0>();
const auto ipc_handle = elem.get<1>();
data[j].input[i] = pointer::offset(open_cached(ipc_handle), offset);
}
}
@@ -34,15 +34,13 @@ from sglang.srt.distributed.device_communicators.custom_all_reduce_v2 import (
from sglang.test.ci.ci_register import register_cuda_ci
register_cuda_ci(
est_time=120,
suite="stage-b-kernel-unit-1-gpu-large",
disabled="requires multi-GPU distributed setup",
est_time=500,
suite="stage-b-kernel-unit-8-gpu-h200",
)
register_cuda_ci(
est_time=120,
suite="nightly-kernel-1-gpu",
est_time=500,
suite="nightly-kernel-8-gpu-h200",
nightly=True,
disabled="requires multi-GPU distributed setup",
)
# ---------------------------------------------------------------------------