Fix mid execution mlir test by replacing topk with convert

Author: Kepontry

tests/lit/NPU/dialect/VPUIP/passes/add_sw_kernel_instruction_prefetch_mid_execution_40XX.mlir

@@ -9,21 +9,21 @@
 !DummyDDRT = memref<32000x1x1x1xf16, @DDR>
 !DummyCMX0T = memref<32000x1x1x1xf16, [@CMX_NN, 0]>
 !DummyCMX1T = memref<32000x1x1x1xf16, [@CMX_NN, 1]>
-!DummyCMX0TopK = memref<16000x1x1x1xsi32, [@CMX_NN, 0]>
-!DummyCMX1TopK = memref<16000x1x1x1xsi32, [@CMX_NN, 1]>
+!DummyCMX0Convert = memref<32000x1x1x1xf32, [@CMX_NN, 0]>
+!DummyCMX1Convert = memref<32000x1x1x1xf32, [@CMX_NN, 1]>
 
 // This test checks following schedule
-//  Barriers :             0         1         2            3          4         5
-//  Cluster 0:             | [ DMA ] | [ DMA ] | [ Softmax] | [ TopK ] | [ DMA ] | [ Softmax ]
-//  Cluster 1:             | [    DMA    ]     | [ Softmax] | [ TopK ]
+//  Barriers :             0         1         2            3             4         5
+//  Cluster 0:             | [ DMA ] | [ DMA ] | [ Softmax] | [ Convert ] | [ DMA ] | [ Softmax ]
+//  Cluster 1:             | [    DMA    ]     | [ Softmax] | [ Convert ]
 //  Other    : [ SyncDMA ] |
 //
 
 module @subgraph attributes {config.arch = #config.arch_kind<NPU40XX>, config.compilationMode = #config.compilation_mode<DefaultHW>} {
   VPURT.SW.Runtime entryPoint : @VPU.SW::@runtime stack_configuration : [4096, 4096, 4096, 4096, 4096, 4096]
   module @VPU.SW {
     func.func private @builtin_SoftMax(memref<*xf16, @CMX_NN>, memref<*xf16, @CMX_NN>, i64, i64) attributes {VPU.kernel_code = "softmax.cpp", VPU.kernel_entry = "softmax", VPU.task_type = @COMPUTE}
-    func.func private @builtin_TopK(memref<*xf16, @CMX_NN>, memref<*xf16, @CMX_NN>, memref<*xsi32, @CMX_NN>, i64, i64, i64, i64) attributes {VPU.kernel_code = "topk.cpp", VPU.kernel_entry = "topk", VPU.task_type = @COMPUTE}
+    func.func private @builtin_Convert(memref<*xf16, @CMX_NN>, memref<*xf32, @CMX_NN>) attributes {VPU.kernel_code = "convert.cpp", VPU.kernel_entry = "convert", VPU.kernel_name = "convert", VPU.task_type = @COMPUTE}
     func.func private @runtime() attributes {VPU.kernel_code = "nnActEntry"}
   }
   config.Resources {activity_factor = 0.078934384661980161 : f64} 2 of @NCE at 1.700000e+03 MHz {
@@ -105,18 +105,18 @@ module @subgraph attributes {config.arch = #config.arch_kind<NPU40XX>, config.co
     }
     }
 
-    %cmx0_top_k = VPURT.DeclareBuffer <CMX_NN> [0] <0> -> !DummyCMX0TopK
+    %cmx0_convert = VPURT.DeclareBuffer <CMX_NN> [0] <0> -> !DummyCMX0Convert
     VPURT.Task waits(%5: !VPURT.Barrier) updates(%6 : !VPURT.Barrier) {
-        %results:2 = VPUIP.SW.Kernel {resultSegmentSizes = array<i32: 2, 0, 0>} @VPU.SW::@builtin_TopK inputs(%cmx_0 as %arg3: !DummyCMX0T) outputs(%cmx_0 as %arg4: !DummyCMX0T, %cmx0_top_k as %arg5: !DummyCMX0TopK) on tile 0 -> (!DummyCMX0T, !DummyCMX0TopK) {
-                VPUIP.SW.Kernel.run {attrs = [1, 0, 0, 1]}(%arg3, %arg4, %arg5) : !DummyCMX0T, !DummyCMX0T, !DummyCMX0TopK
-    }
+      %results = VPUIP.SW.Kernel {resultSegmentSizes = array<i32: 1, 0, 0>} @VPU.SW::@builtin_Convert inputs(%cmx_0 as %arg3: !DummyCMX0T) outputs(%cmx0_convert as %arg4: !DummyCMX0Convert) on tile 0 -> (!DummyCMX0Convert) {
+        VPUIP.SW.Kernel.run {attrs = [[]]}(%arg3, %arg4) : !DummyCMX0T, !DummyCMX0Convert
+      }
     }
 
-    %cmx1_top_k = VPURT.DeclareBuffer <CMX_NN> [1] <0> -> !DummyCMX1TopK
+    %cmx1_convert = VPURT.DeclareBuffer <CMX_NN> [1] <0> -> !DummyCMX1Convert
     VPURT.Task waits(%5: !VPURT.Barrier) updates(%6 : !VPURT.Barrier) {
-        %results:2 = VPUIP.SW.Kernel {resultSegmentSizes = array<i32: 2, 0, 0>} @VPU.SW::@builtin_TopK inputs(%cmx_1 as %arg3: !DummyCMX1T) outputs(%cmx_1 as %arg4: !DummyCMX1T, %cmx1_top_k as %arg5: !DummyCMX1TopK) on tile 1 -> (!DummyCMX1T, !DummyCMX1TopK) {
-                VPUIP.SW.Kernel.run {attrs = [1, 0, 0, 1]}(%arg3, %arg4, %arg5) : !DummyCMX1T, !DummyCMX1T, !DummyCMX1TopK
-    }
+      %results = VPUIP.SW.Kernel {resultSegmentSizes = array<i32: 1, 0, 0>} @VPU.SW::@builtin_Convert inputs(%cmx_1 as %arg3: !DummyCMX1T) outputs(%cmx1_convert as %arg4: !DummyCMX1Convert) on tile 1 -> (!DummyCMX1Convert) {
+        VPUIP.SW.Kernel.run {attrs = [[]]}(%arg3, %arg4) : !DummyCMX1T, !DummyCMX1Convert
+      }
     }
 
     VPURT.Task waits(%6: !VPURT.Barrier) updates(%7 : !VPURT.Barrier) {
@@ -158,15 +158,15 @@ module @subgraph attributes {config.arch = #config.arch_kind<NPU40XX>, config.co
     // CHECK:       VPURT.Task {
     // CHECK-NEXT:        VPUIP.SW.Kernel
     // CHECK-SAME:        skipProfiling
-    // CHECK-SAME:        @VPU.SW::@builtin_TopK
+    // CHECK-SAME:        @VPU.SW::@builtin_Convert
 
     // CHECK:       VPURT.Task waits([[BARRIER_5]] : !VPURT.Barrier) updates([[BARRIER_6]] : !VPURT.Barrier) {
     // CHECK:             VPUIP.SW.Kernel
-    // CHECK-SAME:        @VPU.SW::@builtin_TopK
+    // CHECK-SAME:        @VPU.SW::@builtin_Convert
 
     // CHECK:       VPURT.Task waits([[BARRIER_5]] : !VPURT.Barrier) updates([[BARRIER_6]] : !VPURT.Barrier) {
     // CHECK:             VPUIP.SW.Kernel
-    // CHECK-SAME:        @VPU.SW::@builtin_TopK
+    // CHECK-SAME:        @VPU.SW::@builtin_Convert
 
     // CHECK:       VPURT.Task waits([[BARRIER_6]] : !VPURT.Barrier) updates([[BARRIER_7]] : !VPURT.Barrier) {
     // CHECK-NEXT:        VPUIP.NNDMA