From 7c8815a247cba7a69af472fa5fef83d046a82a20 Mon Sep 17 00:00:00 2001 From: mlevesquedion Date: Fri, 22 Mar 2024 14:26:47 -0700 Subject: [PATCH] Disallow unranked complex tensors (#2120) This was missed in #2045. --- stablehlo/dialect/Base.td | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/stablehlo/dialect/Base.td b/stablehlo/dialect/Base.td index 38bb5e4c66..f811a3b6f7 100644 --- a/stablehlo/dialect/Base.td +++ b/stablehlo/dialect/Base.td @@ -162,7 +162,7 @@ def HLO_NonQuantizedTensor : RankedTensorOf<[HLO_Float, HLO_Pred, HLO_Int, HLO_C def HLO_TensorOrPerAxisQuantizedTensor : RankedTensorOf<[HLO_Float, HLO_Pred, HLO_Int, HLO_Complex, HLO_QuantizedInt, HLO_PerAxisQuantizedInt]>; -def HLO_ComplexTensor : TensorOf<[HLO_Complex]>; +def HLO_ComplexTensor : RankedTensorOf<[HLO_Complex]>; def HLO_Tuple : NestedTupleOf<[HLO_Tensor, HLO_PerAxisQuantizedIntTensor, HLO_Token]>;