Updated tests based on NCCL 2.27.3-1 sync (#1892)
[ROCm/rccl commit: 9839d1c7c8]
Dieser Commit ist enthalten in:
@@ -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
|
||||
|
||||
@@ -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);
|
||||
|
||||
|
||||
In neuem Issue referenzieren
Einen Benutzer sperren