diff --git a/test/EnqueueTests.cpp b/test/EnqueueTests.cpp index 03e81f5791..ad357e5b40 100644 --- a/test/EnqueueTests.cpp +++ b/test/EnqueueTests.cpp @@ -59,8 +59,8 @@ protected: comm->workFifoBytes = 1024; // Power of 2 comm->workFifoBuf = nullptr; comm->workFifoBufDev = nullptr; - comm->workFifoConsumed = nullptr; - comm->workFifoConsumedLeast = 0; + comm->workFifoConsumed = 0; + comm->workFifoProducedLastRecorded = 0; comm->workFifoProduced = 0; // Initialize planner diff --git a/test/P2pTests.cpp b/test/P2pTests.cpp index 8b5b22e7f2..5d31315f86 100644 --- a/test/P2pTests.cpp +++ b/test/P2pTests.cpp @@ -1050,7 +1050,7 @@ TEST_F(P2pTests, P2pCanConnectWithIntermediateRankForkTest) { } TEST_F(P2pTests, IpcRegisterBufferFailures) { - + const size_t TEST_BUFFER_ELEMENTS = 32; struct ncclComm comm; struct ncclTopoSystem system; setupCommAndPeers(&comm, &system, gethostid(), 0, 0, 0, true); @@ -1061,13 +1061,13 @@ TEST_F(P2pTests, IpcRegisterBufferFailures) { void *dptr = nullptr; ASSERT_EQ(hipSetDevice(0), hipSuccess); - hipError_t err = hipMalloc(&dptr, 32 * sizeof(float)); + hipError_t err = hipMalloc(&dptr, TEST_BUFFER_ELEMENTS * sizeof(float)); ASSERT_EQ(err, hipSuccess); struct ncclReg regRecord; memset(®Record, 0, sizeof(regRecord)); - regRecord.addr = (uintptr_t)dptr; - regRecord.pages = 1; + regRecord.begAddr = (uintptr_t)dptr; + regRecord.endAddr = regRecord.begAddr + (TEST_BUFFER_ELEMENTS * sizeof(float)); // Calculate end address based on size for (int i = 0; i < 2; ++i) regRecord.ipcInfos[i] = nullptr; @@ -1080,7 +1080,7 @@ TEST_F(P2pTests, IpcRegisterBufferFailures) { // Test 1: HIP_POINTER_ATTRIBUTE_IS_LEGACY_HIP_IPC_CAPABLE is not supported ncclResult_t result = ipcRegisterBuffer( - &comm, dptr, 32 * sizeof(float), peerRanks, 1, type, ®Record, + &comm, dptr, TEST_BUFFER_ELEMENTS * sizeof(float), peerRanks, 1, type, ®Record, ®BufFlag, &offsetOut, &peerRmtAddrsOut, &isLegacyIpc); EXPECT_EQ(result, ncclUnhandledCudaError);