yes, we can do this. i have tried this and successfully implemented the codeBut I i face one problem, i can only run code on the CPU, while using the cuda compiler shows error regarding batch first command.