Fix TransformLocalIndicesToIters corner case.

[YuhanXu]

PR Category

CINN

PR Types

Bug fixes

Description

修复BUG1：CINN TransformLocalIndicesToIters Pass修改
TransformLocalIndicesToIters支持的普遍场景如下：

  for i:
    for j:
      for k:
        Local_var[IndiceFunc0(i,j)][IndiceFunc1(k)]  # 可被化简为 Local_var[i][j][k]

当IndiceFunc是[Add Sub Mul Div Mod]的组合, Local_var可以被化简为Local_var[i][j][k]

具体的例子：

  for i:
    for j: 
      for k:
         Local_var[i + 1024][j][k]  # 可被化简为 Local_var[i][j][k]

但是存在Corner Case

  for i in [0,16):
    for j:
      for k:
        Local_var[(i + 1024)/2048]  # 错误：被化简为Local_var[i]，正确：被化简为Local_var[0]

Local_var[(i + 1024)/2048]，i取值范围[0,16)，
那么Local_var[(i + 1024)/2048]应该被化简为Local_var[0]，而不是Local_var[i]。
化简成Local_var[i]会生成越界的访问。

修复BUG2：CINN LoopTransformation Pass修改
Slip轴生成新temp_var的时候没有配置新Var的upper_bound和lower_bound

Pcard-90680

[paddle-bot]

你的PR提交成功，感谢你对开源项目的贡献!
请关注后续CI自动化测试结果，详情请参考Paddle-CI手册。
Your PR has been submitted. Thanks for your contribution!
Please wait for the result of CI firstly. See Paddle CI Manual for details.

[CLAassistant]

All committers have signed the CLA.