i32 to i64 (apache#34)

jinhongyii · Nov 28, 2022 · 5034ae1 · 5034ae1
1 parent 87d3e7e
commit 5034ae1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/relax/ir/block_builder.cc b/src/relax/ir/block_builder.cc
@@ -216,7 +216,7 @@ class BlockBuilderNode::ExprNormalizer : public ExprFunctor<Expr(const Expr&)> {
     if (!constant->shape_) {
       Array<PrimExpr> values;
       for (size_t dim = 0; dim < shape_tuple.size(); dim++) {
-        values.push_back(IntImm(DataType::Int(32), shape_tuple[dim]));
+        values.push_back(IntImm(DataType::Int(64), shape_tuple[dim]));
       }
       UpdateShape(constant, relax::ShapeExpr(values));
     }