Skip to content

GitLab

"tests/test_zero/test_sharded_optim_v2.py" did not exist on "3af13a2c3e917ce23d44050cfeea71cfa9f23e81"
Switch branch/tag
Find file BlameHistoryPermalink
  • Hongxin Liu's avatar
    [npu] add npu support for gemini and zero (#5067) · e5ce4c8e
    Hongxin Liu authored 
    * [npu] setup device utils (#5047)

* [npu] add npu device support

* [npu] support low level zero

* [test] update npu zero plugin test

* [hotfix] fix import

* [test] recover tests

* [npu] gemini support npu (#5052)

* [npu] refactor device utils

* [gemini] support npu

* [example] llama2+gemini support npu

* [kernel] add arm cpu adam kernel (#5065)

* [kernel] add arm cpu adam

* [optim] update adam optimizer

* [kernel] arm cpu adam remove bf16 support
    e5ce4c8e
performance_evaluator.py 4.26 KB