MHLO Pretty Print - Enhance type printing for CopyOp, ClampOp, CstrRe…

…shapeOp, ComputeReshapeShapeOp, SelectOp. Based on: openxla/stablehlo#37 PiperOrigin-RevId: 466802406
jax-ml · Oct 27, 2022 · 1900a60 · 1900a60
1 parent 540835f
commit 1900a60
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/tests/filecheck/array.filecheck.py b/tests/filecheck/array.filecheck.py
@@ -77,9 +77,7 @@ def main(_):
 
   # CHECK-LABEL: TEST: select bool[2,7] int32[2,7] int32[2,7]
   # CHECK: mhlo.select
-  # CHECK-SAME: tensor<2x7xi1>
-  # CHECK-SAME: tensor<2x7xi32>
-  # CHECK-SAME: tensor<2x7xi32>
+  # CHECK-SAME: tensor<2x7xi1>, tensor<2x7xi32>
   print_ir(np.empty([2, 7], np.bool_), np.empty([2, 7], np.int32),
            np.empty([2, 7], np.int32))(lax.select)