我应该如何设置 CUDA_BATCH_MEM_OP_NODE_PARAMS 的标志字段？

Question

CUDA 图 API 公开了一个 function 调用，用于将“批处理 memory 操作”节点添加到图中：

CUresult cuGraphAddBatchMemOpNode (
   CUgraphNode* phGraphNode, 
   CUgraph hGraph, 
   const CUgraphNode* dependencies, 
   size_t numDependencies, 
   const CUDA_BATCH_MEM_OP_NODE_PARAMS* nodeParams 
);

但是这个 API 调用的文档没有解释...的标志字段的用途，以及应该将标志设置为什么。 那么我应该传递什么价值呢？

Answer 1

一个相关的 API function 是cuStreamBatchMemOp

CUresult cuStreamBatchMemOp (
    CUstream stream, 
    unsigned int count,
    CUstreamBatchMemOpParams* paramArray, 
    unsigned int flags 
);

它本质CUDA_BATCH_MEM_OP_NODE_PARAMS的字段作为其单独的参数。 它的文档说flags “为将来的扩展保留；必须为 0”。