From dc3a3f1e7f9678cc10bf7766078de91c2a7fe400 Mon Sep 17 00:00:00 2001
From: Hui Zhang <zhtclz@foxmail.com>
Date: Tue, 27 Sep 2022 14:52:56 +0800
Subject: [PATCH] tirl bool for jit (#46513)

---
 python/paddle/tensor/creation.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/python/paddle/tensor/creation.py b/python/paddle/tensor/creation.py
index ae785f1d0fe..540c03e36ba 100644
--- 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)
-- 
GitLab