================ ---------------- adam-smnk wrote: nit: I think it'd make sense to add all three check variants to all test cases here Also, maybe one more op variant like `batch_matmul` for completness https://github.com/llvm/llvm-project/pull/211354