7.8. cudaConditionalNodeParams 结构体参考
[CUDA 运行时使用的数据类型]
CUDA 条件节点参数
公共变量
- cudaGraphConditionalHandle handle
- cudaGraph_t* * phGraph_out
- unsigned int size
- enum cudaGraphConditionalNodeType type
变量
- cudaGraphConditionalHandlecudaConditionalNodeParams::handle [继承]
-
条件节点句柄。句柄必须在使用 cudaGraphConditionalHandleCreate 创建节点之前预先创建。
- cudaGraph_t* * cudaConditionalNodeParams::phGraph_out [继承]
-
CUDA 拥有的数组,在节点创建期间填充条件节点子图。在条件节点的生命周期内有效。图的内容受以下约束的约束
-
允许的节点类型是内核节点、空节点、子图、内存设置、内存复制和条件节点。这递归地应用于子图和条件体。
-
所有内核,包括任何级别的嵌套条件或子图中的内核,都必须属于同一 CUDA 上下文。
这些图可以使用图节点创建 API 或 cudaStreamBeginCaptureToGraph 进行填充。cudaGraphCondTypeIf:当条件为非零时,执行 phGraph_out[0]。如果size== 2,当条件为零时,将执行 phGraph_out[1]。cudaGraphCondTypeWhile:只要条件为非零,就执行 phGraph_out[0]。cudaGraphCondTypeSwitch:当条件等于 n 时,执行 phGraph_out[n]。如果条件 >=size,则不执行任何主体图。
-
- unsigned int cudaConditionalNodeParams::size [继承]
-
图输出数组的大小。允许的值为 cudaGraphCondTypeWhile 为 1,cudaGraphCondTypeWhile 为 1 或 2,或者 cudaGraphCondTypeSwitch 为任何大于零的值。
- enum cudaGraphConditionalNodeTypecudaConditionalNodeParams::type [继承]
-
条件节点类型。