7.16. CUDA_CONDITIONAL_NODE_PARAMS 结构体参考

[CUDA 驱动程序使用的数据类型]

条件节点参数

公共变量

CUcontext ctx
CUgraphConditionalHandle handle
CUgraph*phGraph_out
unsigned int  size
CUgraphConditionalNodeType type

变量

CUcontextCUDA_CONDITIONAL_NODE_PARAMS::ctx [继承]

节点运行的上下文。必须与用于创建句柄和所有主体节点的上下文匹配。

CUgraphConditionalHandleCUDA_CONDITIONAL_NODE_PARAMS::handle [继承]

条件节点句柄。句柄必须在使用 cuGraphConditionalHandleCreate 创建节点之前创建。

CUgraph* * CUDA_CONDITIONAL_NODE_PARAMS::phGraph_out [继承]

CUDA 拥有的数组,在创建节点期间填充条件节点子图。在条件节点的生命周期内有效。图的内容受以下约束的约束

  • 允许的节点类型是内核节点、空节点、子图、内存设置、内存复制和条件节点。这递归地应用于子图和条件主体。

  • 所有内核,包括任何级别的嵌套条件或子图中的内核,都必须属于同一 CUDA 上下文。

这些图可以使用图节点创建 API 或 cuStreamBeginCaptureToGraph 填充。

CU_GRAPH_COND_TYPE_IF:当条件为非零时,执行 phGraph_out[0]。如果size== 2,则当条件为零时,将执行 phGraph_out[1]。CU_GRAPH_COND_TYPE_WHILE:只要条件为非零,就执行 phGraph_out[0]。CU_GRAPH_COND_TYPE_SWITCH:当条件等于 n 时,执行 phGraph_out[n]。如果条件 >=size,则不执行任何主体图。

unsigned int CUDA_CONDITIONAL_NODE_PARAMS::size [继承]

图输出数组的大小。允许的值为:CU_GRAPH_COND_TYPE_WHILE 为 1,CU_GRAPH_COND_TYPE_IF 为 1 或 2,或 CU_GRAPH_COND_TYPE_SWITCH 为任何大于零的值。

CUgraphConditionalNodeTypeCUDA_CONDITIONAL_NODE_PARAMS::type [继承]

条件节点类型。