TypeError: Got unsupported ScalarType BFloat16 how to solve it ?

[gukejun1]

File "xx/open-omics-alphafold/alphafold_pytorch_jit/utils.py", line 93, in detached
    return x.detach().cpu().numpy()
           ^^^^^^^^^^^^^^^^^^^^^^^^
TypeError: Got unsupported ScalarType BFloat16

When I run with the parameter output AF2_BF16=1, an error is reported. How to run with the BF16 data type?

[intelyoungway]

It requires your CPU to support intrinsics of AMX BF16.
Please run lscpu | grep amx to check if it exists. If not, set AF2_BF16=0.