gguf-py : __pos__ is also unary

This commit is contained in:
Francis Couture-Harpin 2025-11-06 21:51:20 -05:00
parent 33dcb44aa2
commit 987862ad8c
1 changed files with 2 additions and 2 deletions

View File

@ -50,7 +50,7 @@ class LazyMeta(ABCMeta):
for binary_op in ( for binary_op in (
"lt", "le", "eq", "ne", "ge", "gt", "lt", "le", "eq", "ne", "ge", "gt",
"add", "and", "floordiv", "lshift", "mod", "mul", "matmul", "add", "and", "floordiv", "lshift", "mod", "mul", "matmul",
"or", "pos", "pow", "rshift", "sub", "truediv", "xor", "or", "pow", "rshift", "sub", "truediv", "xor",
"iadd", "iand", "ifloordiv", "ilshift", "imod", "imul", "ior", "irshift", "isub", "ixor", "iadd", "iand", "ifloordiv", "ilshift", "imod", "imul", "ior", "irshift", "isub", "ixor",
"radd", "rand", "rfloordiv", "rmul", "ror", "rpow", "rsub", "rtruediv", "rxor", "radd", "rand", "rfloordiv", "rmul", "ror", "rpow", "rsub", "rtruediv", "rxor",
): ):
@ -58,7 +58,7 @@ class LazyMeta(ABCMeta):
# evaluation on the meta tensor is needed in case there's broadcasting # evaluation on the meta tensor is needed in case there's broadcasting
namespace[attr_name] = mk_wrap(attr_name, meta_noop=False) namespace[attr_name] = mk_wrap(attr_name, meta_noop=False)
for unary_op in ("not", "abs", "invert", "neg"): for unary_op in ("not", "abs", "invert", "neg", "pos"):
attr_name = f"__{unary_op}__" attr_name = f"__{unary_op}__"
# the result of these operators usually has the same shape and dtype as the input, # the result of these operators usually has the same shape and dtype as the input,
# so evaluation on the meta tensor can be skipped. # so evaluation on the meta tensor can be skipped.