transformers
58152438 - MLU devices : Checks if mlu is available via an cndev-based check which won't trigger the drivers and leave mlu (#34326)

Commit
1 year ago
MLU devices : Checks if mlu is available via an cndev-based check which won't trigger the drivers and leave mlu (#34326) * add Cambricon MLUs support * fix mlu device rng state * up for quality check * up mlu to support fp16 * fix mlu device dependency error * fix mlu device dependency error * enable mlu device for bf16 * fix mlu device memory tracker * Cambricon support SDPA and flash_attn * MLU devices : Checks if `mlu` is available via an `cndev-based` check which won't trigger the drivers and leave mlu
Author
Parents
Loading