Closed edknv closed 4 months ago
24.03
branch-24.03
TensoryMemory does not seem to work with certain data types, e.g. uint8.
>>> import cupy as cp >>> import numpy as np >>> from morpheus._lib.messages import TensorMemory as CppTensorMemory >>> array = np.array([[1, 2], [3, 4]], dtype=np.uint8) >>> array = cp.asarray(array) >>> CppTensorMemory(count=array.shape[0], tensors={"array": array}) Traceback (most recent call last): File "<stdin>", line 1, in <module> ValueError: stoi
The same code does work if dtype is switched to int16.
int16
>>> array = np.array([[1, 2], [3, 4]], dtype=np.int16) >>> array = cp.asarray(array) >>> CppTensorMemory(count=array.shape[0], tensors={"array": array}) <morpheus._lib.messages.TensorMemory object at 0x7c309476aaf0>
[Paste the error here, it will be hidden by default]
[Paste the results of print_env.sh here, it will be hidden by default]
No response
Version
24.03
Which installation method(s) does this occur on?
branch-24.03
Describe the bug.
TensoryMemory does not seem to work with certain data types, e.g. uint8.
Minimum reproducible example
The same code does work if dtype is switched to
int16
.Relevant log output
Click here to see error details
Full env printout
Click here to see environment details
Other/Misc.
No response
Code of Conduct