diff --git a/src/relax/ir/block_builder.cc b/src/relax/ir/block_builder.cc index 8e12bbf42e34..d6b80fd8d91e 100644 --- a/src/relax/ir/block_builder.cc +++ b/src/relax/ir/block_builder.cc @@ -212,7 +212,7 @@ class BlockBuilderNode::ExprNormalizer : public ExprFunctor { if (!constant->shape_) { Array values; for (size_t dim = 0; dim < shape_tuple.size(); dim++) { - values.push_back(IntImm(DataType::Int(64), shape_tuple[dim])); + values.push_back(IntImm(DataType::Int(32), shape_tuple[dim])); } UpdateShape(constant, relax::ShapeExpr(values)); }