================ @@ -1140,6 +1140,25 @@ static LogicalResult verifyOutputShape( return success(); } +// Find a pure reduction dims, i.e. a reduction dim that is only ---------------- banach-space wrote: Great point, thank you! https://github.com/llvm/llvm-project/pull/237709