Added wrapper for pycudann.