From 1900a60a59c2e8f1c7a0c8d3bd32da9434172a49 Mon Sep 17 00:00:00 2001 From: jax authors Date: Wed, 10 Aug 2022 15:38:48 -0700 Subject: [PATCH] MHLO Pretty Print - Enhance type printing for CopyOp, ClampOp, CstrReshapeOp, ComputeReshapeShapeOp, SelectOp. Based on: https://github.com/openxla/stablehlo/pull/37 PiperOrigin-RevId: 466802406 --- tests/filecheck/array.filecheck.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/tests/filecheck/array.filecheck.py b/tests/filecheck/array.filecheck.py index d22faa243b9b..ba2e3392ecbe 100644 --- 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)