Hey @zhiics and @comaniac,

I am working on an external codegen that will run on GPU. My external codegen 
module is a CSourceModule. The code generated in this module will call some 
CUDA APIs.

If I go through the external codegen workflow and set the target to `cuda 
-libs=cublas,cudnn`, will my C  wrapper get the correct data? In my first test, 
the program crashes when I try to read the data.

I want to run the whole model on GPU and use my custom GPU code for some 
subgraphs.

Thanks!





---
[Visit 
Topic](https://discuss.tvm.ai/t/external-codegen-with-cuda-target/6159/1) to 
respond.

You are receiving this because you enabled mailing list mode.

To unsubscribe from these emails, [click 
here](https://discuss.tvm.ai/email/unsubscribe/025b846457aacfbee4b0f694d42b6951e374ef730b04c88a22597d9b04e4ab09).

Reply via email to