From 291435e0b4927a8efe8a95a989d3a3c244bc0d7a Mon Sep 17 00:00:00 2001
From: Rob Armstrong <roarmstrong@nvidia.com>
Date: Wed, 5 Mar 2025 18:17:27 -0800
Subject: [PATCH] graphConditionalNodes: Additional tweaks to launch dimension
 initialization (#348)

---
 .../graphConditionalNodes/graphConditionalNodes.cu        | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/Samples/3_CUDA_Features/graphConditionalNodes/graphConditionalNodes.cu b/Samples/3_CUDA_Features/graphConditionalNodes/graphConditionalNodes.cu
index ebf46cad..9fb6deb0 100644
--- a/Samples/3_CUDA_Features/graphConditionalNodes/graphConditionalNodes.cu
+++ b/Samples/3_CUDA_Features/graphConditionalNodes/graphConditionalNodes.cu
@@ -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;