search

GreenWaves-Technologies / bfloat16

bfloat16 dtype for numpy
Apache License 2.0
17 stars 13 forks source link

memoryview fail: ValueError: cannot include dtype 'E' in a buffer #9

Open high-cloud opened 5 months ago

high-cloud commented 5 months ago

When I want to get memoryview of numpy array of bfloat16, I got this:

>>> from bfloat16 import bfloat16
>>> import numpy as np
>>> a = np.full((2,2), 1, bfloat16)
>>> a
array([[1.000000, 1.000000],
       [1.000000, 1.000000]], dtype=bfloat16)
>>> m = memoryview(a)
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
ValueError: cannot include dtype 'E' in a buffer