lukasplevac/cuda-samples
1
0
Fork 0
mirror of https://github.com/NVIDIA/cuda-samples.git synced 2025-04-04 07:21:33 +01:00
Code Issues Packages Projects Releases Wiki Activity

Update source code for cudaGraphPerfScaling

Browse Source
This commit is contained in:
shawnz 2025-02-12 12:22:55 +08:00
parent 93f1c78c5b
commit 24a617c043
1 changed files with 10 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
Samples/6_Performance/cudaGraphsPerfScaling/cudaGraphPerfScaling.cu
View File

@ -350,8 +350,10 @@ int main(int argc, char **argv)
cudaFree(0);
cudaMallocHost(&hostData, sizeof(*hostData));
stream.resize(width);
for (int i = 0; i < width; i++)
int numStreams = width;
if (numStreams == 1) numStreams = 2; // demo needs two streams even if capture only needs 1.
stream.resize(numStreams);
for (int i = 0; i < numStreams; i++)
{
cudaStreamCreate(&stream[i]);
}
@ -386,7 +388,7 @@ int main(int argc, char **argv)
if (!(outputFmt & 6)) {
printf("skipping trials since no output is expected\n");
return EXIT_FAILURE;
return 1;
}
std::vector<double> metricTotal;
@ -429,6 +431,9 @@ int main(int argc, char **argv)
length += stride;
}
printf("\n");
}
cudaFreeHost(hostData);
printf("\n");
printf("Test passed\n");
return 0;
}