Re-route thrust memory allocation to THCudaMalloc / THCudaFree in cunn.
6 files changed