一 、构造函数增加
QueryAvailableComputeDevices(&hv_DeviceIdentifier);
OpenComputeDevice(HTuple(hv_DeviceIdentifier[0]), &hv_DeviceHandle);
SetComputeDeviceParam(hv_DeviceHandle, "alloc_pinned", "false");
InitComputeDevice(hv_DeviceHandle, "all");

二、在使用可以GPU算子的所在线程或者函数中 (只对当前线程有效)
开始时增加
ActivateComputeDevice(hv_DeviceHandle);
结束时增加
DeactivateComputeDevice(hv_DeviceHandle);

4714

被折叠的 条评论
为什么被折叠?



