github/concrete
1
1
Fork 0
mirror of https://github.com/zama-ai/concrete.git synced 2026-04-17 03:00:54 -04:00
Code Issues Packages Projects Releases Wiki Activity

feat: add neg support to BaseTracer

Browse Source 
closes #853
This commit is contained in:
Arthur Meyre
2021-11-09 16:07:24 +01:00
parent 4237b9df32
commit 565756810d
2 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
concrete/common/tracing/base_tracer.py
View File

@@ -122,6 +122,9 @@ class BaseTracer(ABC):
# some changes
__radd__ = __add__
def __neg__(self) -> "BaseTracer":
return 0 - self
def __sub__(self, other: Union["BaseTracer", Any]) -> "BaseTracer":
if not self._supports_other_operand(other):
return NotImplemented

1
tests/numpy/test_compile.py
View File

@@ -513,6 +513,7 @@ def test_compile_function_multiple_outputs(
pytest.param(lambda x, y: x + y + 64, ((0, 20), (0, 20)), ["x", "y"]),
pytest.param(lambda x, y: 100 - y + x, ((0, 20), (0, 20)), ["x", "y"]),
pytest.param(lambda x, y: 50 - y * 2 + x, ((0, 20), (0, 20)), ["x", "y"]),
pytest.param(lambda x: -x + 50, ((0, 20),), ["x"]),
],
)
def test_compile_and_run_correctness(