graphConditionalNodes: Change launch dimension initialization for better cross-platform compatibility (#346)

This commit is contained in:
Rob Armstrong 2025-03-05 08:32:58 -08:00 committed by GitHub
parent e7b23470d5
commit 990ebc01c2
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -380,8 +380,8 @@ void simpleIfElseGraph(void)
// Use a kernel upstream of the conditional to set the handle value
cudaGraphNodeParams params = {cudaGraphNodeTypeKernel};
params.kernel.func = (void *)ifGraphKernelA;
params.kernel.gridDim = 1;
params.kernel.blockDim = 1;
params.kernel.blockDim.x = params.kernel.blockDim.y = params.kernel.blockDim.z = 1;
params.kernel.gridDim.x = params.kernel.gridDim.y = params.kernel.gridDim.z = 1;
params.kernel.kernelParams = kernelArgs;
kernelArgs[0] = &dPtr;
kernelArgs[1] = &handle;