CUDA 图形

populate_cuda_graph

populate_cuda_graph 函数是 Graph 类的成员函数。它用于使用必要的数据和操作填充 CUDA 图形。

参数

• handle：一个 cuDNN 句柄。

• uid_to_device_ptrs：张量 UID 到设备指针的映射。

• workspace：指向工作区内存的指针。

• cudnn_cuda_graph：指向 CUDA 图形的指针。

返回值

• 一个 error_t 对象，指示函数成功或失败。

update_cuda_graph

update_cuda_graph 函数是 Graph 类的成员函数。它用于使用必要的数据和操作更新 CUDA 图形。

参数

• handle：一个 cuDNN 句柄。

• uid_to_device_ptrs：张量 UID 到设备指针的映射。

• workspace：指向工作区内存的指针。

• cudnn_cuda_graph：指向 CUDA 图形的指针。

返回值

• 一个 error_t 对象，指示函数成功或失败。