tirl bool for jit (#46513)

PaddlePaddle · Sep 27, 2022 · dc3a3f1 · dc3a3f1
1 parent 8089a1f
commit dc3a3f1
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/python/paddle/tensor/creation.py b/python/paddle/tensor/creation.py
@@ -926,7 +926,8 @@ def _tril_triu_op(helper):
 
     assert x is not None, 'x cannot be None in {}'.format(op_type)
     check_variable_and_dtype(
-        x, 'x', ['float16', 'float32', 'float64', 'int32', 'int64'], op_type)
+        x, 'x', ['float16', 'float32', 'float64', 'int32', 'int64', 'bool'],
+        op_type)
     if len(x.shape) < 2:
         raise ValueError("x shape in {} must be at least 2-D".format(op_type))
     diagonal = helper.kwargs.get('diagonal', 0)