diff --git a/triton_viz/clients/sanitizer/sanitizer.py b/triton_viz/clients/sanitizer/sanitizer.py index 06f47574..d6c46305 100644 --- a/triton_viz/clients/sanitizer/sanitizer.py +++ b/triton_viz/clients/sanitizer/sanitizer.py @@ -697,6 +697,8 @@ class SymbolicExpr: "maximum": "max", "minimum": "min", "bitwise_and": "&", + "bitwise_or": "|", + "bitwise_xor": "^", "right_shift": ">>", "left_shift": "<<", "ashr": ">>>", @@ -767,6 +769,8 @@ class SymbolicExpr: "maximum", "minimum", "bitwise_and", + "bitwise_or", + "bitwise_xor", "right_shift", "left_shift", "ashr", @@ -1762,6 +1766,8 @@ def op_binary_op_overrider(lhs, rhs, op): np.maximum: lambda lhs, rhs: SymbolicExpr("maximum", lhs, rhs), np.minimum: lambda lhs, rhs: SymbolicExpr("minimum", lhs, rhs), np.bitwise_and: lambda lhs, rhs: SymbolicExpr("bitwise_and", lhs, rhs), + np.bitwise_or: lambda lhs, rhs: SymbolicExpr("bitwise_or", lhs, rhs), + np.bitwise_xor: lambda lhs, rhs: SymbolicExpr("bitwise_xor", lhs, rhs), np.right_shift: lambda lhs, rhs: SymbolicExpr("right_shift", lhs, rhs), np.left_shift: lambda lhs, rhs: SymbolicExpr("left_shift", lhs, rhs), }