(a, b)
| 37 | return tirx.And(a, b) |
| 38 | |
| 39 | def _or(a, b): |
| 40 | if isinstance(a, bool): |
| 41 | a = IntImm("bool", a) |
| 42 | if isinstance(b, bool): |
| 43 | b = IntImm("bool", b) |
| 44 | if DataType(a.dtype).lanes > 1 or DataType(b.dtype).lanes > 1: |
| 45 | return a | b |
| 46 | else: |
| 47 | return tirx.Or(a, b) |
| 48 | |
| 49 | def _get_type_str(dtype: str): |
| 50 | if DataType(dtype).lanes == 1: |