accelerate
937e08ce - add bf16 mixed precision support for NPU (#1949)

Commit
2 years ago
add bf16 mixed precision support for NPU (#1949) * add bf16 mixed precision support for NPU * Explicitly register the NPU backend to PyTorch via `import torch_npu` --------- Co-authored-by: statelesshz <jihuazhong1@huawei.com>
Author
Parents
Loading