obscura/nccl-tests
1
0
Fork 0
mirror of https://github.com/NVIDIA/nccl-tests.git synced 2026-09-06 02:19:56 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge branch 'v2.29_check-devapi-support' into 'master'

Browse Source 
check deviceAPI support

See merge request nccl/nccl-tests-internal!23
This commit is contained in:
Katie Gioioso
2026-01-08 11:49:16 -08:00
parent cd773dbe2f 86c1297a95
commit ca7625d41a
3 changed files with 12 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/all_reduce.cu
+1 -1
View File
@@ -78,7 +78,7 @@ testResult_t AllReduceGetDevCommRequirements(int deviceImpl, ncclDevCommRequirem
case 4: // allReduceMultimemVectorizedKernel
if (!commProperties->multimemSupport) {
fprintf(stderr, "This test requires multimem support, but multimem support is not enabled for this communicator.\n");
return testInternalError;
return testInvalidUsage;
}
reqs->lsaMultimem = true;
reqs->lsaBarrierCount = deviceCtaCount;
src/alltoall.cu
+1 -1
View File
@@ -65,7 +65,7 @@ testResult_t AlltoAllGetDevCommRequirements(int deviceImpl, ncclDevCommRequireme
case 4: // HybridAlltoAllKernel (LSA+GIN)
if (commProperties->ginType == NCCL_GIN_TYPE_NONE) {
fprintf(stderr, "This test requires GIN support, but GIN support is not enabled for this communicator.\n");
return testInternalError;
return testInvalidUsage;
}
reqs->barrierCount = deviceCtaCount;
reqs->ginSignalCount = deviceCtaCount;
src/common.cu
+10 -2
View File
@@ -834,8 +834,12 @@ testResult_t threadInit(struct threadArgs* args) {
fprintf(stderr, "Device implementation %d is not supported by this test\n", deviceImpl);
return testNotImplemented;
}
ncclCommProperties commProperties = NCCL_COMM_PROPERTIES_INITIALIZER;
ncclCommProperties_t commProperties = NCCL_COMM_PROPERTIES_INITIALIZER;
NCCLCHECK(ncclCommQueryProperties(args->comms[0], &commProperties));
if (!commProperties.deviceApiSupport) {
fprintf(stderr, "Device API is not supported by this communicator.\n");
return testInvalidUsage;
}
TESTCHECK(ncclTestEngine.getDevCommRequirements(deviceImpl, &reqs, &commProperties));
#else
if (test_ncclVersion >= NCCL_VERSION(2,29,0)) {
@@ -1410,8 +1414,12 @@ testResult_t run() {
fprintf(stderr, "Device implementation %d is not supported by this test\n", deviceImpl);
return testNotImplemented;
}
ncclCommProperties commProperties = NCCL_COMM_PROPERTIES_INITIALIZER;
ncclCommProperties_t commProperties = NCCL_COMM_PROPERTIES_INITIALIZER;
NCCLCHECK(ncclCommQueryProperties(comms[0], &commProperties));
if (!commProperties.deviceApiSupport) {
fprintf(stderr, "Device API is not supported by this communicator.\n");
return testInvalidUsage;
}
TESTCHECK(ncclTestEngine.getDevCommRequirements(deviceImpl, &reqs, &commProperties));
#else
if (test_ncclVersion >= NCCL_VERSION(2,29,0)) {