[Mlir-commits] [mlir] [mlir][spirv] Add an argmax integration test with `mlir-vulkan-runner` (PR #106426)

[Ivan Butygin]

================
@@ -0,0 +1,108 @@
+// RUN: mlir-vulkan-runner %s \
+// RUN:  --shared-libs=%vulkan-runtime-wrappers,%mlir_runner_utils \
+// RUN:  --entry-point-result=void | FileCheck %s
+
+// This kernel computes the argmax (index of the maximum element) from an array
----------------
Hardcode84 wrote:

This code only works if we have single workgroup, we should probably document it too.

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