Commit e6b4f527 authored by Seth Howell's avatar Seth Howell
Browse files

buffer.py: Do not force the NIC handler to GPU.



This enables the CPU-Assisted data path.
Signed-off-by: default avatarSeth Howell <sethh@nvidia.com>
parent 35e1cd1b
...@@ -83,7 +83,6 @@ class Buffer: ...@@ -83,7 +83,6 @@ class Buffer:
assert num_qps_per_rank > 0 assert num_qps_per_rank > 0
os.environ['NVSHMEM_DISABLE_P2P'] = '0' if allow_nvlink_for_low_latency_mode else '1' os.environ['NVSHMEM_DISABLE_P2P'] = '0' if allow_nvlink_for_low_latency_mode else '1'
os.environ['NVSHMEM_IB_ENABLE_IBGDA'] = '1' os.environ['NVSHMEM_IB_ENABLE_IBGDA'] = '1'
os.environ['NVSHMEM_IBGDA_NIC_HANDLER'] = 'gpu'
os.environ['NVSHMEM_IBGDA_NUM_RC_PER_PE'] = f'{num_qps_per_rank}' os.environ['NVSHMEM_IBGDA_NUM_RC_PER_PE'] = f'{num_qps_per_rank}'
# Make sure QP depth is always larger than the number of on-flight WRs, so that we can skip WQ slot check # Make sure QP depth is always larger than the number of on-flight WRs, so that we can skip WQ slot check
os.environ['NVSHMEM_QP_DEPTH'] = '1024' os.environ['NVSHMEM_QP_DEPTH'] = '1024'
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment