Did fastfold support bfloat16 inference?

Open semal opened this issue 2 years ago • 5 comments

fastfold/utils/geometry/vector.py seems got an unexpected data type.

Mar 24 '23 07:03 semal

Thanks for your question, this feature is still under development.

Mar 25 '23 05:03 Gy-Lu

more information about the data type change.

Mar 27 '23 05:03 semal

This error comes from xx = 1 - 2* (y ** 2 + z** 2), dtype of 1 should give a specific datatype. Another error comes from w *= inv_normwhich is a inplace operation, maybe should change to w = w * inv_norm

Mar 27 '23 06:03 semal

hi! have you achieved inference with bf16? which part of the source code did you modified?

Dec 10 '23 04:12 Ian-pku

Readme says: We can inference a sequence of length 10000 in bf16 with 61GB memory on a Nvidia A100(80GB).

How can I infer with bf16?

Jul 11 '24 12:07 Lix1993