Rate this Page
★ ★ ★ ★ ★

torch.mtia.is_bf16_supported#

torch.mtia.is_bf16_supported(including_emulation=True)[source]#

Return a bool indicating if the current MTIA device supports dtype bfloat16.

On this page
Show Source
PyTorch Libraries