[Mlir-commits] [mlir] [mlir][affine]fix create affine.for bug. (PR #117721)

lonely eagle llvmlistbot at llvm.org
Tue Nov 26 07:27:01 PST 2024


linuxlonelyeagle wrote:

I believe this issue could be made even clearer.Below are the results after I fixed this bug.If you have any questions, welcome to tell me.
```
  gpu.module @gpu {
    gpu.func @gemm(%arg0: memref<128x32xf32>, %arg1: memref<32x64xf32>, %arg2: memref<128x64xf32>) kernel {
      %0 = gpu.dynamic_shared_memory : memref<?xi8, #gpu.address_space<workgroup>>
       .....
      %c1 = arith.constant 1 : index
      %dim = memref.dim %arg0, %c1 : memref<128x32xf32>
      %c0_3 = arith.constant 0 : index
      affine.for %arg3 = %c0_3 to %dim step 32 {
      }
      gpu.return
    }
  }
```

https://github.com/llvm/llvm-project/pull/117721


More information about the Mlir-commits mailing list