fix standalone mode (apache#148)

stefanhenneking · Jul 10, 2016 · 46ead0c · 46ead0c
1 parent 0186f06
commit 46ead0c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/mshadow/dot_engine-inl.h b/mshadow/dot_engine-inl.h
@@ -114,7 +114,7 @@ struct BLASEngine<cpu, float> {
   }
   inline static void batched_gemm(Stream<cpu> *stream,
                                   bool transa, bool transb,
-                                  int m, int n, int k, DType alpha,
+                                  int m, int n, int k, float alpha,
                                   const float *A, int lda, const float *B, int ldb,
                                   float beta, float *C, int ldc, int batch_count) {
     for (int i = 0; i < batch_count; ++i) {
@@ -191,7 +191,7 @@ struct BLASEngine<cpu, double> {
   }
   inline static void batched_gemm(Stream<cpu> *stream,
                                   bool transa, bool transb,
-                                  int m, int n, int k, DType alpha,
+                                  int m, int n, int k, double alpha,
                                   const double *A, int lda, const double *B, int ldb,
                                   double beta, double *C, int ldc, int batch_count) {
     for (int i = 0; i < batch_count; ++i) {