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: warn possible problematic usage of astype

Browse Source
This commit is contained in:
Umut
2022-04-06 17:12:27 +02:00
parent 72c13c54be
commit 38ccacca69
1 changed files with 8 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
concrete/numpy/tracing/tracer.py
View File

@@ -477,10 +477,16 @@ class Tracer:
"""
normalized_dtype = np.dtype(dtype)
output_dtype = Value.of(normalized_dtype.type(0)).dtype
if normalized_dtype != np.int64:
print(
"Warning: when using `value.astype(newtype)`, "
"only use `np.int64` as the newtype "
"to avoid unexpected overflows "
"during inputset evaluation"
)
output_value = deepcopy(self.output)
output_value.dtype = output_dtype
output_value.dtype = Value.of(normalized_dtype.type(0)).dtype
computation = Node.generic(
"astype",