Skip to content

[Feat] Wire TransProviderType::AIV to the static-library backend#1028

Merged
Infinite666 merged 3 commits into
ModelEngine-Group:feature_26h1from
hczl:feat/aiv-transport-wiring
Jun 17, 2026
Merged

[Feat] Wire TransProviderType::AIV to the static-library backend#1028
Infinite666 merged 3 commits into
ModelEngine-Group:feature_26h1from
hczl:feat/aiv-transport-wiring

Conversation

@hczl

@hczl hczl commented Jun 16, 2026

Copy link
Copy Markdown

内容

asu_transport_impl.cppcase TransProviderType::AIV 的 UNSUPPORTED 占位替换为构造 AIVTransProvider;CMake 增加 ASU_AIV_PROVIDER_ROOT 配置项

Comment thread ucm/transport/kv/asu/trans/include/aiv_transport/types.h Outdated
@hczl hczl force-pushed the feat/aiv-transport-wiring branch 5 times, most recently from 7861953 to 0e9ba55 Compare June 16, 2026 08:12
Comment thread ucm/transport/kv/asu/trans/include/aiv_transport/aiv_transport.h Outdated
Comment thread ucm/transport/kv/asu/trans/src/aiv_trans_provider.h Outdated
@hczl hczl force-pushed the feat/aiv-transport-wiring branch 4 times, most recently from ea489c1 to bc9d200 Compare June 16, 2026 09:52
Comment thread ucm/transport/kv/asu/trans/src/asu_transport_impl.cpp
@hczl hczl force-pushed the feat/aiv-transport-wiring branch 2 times, most recently from 90ceeba to 2777dfb Compare June 17, 2026 02:37
hczl and others added 3 commits June 17, 2026 10:53
Vendor the public interface shipped by libumc_asu_transport_provider.a
(npu2cpu_rdma) under trans/include/aiv_transport/: aiv_transport.h declares
the UC::ASU::AIVTransport contract plus the CreateAIVTransProvider() factory.
Status/StatusCode reuse the existing UCM asu_transport/types.h to avoid
shipping a duplicate vendor copy.
src/aiv_trans_provider.h provides a header-only AIVTransProvider : public
TransProvider that forwards every call to an impl_ created by
CreateAIVTransProvider() (libumc_asu_transport_provider.a). After renaming the
vendor contract to AIVTransport there is no longer a name clash with UCM's own
TransProvider, so the adapter includes aiv_transport/aiv_transport.h directly
and stays header-only, matching aicpu_trans_provider.h. SendIoBatch /
RegisterMemoryDesc / UnregisterMemoryDesc are rebuilt field-by-field when
forwarding since the vendor and UCM structs are now distinct types.
asu_transport_impl.cpp constructs AIVTransProvider for the AIV backend; the asu
CMake adds an ASU_AIV_PROVIDER_ROOT cache entry plus find_path / find_library
checks (FATAL_ERROR on miss, mirroring the existing ASCEND_ROOT / ASCENDCL
discovery style) and links libumc_asu_transport_provider.a.

Co-authored-by: Cursor <cursoragent@cursor.com>
@hczl hczl force-pushed the feat/aiv-transport-wiring branch from 2777dfb to 58c50a3 Compare June 17, 2026 02:53
@Infinite666 Infinite666 merged commit abf31d1 into ModelEngine-Group:feature_26h1 Jun 17, 2026
8 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants