Hi Guys
I would like to copy data from CPU to GPU
After inference the result is different every time, I think the problem is that the data do not copy completely, I also tried TVMSynchronize () … but still not working
Any Ideas ?
nds[node] = tvm::runtime::NDArray::Empty(shape, DLDataType{ kDLFloat , 32, 1 }, device);
nds_cpu[node] = tvm::runtime::NDArray::Empty(shape, DLDataType{ kDLFloat , 32, 1 }, device_cpu);
set_val(nds_cpu[node]);
nds[node].CopyFrom(nds_cpu[node]);