np.load gets wrong descriptor when using bfloat16 as the type

GreenWaves-Technologies / bfloat16

bfloat16 dtype for numpy

Apache License 2.0

17 stars 13 forks source link

np.load gets wrong descriptor when using bfloat16 as the type #6

Open TaikunZhang opened 1 year ago

TaikunZhang commented 1 year ago

ValueError: descr is not a valid dtype descriptor: '<g2'

This happens after saving a numpy array to a file using np.save and loading it again using np.load. It seems like the descriptor type is not registered