CUDAgraph block/grid size dependent on previous node's result?

Accelerated Computing CUDA CUDA Programming and Performance
1

Hi,
is there a way, in a CUDAgraph, to set the parameters to a kernel (block/grid size, shared memory) using the result of a previous node (computed in the same CUDAgraph, just a previous step) ?
Thanks in advance for your help/

2

cudaGraphKernelNodeSetGridDim allows you to set the grid dimentions of a kernel node from the device as long as you set that node to be updatable & have a reference to its device node handle.

https://docs.nvidia.com/cuda/cuda-runtime-api/group__CUDART__GRAPH.html#group__CUDART__GRAPH_1g646ec1a2a562a5a759b668d10cf0f95d