关于FP16加速的问题

[Copilot-X]

main_bert.py里的一行代码： apex.amp.register_half_function(torch, "einsum") 我看到网上说的:函数（如einsum）暂不支持FP16加速， 这是怎么肥事？

url: https://www.cnblogs.com/sddai/p/14597434.html