when I use GPU api, the run
throw error:
code:
const int device_type = kDLGPU;
...
TVMArrayAlloc(in_shape, in_ndim, dtype_code, dtype_bits, dtype_lanes, device_type, device_id, &input);//
TVMArrayCopyFromBytes(input, data, WIDTH*3*HEIGHT*4);
...
set_input("data", input);
run();
error:
'dmlc::Error'
what(): [14:14:29] incubator-tvm/src/runtime/library_module.cc:89: Check failed: ret == 0 (-1 vs. 0) : Assert fail: (255 == int32(arg1.shape[1])), Argument arg1.shape[1] has an unsatisfied constraint
Stack trace:
[bt] (0) TVM/install/lib/libtvm.so(+0x139ab44) [0x7f55158dbb44]
[bt] (1) TVM/install/lib/libtvm.so(+0x14036b7) [0x7f55159446b7]
[bt] (2) TVM/install/lib/libtvm.so(tvm::runtime::GraphRuntime::Run()+0x37) [0x7f5515944737]