Nuphar: Fix a bug in weight layout where read may go out of bound (#2129)

This commit is contained in:
KeDengMS 2019-10-15 00:11:41 -07:00 коммит произвёл GitHub
Родитель 5c2803f2d5
Коммит b101f1bcee
Не найден ключ, соответствующий данной подписи
Идентификатор ключа GPG: 4AEE18F83AFDEB23
1 изменённых файлов: 1 добавлений и 1 удалений

Просмотреть файл

@ -78,7 +78,7 @@ void WeightLayout::CreateLayoutMarshallingTVMOp(tvm::Array<tvm::Tensor>& inputs,
for (size_t dim = 1; dim < input_coord.size(); ++dim)
in_range = in_range && (input_coord[dim] >= 0) && (input_coord[dim] < placeholder->shape[dim]);
return tvm::ir::Select::make(in_range, placeholder(input_coord), pad_zero_expr);
return tvm::if_then_else(in_range, placeholder(input_coord), pad_zero_expr);
} else {
// scalar
return placeholder(input_coord);