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

feat: adding management of numpy.isfinite, numpy.isinf, numpy.isnan

Browse Source 
refs #126
This commit is contained in:
Benoit Chevallier-Mames
2021-09-29 14:24:49 +02:00
committed by Benoit Chevallier
parent ab7cf24285
commit 4adb0eb18e
2 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
concrete/numpy/np_dtypes_helpers.py
View File

@@ -27,6 +27,7 @@ NUMPY_TO_COMMON_DTYPE_MAPPING: Dict[numpy.dtype, BaseDataType] = {
numpy.dtype(numpy.uint64): Integer(64, is_signed=False),
numpy.dtype(numpy.float32): Float(32),
numpy.dtype(numpy.float64): Float(64),
numpy.dtype(bool): Integer(32, is_signed=False),
}
SUPPORTED_NUMPY_DTYPES = tuple(NUMPY_TO_COMMON_DTYPE_MAPPING)

6
concrete/numpy/tracing.py
View File

@@ -230,9 +230,9 @@ class NPTracer(BaseTracer):
# numpy.heaviside,
# numpy.hypot,
# numpy.invert,
# numpy.isfinite,
# numpy.isinf,
# numpy.isnan,
numpy.isfinite,
numpy.isinf,
numpy.isnan,
# numpy.isnat,
# numpy.lcm,
# numpy.ldexp,