[mlir] Print bbArgs of linalg.map/reduce/tranpose on the next line.
``` %mapped = linalg.map ins(%arg0 : tensor<64xf32>) outs(%arg1 : tensor<64xf32>) (%in: f32) { %0 = math.absf %in : f32 linalg.yield %0 : f32 } %reduced = linalg.reduce ins(%arg0 : tensor<16x32x64xf32>) outs(%arg1 : tensor<16x64xf32>) dimensions = [1] (%in: f32, %init: f32) { %0 = arith.addf %in, %init : f32 linalg.yield %0 : f32 } %transposed = linalg.transpose ins(%arg0 : tensor<16x32x64xf32>) outs(%arg1 : tensor<32x64x16xf32>) permutation = [1, 2, 0] ``` Differential Revision: https://reviews.llvm.org/D136818
Loading
Please sign in to comment