onnxruntime
[ARM CPU] Add fp16 mlas kernels for exp, tanh, softmax, logsoftmax, softcap
#23597

Merged

[ARM CPU] Add fp16 mlas kernels for exp, tanh, softmax, logsoftmax, softcap #23597

fajin-corp merged 11 commits into main from fajin/gqa-softmax

fajin-corp requested a review 341 days ago

github-actions commented on 2025-02-06

added the apis for fp16 softmax kernels

f27df5be

added softcap api

7d7b5353

finished exp and sum exp kernels

7dd6ceed

finished tanh and softcap

cc22f530

finished reduce max kernel

30eef71d

finished softmax kernels

e5949185

fix build

5f99390c

remove warning

c93c17a6

finished ut

d9919fb3

fixed linting

56b85ed2

fajin-corp force pushed from 3a78168a to 56b85ed2 340 days ago

fix windows build

94e80536

amarin16 approved these changes on 2025-02-10

yihonglyu approved these changes on 2025-02-10

fajin-corp merged e206950f into main 336 days ago

fajin-corp deleted the fajin/gqa-softmax branch 336 days ago

Reviewers

yihonglyu

amarin16

github-actions

Assignees

No one assigned

Labels

None yet

Milestone

No milestone