fix compile error

apache · May 26, 2020 · 40a14d9 · 40a14d9
1 parent 0577671
commit 40a14d9
Show file tree

Hide file tree

Showing 5 changed files with 6 additions and 6 deletions.
diff --git a/src/operator/tensor/index_add-inl.h b/src/operator/tensor/index_add-inl.h
@@ -170,7 +170,7 @@ struct IndexAddBackwardAKernel {
                                   const mshadow::Shape<MXNET_SPECIAL_MAX_NDIM>& stride,
                                   const int tail_size, const int ind_num, const int ind_ndim,
                                   const int32_t* ind_vec, const int req, const int out_ndim) {
-    size_t id = 0;
+    index_t id = 0;
     int seg = MXNET_SPECIAL_MAX_NDIM - out_ndim;
     for (int dim = 0; dim < ind_ndim; ++dim) {
       id += stride[seg + dim] * ind_vec[dim * ind_num + i];

diff --git a/src/operator/tensor/index_add_backward.cc b/src/operator/tensor/index_add_backward.cc
@@ -51,7 +51,7 @@ struct IndexAddBackwardValCPUKernel {
                                   const int ograd_tail_size, const int ind_num,
                                   const int ind_ndim, const int32_t* ind_vec,
                                   const int out_ndim) {
-    size_t id = 0;
+    index_t id = 0;
     int seg = MXNET_SPECIAL_MAX_NDIM - out_ndim;
     for (int dim = 0; dim < ind_ndim; ++dim) {
       id += ograd_pre_stride[seg + dim] * ind_vec[dim * ind_num + i];

diff --git a/src/operator/tensor/index_add_backward.cu b/src/operator/tensor/index_add_backward.cu
@@ -55,7 +55,7 @@ struct IndexAddBackwardValGPUKernel {
                                   const int ograd_tail_size, const int ind_num,
                                   const int ind_ndim, const int32_t* ind_vec,
                                   const int out_ndim) {
-    size_t id = 0;
+    index_t id = 0;
     int seg = MXNET_SPECIAL_MAX_NDIM - out_ndim;
     for (int dim = 0; dim < ind_ndim; ++dim) {
       id += ograd_pre_stride[seg + dim] * ind_vec[dim * ind_num + i];

diff --git a/src/operator/tensor/index_add_forward.cc b/src/operator/tensor/index_add_forward.cc
@@ -35,10 +35,10 @@ struct IndexAddForwardCPUKernel {
                                   const mshadow::Shape<MXNET_SPECIAL_MAX_NDIM>& val_stride,
                                   const mshadow::Shape<MXNET_SPECIAL_MAX_NDIM>& val_shape,
                                   const mshadow::Shape<MXNET_SPECIAL_MAX_NDIM>& a_shape,
-                                  const size_t a_tail_size, const int ind_num,
+                                  const int a_tail_size, const int ind_num,
                                   const int ind_ndim, const int* ind,
                                   const int a_ndim) {
-    size_t id = 0;
+    index_t id = 0;
     int seg = MXNET_SPECIAL_MAX_NDIM - a_ndim;
     for (int dim = 0; dim < ind_ndim; ++dim) {
       CHECK_LT(ind[dim * ind_num + i], a_shape[seg + dim])

diff --git a/src/operator/tensor/index_add_forward.cu b/src/operator/tensor/index_add_forward.cu
@@ -41,7 +41,7 @@ struct IndexAddForwardGPUKernel {
                                   const int a_tail_size, const int ind_num,
                                   const int ind_ndim, const int* ind,
                                   const int a_ndim) {
-    size_t id = 0;
+    index_t id = 0;
     int seg = MXNET_SPECIAL_MAX_NDIM - a_ndim;
     for (int dim = 0; dim < ind_ndim; ++dim) {
       id += a_pre_stride[seg + dim] * ind[dim * ind_num + i];