     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, 

    long k = weight->size[1];

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

                1,
                THCudaTensor_data(gradWeight), n
                );
    
    // Do Bias: 
    // M,N,K are dims of matrix A and B
