[CINN] Add type casting for iter_vals (PaddlePaddle#72204)

Author: ggggxm

paddle/cinn/ast_gen_ius/ast_gen.cc

@@ -310,6 +310,7 @@ StmtRef AstGen::Build(const ir::Tensor& tensor, TensorGroup* tensor_group) {
       block_vars.push_back(Var(
           Expr(0), shape[i], cinn::UniqName("i" + std::to_string(i)), false));
       optim::ReplaceVarWithExpr(body, axis[i], block_vars[i]);
+      if (shape[i].type() == Int(64)) axis_vars[i]->set_type(Int(64));
       axis_vars[i]->is_reduce_axis = false;
       iter_values.push_back(axis_vars[i]);
     }

paddle/cinn/hlir/pe/elementwise.cc

@@ -338,7 +338,6 @@ ir::Tensor Tril(const ir::Tensor& A,
                               "The Tril op input tensor must have a rank "
                               "greater than or equal to 2."));
         std::vector<Expr> new_indice(indice.end() - 2, indice.end());
-        Expr col_indice = indice.back();
         return ir::Select::Make(new_indice[0] >= new_indice[1] - diagonal,
                                 A(indice),
                                 ir::Zero(A->type()));