PR #13371 [DML EP] Attention Kernel

[DML EP] Attention Kernel #13371

sumitsays merged 14 commits into main from user/sumita/attention

Attention Kernel First Commit

ee99225d

Fixed output stride and transposed final output

aaa9acfc

Used packed strided output for first GEMM

4d92c374

Fixed final output stride to transpose the output

569e1965

Merge branch 'master' into user/sumita/attention

05d39c05

Store the return object into a variable and then use it. Otherwise Re…

b8b526a2

fdwr commented on 2022-10-19

Clean up

3deee222

Removed fusion

dfeac936

Added some comments

eb11002f

sumitsays marked this pull request as ready for review 3 years ago

sumitsays requested a review from

jeffbloo 3 years ago

Store the temporary object into a variable

8f647652

enable fusion

f9bdbe43

fdwr commented on 2022-10-20

Addressed nit comments on the PR

9070f556

fdwr commented on 2022-10-21

Minor intrafile style consistency cleanup

af14e80a

Move division by zero check sooner

9f1b7bd5

fdwr approved these changes on 2022-10-21

sumitsays merged 24818cfd into main 3 years ago

sumitsays deleted the user/sumita/attention branch 3 years ago

Reviewers

fdwr

jeffbloo

Assignees

No one assigned

Labels

None yet

Milestone

No milestone

onnxruntime [DML EP] Attention Kernel #13371 Merged

[DML EP] Attention Kernel #13371

onnxruntime
[DML EP] Attention Kernel
#13371

Merged