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

    long k_ = 1;

    // Do GEMM (note: this is a bit confusing because gemm assumes column-major matrices)
    cublasSgemm(
      't', 'n',
      n_, m_, k_,
      1, 

                1,
                THCudaTensor_data(output_n), n
                );
  }

  // Free
