diff --git a/python/extras/util.py b/python/extras/util.py index 3a292a0841..0b863fcda8 100644 --- a/python/extras/util.py +++ b/python/extras/util.py @@ -48,7 +48,8 @@ def __init__(*args, **kwargs): except: # Numpy doesn't support bfloat16 at this time. Using hacky extension of np.float16 as a placeholder, for now. print( - "Warning! bfloat16 python package not available, so using placeholder bfloat16 type" + "Warning! bfloat16 python package not available, so using placeholder bfloat16 type", + file=sys.stderr, ) bfloat16 = _pseudo_bfloat16