Does the Rust wrapper work for cuda?

Troubleshooting
#1

I tried the example tvm/rust/tvm/examples/resnet at main · apache/tvm · GitHub and changed the cpu to cuda, but got segmentation fault. Does the Rust wrapper work for cuda?

#2

I think its necessary to do a two step conversion, e.g.:

    let input = NDArray::from_rust_ndarray(&arr, Device::cpu(0), DataType::float(32, 1))?;
    let input = input.copy_to_device(&Device::cuda(0))?;