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

Gather size 4

Browse Source
This commit is contained in:
Changho Hwang
2023-09-06 12:30:04 +00:00
parent 8232ec731f
commit e48f942068
2 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/ib.cc
View File

@@ -72,8 +72,8 @@ IbQp::IbQp(ibv_context* ctx, ibv_pd* pd, int port, int maxCqSize, int maxCqPollN
qpInitAttr.qp_type = IBV_QPT_RC;
qpInitAttr.cap.max_send_wr = maxSendWr;
qpInitAttr.cap.max_recv_wr = maxRecvWr;
qpInitAttr.cap.max_send_sge = 1;
qpInitAttr.cap.max_recv_sge = 1;
qpInitAttr.cap.max_send_sge = maxNumSgesPerWr;
qpInitAttr.cap.max_recv_sge = maxNumSgesPerWr;
qpInitAttr.cap.max_inline_data = 0;
struct ibv_qp* _qp = ibv_create_qp(pd, &qpInitAttr);

4
test/mp_unit/ib_tests.cu
View File

@@ -36,7 +36,7 @@ void IbPeerToPeerTest::SetUp() {
bootstrap->initialize(id);
ibCtx = std::make_shared<mscclpp::IbCtx>(ibDevName);
qp = ibCtx->createQp(1024, 1, 8192, 0, 64, 1);
qp = ibCtx->createQp(1024, 1, 8192, 0, 64, 4);
int remoteRank = (gEnv->rank == 0) ? 1 : 0;
@@ -315,7 +315,7 @@ TEST_F(IbPeerToPeerTest, SendGather) {
mscclpp::Timer timeout(3);
const int numDataSrcs = 1;
const int numDataSrcs = 4;
const int nelemPerMr = 1024;
// Gather send from rank 0 to 1