`MoveLoadToAffine` in `affine-cfg` doesn't work for affine maps of indices? about polygeist

[jumerckx]

I was able to reduce the example quite a bit, but I'm still unable to see why this doesn't simply work.

// polygeist-opt --allow-unregistered-dialect --affine-cfg
#map = affine_map<(d0) -> (d0 floordiv 10)>
#map1 = affine_map<(d0) -> (d0 mod 10)>
module {
  func.func @sum(%arg0: memref<10x2xf64>) -> f64 attributes {llvm.emit_c_interface} {
    %cst = arith.constant 0.000000e+00 : f64
    %0 = affine.for %arg1 = 0 to 10 iter_args(%arg2 = %cst) -> (f64) {
      %2 = affine.apply #map(%arg1)
      %3 = affine.apply #map1(%arg1)
      %4 = memref.load %arg0[%3, %2] : memref<10x2xf64>
      %5 = arith.addf %arg2, %4 : f64
      affine.yield %5 : f64
    }
    return %0 : f64
  }
}