Removed import of cudamat (not used)