GitOrigin-RevId: 69e6f32e7e
tags/v1.11.1
| @@ -36,8 +36,8 @@ __global__ void kgen_offset_base(GenOffsetBaseParam<nidx, idx_ndim> param) { | |||||
| set_async_error_info( | set_async_error_info( | ||||
| param.error_info, param.error_tracker, | param.error_info, param.error_tracker, | ||||
| "invalid advanced indexing: " | "invalid advanced indexing: " | ||||
| "indexer=%d idx=%d shape=%d", | |||||
| i, data_idx, param.data_shape[i]); | |||||
| "input index %d is out of bounds for axis %d with size %d", | |||||
| data_idx, i, param.data_shape[i]); | |||||
| data_idx = 0; | data_idx = 0; | ||||
| } | } | ||||
| // calculate offset from current index | // calculate offset from current index | ||||
| @@ -56,7 +56,9 @@ void do_exec( | |||||
| data_idx += data_shape; | data_idx += data_shape; | ||||
| megdnn_assert( | megdnn_assert( | ||||
| data_idx >= 0 && static_cast<size_t>(data_idx) < data_shape, | data_idx >= 0 && static_cast<size_t>(data_idx) < data_shape, | ||||
| "bad index value for index %zu at output %zu", i, *index_idx); | |||||
| "invalid advanced indexing: " | |||||
| "input index %d is out of bounds for axis %zu with size %zu", | |||||
| data_idx, i, data_shape); | |||||
| offset += data_stride * data_idx; | offset += data_stride * data_idx; | ||||
| } | } | ||||
| for (size_t i = 0; i < nr_nonidx_axes; ++i) { | for (size_t i = 0; i < nr_nonidx_axes; ++i) { | ||||
| @@ -37,8 +37,8 @@ namespace { | |||||
| // cast to uint32 to handle both negative and overflow | // cast to uint32 to handle both negative and overflow | ||||
| set_async_error_info(param.error_info, param.error_tracker, | set_async_error_info(param.error_info, param.error_tracker, | ||||
| "invalid advanced indexing: " | "invalid advanced indexing: " | ||||
| "indexer=%d idx=%d shape=%d", | |||||
| i, data_idx, param.data_shape[i]); | |||||
| "input index %d is out of bounds for axis %d with size %d", | |||||
| data_idx, i, param.data_shape[i]); | |||||
| data_idx = 0; | data_idx = 0; | ||||
| } | } | ||||
| offset += data_idx * param.data_stride[i]; | offset += data_idx * param.data_stride[i]; | ||||