Fix Compiler warning on signed comparison (#174)

dmlc · Nov 1, 2016 · 933ed02 · 933ed02
1 parent f67e112
commit 933ed02
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/mshadow/extension/reduce_with_axis.h b/mshadow/extension/reduce_with_axis.h
@@ -37,13 +37,13 @@ struct ReduceWithAxisExp:
     bool keepdim = (dimsrc == dimdst);
     CHECK(dimsrc > axis) << "reduce axis out of bound";
     Shape<dimsrc> src_shape = ShapeCheck<dimsrc, SrcExp>::Check(src_);
-    for (index_t i = 0; i < axis; ++i) {
+    for (int i = 0; i < axis; ++i) {
       this->shape_[i] = src_shape[i];
     }
     this->size_ = src_shape[axis];
     this->trailing_ = 1;
     if (!keepdim) {
-      for (index_t i = axis + 1; i < dimsrc; ++i) {
+      for (int i = axis + 1; i < dimsrc; ++i) {
         this->trailing_ *= src_shape[i];
         this->shape_[i - 1] = src_shape[i];
       }

diff --git a/mshadow/tensor_cpu-inl.h b/mshadow/tensor_cpu-inl.h
@@ -457,7 +457,7 @@ inline void BatchGEMM(Tensor<Device, 3, DType> dst,
                       DType alpha,
                       DType beta,
                       Tensor<Device, 1, DType*> workspace) {
-  int batch_size = dst.shape_[0];
+  index_t batch_size = dst.shape_[0];
   expr::BLASEngine<Device, DType>::SetStream(dst.stream_);
   Shape<3> sleft = transpose_left ? Shape3(lhs.shape_[0], lhs.shape_[2], lhs.shape_[1])
     : lhs.shape_;