search

GreenWaves-Technologies / bfloat16

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

Crash observed when importing tensorflow and bfloat16 module #2

Open murthy-l opened 2 years ago

murthy-l commented 2 years ago

Bfloat16 module crashes when tried to import along with tensorflow. Tensorflow supports bfloat16, i am aware of it, but in larger project this is affecting the bfloat16 usage Simple steps to reproduce the issue

$ python Python 3.8.10 (default, Nov 26 2021, 20:14:08) [GCC 9.3.0] on linux Type "help", "copyright", "credits" or "license" for more information.

import tensorflow import bfloat16 Segmentation fault (core dumped)

tensorflow version 2.6.2 bfloat16 version 1.1

zhengbuqian commented 7 months ago

In my experiment importing bfloat16 first and then tensorflow did the trick.

But I only tested on a small script, not sure what the root cause is and whether it'll cause issue in a more complex program/project.