From a37de789b458c72960a61da01ec453e270a8ed26 Mon Sep 17 00:00:00 2001 From: chenkun Date: Thu, 31 Jul 2025 17:29:14 +0800 Subject: [PATCH] HcclAllGatherV add A3 --- torch_npu/csrc/distributed/HcclCompile.h | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/torch_npu/csrc/distributed/HcclCompile.h b/torch_npu/csrc/distributed/HcclCompile.h index de4a2ba619..685bb8ee0b 100644 --- a/torch_npu/csrc/distributed/HcclCompile.h +++ b/torch_npu/csrc/distributed/HcclCompile.h @@ -171,8 +171,9 @@ bool hcclAllGatherVExist() c10::call_once(flag, [&]() { auto func = GET_FUNC(HcclAllGatherV) if (func != nullptr && - c10_npu::GetSocVersion() >= c10_npu::SocVersion::Ascend310P1 && - c10_npu::GetSocVersion() < c10_npu::SocVersion::Ascend310B1) { + ((c10_npu::GetSocVersion() >= c10_npu::SocVersion::Ascend310P1 && + c10_npu::GetSocVersion() < c10_npu::SocVersion::Ascend310B1) || + c10_npu::GetSocVersion() >= c10_npu::SocVersion::Ascend910_9391)) { exist = true; } }); -- Gitee