update

Author: XieYunshen

python/env_dict.py.in

@@ -116,4 +116,11 @@ env_dict={
     'WITH_FLAGCX':'@WITH_FLAGCX@',
     'FLAGCX_LIB':'@FLAGCX_LIB@',
     'COMPILED_CUDA_ARCHS':'@COMPILED_CUDA_ARCHS@'
+    'WITH_DISTRIBUTE':'@WITH_DISTRIBUTE@',
+    'WITH_NVSHMEM':'@WITH_NVSHMEM@',
+    'NVSHMEM_BOOTSTRAP_UID_LIB':'@NVSHMEM_BOOTSTRAP_UID_LIB@',
+    'NVSHMEM_BOOTSTRAP_PMI_LIB':'@NVSHMEM_BOOTSTRAP_PMI_LIB@',
+    'NVSHMEM_BOOTSTRAP_PMI2_LIB':'@NVSHMEM_BOOTSTRAP_PMI2_LIB@',
+    'NVSHMEM_TRANSPORT_IBRC_LIB':'@NVSHMEM_TRANSPORT_IBRC_LIB@',
+    'NVSHMEM_TRANSPORT_IBGDA_LIB':'@NVSHMEM_TRANSPORT_IBGDA_LIB@'
 }