Disallow unranked complex tensors (#2120)

This was missed in #2045.
openxla · Mar 22, 2024 · 7c8815a · 7c8815a
1 parent 2c82fb4
commit 7c8815a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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]>;