[Mlir-commits] [mlir] [mlir][spirv] Add tests for `scf.while` and `scf.for` in `convert-to-spirv` pass (PR #102528)

Angel Zhang llvmlistbot at llvm.org
Mon Aug 12 07:26:46 PDT 2024


================
@@ -33,6 +33,17 @@ func.func @if_yield(%arg0: i1) -> f32 {
 }
 
 // CHECK-LABEL: @while
+// CHECK:       spirv.mlir.loop {
+// CHECK:         spirv.Branch ^[[HEADER:.*]](%{{.*}} : i32)
+// CHECK:       ^[[HEADER]]
+// CHECK:         spirv.Store "Function"
----------------
angelz913 wrote:

I think that when we lower a `scf.while` with `scf.yield`, the header block contains `spirv.Store "Function"`, so I included this to ensure that the `while` is lowered properly.

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


More information about the Mlir-commits mailing list