Browse Source

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

pull/53/merge
Rob Armstrong 4 months ago
parent
commit
9370f11e69
  1. 8
      Samples/3_CUDA_Features/graphConditionalNodes/graphConditionalNodes.cu

8
Samples/3_CUDA_Features/graphConditionalNodes/graphConditionalNodes.cu

@ -95,8 +95,8 @@ void simpleIfGraph(void) @@ -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) @@ -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;

Loading…
Cancel
Save