PR #1949 add bf16 mixed precision support for NPU

add bf16 mixed precision support for NPU #1949

muellerzr merged 2 commits into huggingface:main from bf16

add bf16 mixed precision support for NPU

ca10be7a

ji-huazhong marked this pull request as draft 2 years ago

ji-huazhong changed the title ~~[wip]add bf16 mixed precision support for NPU~~ add bf16 mixed precision support for NPU 2 years ago

ji-huazhong marked this pull request as ready for review 2 years ago

muellerzr approved these changes on 2023-09-11

muellerzr requested a review from

BenjaminBossan 2 years ago

BenjaminBossan approved these changes on 2023-09-11

Explicitly register the NPU backend to PyTorch via `import torch_npu`

dc0c998e

muellerzr merged 937e08ce into main 2 years ago

ji-huazhong deleted the bf16 branch 2 years ago

Reviewers

BenjaminBossan

muellerzr

Assignees

No one assigned

Labels

None yet

Milestone

No milestone