feat(shmem): add MORI_MULTITHREAD_SUPPORT for SPMT#308

Merged

jhchouuu merged 15 commits into

mainfrom

jiahzhou/shmem-multithread-support

May 21, 2026

Commits on May 13, 2026

feat(shmem): add MORI_MULTITHREAD_SUPPORT for single-process multi-thread (SPMT)

jhchouuu
and
claude
committed
fix(shmem): SPMT review fixes — pybind dup, peer access, finalize leak

jhchouuu
and
claude
committed
feat(jax): enable EP dispatch/combine in single-process multi-thread (SPMT) JAX

jhchouuu
and
claude
committed
fix(jax): clear_ep_handle_cache must touch only the calling thread's slot

jhchouuu
and
claude
committed
refactor(spmt): centralize kMaxGpusPerNode, drop dead code, RAII device guard

jhchouuu
and
claude
committed
shmem: restore Finalized enum value and CheckStatusValid check

jhchouuu
and
claude
committed
examples: drop multithread_multi_gpu.cpp (superseded by Python SPMT tests)

jhchouuu
and
claude
committed
fix(pre-commit): apply black/clang-format/cmake-format auto-fixes

jhchouuu
and
claude
committed
review: ctypes signatures, dedup ep_config parse, spdlog null guard

jhchouuu
and
claude
committed
test(jax-spmt): add real data verification (matches multi-process test)

jhchouuu
and
claude
committed
fix(pre-commit): apply black auto-fix on test_dispatch_combine_jax_spmt

jhchouuu
and
claude
committed

Commits on May 19, 2026

Commits on May 21, 2026

refactor: remove SPMT JAX tests, fix HostName return type

jhchouuu
and
cursoragent
committed