     long k = weight->size[0];
          
     // Do GEMM (note: this is a bit confusing because gemm assumes column-major matrices)
-    cublasSgemm(
+    THCudaBlas_gemm(
                 'n', 't',
                 n, m, k,
                 1, 

                0,
                THCudaTensor_data(gradColumns), n
                );
    THCublasCheck();
          
    // Unpack columns back into input:
    col2im(

                1,
                THCudaTensor_data(output_n), n
                );
  }

  // Free
