[Mlir-commits] [mlir] [ROCDL] Added `wave.id` to rocdl; add `rsq` to rocdl.math (PR #176028)

[Krzysztof Drewniak]

================
@@ -39,11 +39,14 @@ llvm.func @rocdl_special_regs() -> i32 {
   // CHECK: call range(i64 1, 65) i64 @__ockl_get_local_size(i32 0)
   %17 = rocdl.workgroup.dim.x range <i32, 1, 65> : i64
 
+  // CHECK: call i32 @llvm.amdgcn.wave.id()
+  %18 = rocdl.wave.id : i32
----------------
krzysz00 wrote:

Do we also want to add a range test here?

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