Added support for CUDA modules