Commit dcf58e2a authored by lijian6's avatar lijian6
Browse files

fix size err.


Signed-off-by: lijian6's avatarlijian <lijian6@sugon.com>
parent fe8eb468
......@@ -122,9 +122,9 @@ class Buffer:
self.gda_num_qps_per_pe = max(int(os.environ.get('ROCSHMEM_GDA_NUM_QPS_PER_PE_DEFAULT_CTX', str(num_qps_per_rank))), num_qps_per_rank * self.group_size)
os.environ["ROCSHMEM_GDA_NUM_QPS_DEFAULT_CTX"] = str(self.gda_num_qps_per_pe)
if self.num_rdma_bytes > (1 << 30):
rocshmem_num_rdma_bytes = ((self.num_rdma_bytes + (1 << 31) - 1) // (1 << 31)) * (1 << 31)
os.environ["ROCSHMEM_HEAP_SIZE"] = str(rocshmem_num_rdma_bytes)
# if self.num_rdma_bytes > (1 << 30):
# rocshmem_num_rdma_bytes = ((self.num_rdma_bytes + (1 << 31) - 1) // (1 << 31)) * (1 << 31)
# os.environ["ROCSHMEM_HEAP_SIZE"] = str(rocshmem_num_rdma_bytes)
if self.group_size <= 8:
os.environ["ROCSHMEM_BACKEND"] = "ipc"
......
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