[Mlir-commits] [mlir] [MLIR][AMDGPU] Added l2-prefetch op to AMDGPU (PR #188457)

[Ravil Dorozhinskii]

================
@@ -0,0 +1,15 @@
+// RUN: mlir-opt %s --convert-amdgpu-to-rocdl=chipset=gfx1250 --split-input-file --verify-diagnostics | FileCheck %s
+
+// CHECK-LABEL: @glb_prefetch0
+func.func @glb_prefetch0(%src : memref<64x64xf16, 1>, %i : i64, %j : i64) {
+  // CHECK: rocdl.global.prefetch %{{.*}}, scope 0 : !llvm.ptr<1>
+  amdgpu.global_prefetch %src[%i, %j] RT speculative : memref<64x64xf16, 1>
+  func.return
+}
+
+// CHECK-LABEL: @glb_prefetch1
+func.func @glb_prefetch1(%src : memref<64x64xf16, 1>, %i : i64, %j : i64) {
+  // CHECK: rocdl.global.prefetch %{{.*}}, scope 3 : !llvm.ptr<1>
+  amdgpu.global_prefetch %src[%i, %j] HT : memref<64x64xf16, 1>
----------------
ravil-mobile wrote:

Done

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