diff --git a/concrete/numpy/np_dtypes_helpers.py b/concrete/numpy/np_dtypes_helpers.py index 0e942ec8f..f37a41aea 100644 --- a/concrete/numpy/np_dtypes_helpers.py +++ b/concrete/numpy/np_dtypes_helpers.py @@ -22,8 +22,12 @@ from ..common.tracing import BaseTracer from ..common.values import BaseValue, TensorValue NUMPY_TO_COMMON_DTYPE_MAPPING: Dict[numpy.dtype, BaseDataType] = { + numpy.dtype(numpy.int8): Integer(8, is_signed=True), + numpy.dtype(numpy.int16): Integer(16, is_signed=True), numpy.dtype(numpy.int32): Integer(32, is_signed=True), numpy.dtype(numpy.int64): Integer(64, is_signed=True), + numpy.dtype(numpy.uint8): Integer(8, is_signed=False), + numpy.dtype(numpy.uint16): Integer(16, is_signed=False), numpy.dtype(numpy.uint32): Integer(32, is_signed=False), numpy.dtype(numpy.uint64): Integer(64, is_signed=False), numpy.dtype(numpy.float32): Float(32), diff --git a/tests/numpy/test_np_dtypes_helpers.py b/tests/numpy/test_np_dtypes_helpers.py index 13f8c9ef6..7f2a9cfab 100644 --- a/tests/numpy/test_np_dtypes_helpers.py +++ b/tests/numpy/test_np_dtypes_helpers.py @@ -16,10 +16,18 @@ from concrete.numpy.np_dtypes_helpers import ( @pytest.mark.parametrize( "numpy_dtype,expected_common_type", [ + pytest.param(numpy.int8, Integer(8, is_signed=True)), + pytest.param("int8", Integer(8, is_signed=True)), + pytest.param(numpy.int16, Integer(16, is_signed=True)), + pytest.param("int16", Integer(16, is_signed=True)), pytest.param(numpy.int32, Integer(32, is_signed=True)), pytest.param("int32", Integer(32, is_signed=True)), pytest.param(numpy.int64, Integer(64, is_signed=True)), pytest.param("int64", Integer(64, is_signed=True)), + pytest.param(numpy.uint8, Integer(8, is_signed=False)), + pytest.param("uint8", Integer(8, is_signed=False)), + pytest.param(numpy.uint16, Integer(16, is_signed=False)), + pytest.param("uint16", Integer(16, is_signed=False)), pytest.param(numpy.uint32, Integer(32, is_signed=False)), pytest.param("uint32", Integer(32, is_signed=False)), pytest.param(numpy.uint64, Integer(64, is_signed=False)),