Skip to content

GitLab

"vllm/vscode:/vscode.git/clone" did not exist on "5e11b403657ebd5507e07200c2ba2b8f186d07da"
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
builder.py 9.04 KB