NASP/cuda-samples
49
0
Fork 0
mirror of https://github.com/NVIDIA/cuda-samples.git synced 2025-07-30 02:00:30 +08:00
Code Issues Packages Projects Releases Wiki Activity

graphConditionalNodes: Additional tweaks to launch dimension initialization (#348)

Browse Source
This commit is contained in:
Rob Armstrong 2025-03-05 18:17:27 -08:00
parent 8d901e745d
commit 9370f11e69
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
Samples/3_CUDA_Features/graphConditionalNodes/graphConditionalNodes.cu
View File

@ -95,8 +95,8 @@ void simpleIfGraph(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;
@ -496,8 +496,8 @@ void simpleSwitchGraph(void)
// Use a kernel upstream of the conditional to set the handle value
cudaGraphNodeParams params = {cudaGraphNodeTypeKernel};
params.kernel.func = (void *)switchGraphKernelA;
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;