microsoft/mscclpp
1
0
Fork 0
mirror of https://github.com/microsoft/mscclpp.git synced 2026-05-21 21:39:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add missing env fields to Python binding

Browse Source
This commit is contained in:
Ubuntu
2026-03-09 23:15:40 +00:00
parent a76dbe8587
commit 8c7298a357
1 changed files with 6 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
python/csrc/env_py.cpp
View File

@@ -23,7 +23,12 @@ void register_env(nb::module_& m) {
.def_ro("ibv_mode", &Env::ibvMode)
.def_ro("cache_dir", &Env::cacheDir)
.def_ro("npkit_dump_dir", &Env::npkitDumpDir)
.def_ro("cuda_ipc_use_default_stream", &Env::cudaIpcUseDefaultStream);
.def_ro("cuda_ipc_use_default_stream", &Env::cudaIpcUseDefaultStream)
.def_ro("nccl_shared_lib_path", &Env::ncclSharedLibPath)
.def_ro("force_nccl_fallback_operation", &Env::forceNcclFallbackOperation)
.def_ro("nccl_symmetric_memory", &Env::ncclSymmetricMemory)
.def_ro("force_disable_nvls", &Env::forceDisableNvls)
.def_ro("ib_gid_index", &Env::ibGidIndex);
m.def("env", &env);
}