feat: Wx/modify ops for npu - #134

Open

POI-WX wants to merge 9 commits into

mainfrom

wx/modify_ops_for_npu

Commits on Oct 21, 2024

fix bug of rope for npu and support the interleaved option
POI-WX
committed
49d1a9d

Commits on Oct 22, 2024

optimize when interleaved is True
POI-WX
committed
0eebc4b
use fused rope op from mindspeed
POI-WX
committed
d69d17b

Commits on Oct 23, 2024

optimize
POI-WX
committed
6681c0a

Commits on Jan 8, 2025

using fused rms norm op with 8.0 RC3 software stack on npu
wangxing2
committed
ee0baeb

Commits on Jan 9, 2025

optimize
wangxing2
committed
5565fbc
modify atol and rtol
wangxing2
committed
c39b8c5

Commits on Jan 22, 2025

fix
wangxing2
committed
ee852e4
add check for fa op
wangxing2
committed
60ec7fd