temporarily use i32 for e2e workloads (apache#10)

MasterJH5574 · Nov 19, 2022 · 35cfadd · 35cfadd
1 parent 7425128
commit 35cfadd
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
@@ -212,7 +212,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(64), shape_tuple[dim]));
+        values.push_back(IntImm(DataType::Int(32), shape_tuple[dim]));
       }
       UpdateShape(constant, relax::ShapeExpr(values));
     }