[Apache TVM Discuss] [Questions] Use cuda pointers with TVM

[Luca Fancellu via Apache TVM Discuss]

So can I simply do that?

    void* ptr;
    cudaMalloc(&ptr, 10 * 10 * 2 * sizeof(float));
    TVMArray *x{};
    std::vector<int64_t> shape ={1, 10, 10, 2};
    TVMArrayAlloc(&shape[0], shape.size(), kDLFloat, 32, 1, kDLGPU, 0, &x);
    x->data = ptr;





---
[Visit 
Topic](https://discuss.tvm.apache.org/t/use-cuda-pointers-with-tvm/7853/3) to 
respond.

You are receiving this because you enabled mailing list mode.

To unsubscribe from these emails, [click 
here](https://discuss.tvm.apache.org/email/unsubscribe/20162c98264f52b829ae2c1edf098864cdedadd873c0dc8b9c090c09b1e31d9b).