onnxruntime
Implement block wise softmax for reduction dimention > 1024 cases.
#9696

Merged

Implement block wise softmax for reduction dimention > 1024 cases. #9696

pengwa merged 9 commits into master from pengwa/softmax_large_dim

implement block wise softmax for reduction dimention > 1024 cases.

2b15bf3c

pengwa requested a review from

SherlockNoMad 4 years ago

pengwa requested a review from

mindest 4 years ago

pengwa added training

fix builds

1ab19c1c

fix

0dd84412

fix amd build

6c1df7c8

fix amd build

b4a2ce72

fix win-gpu build

51465e12

pengwa requested a review from

weixingzhang 4 years ago

add tests

4ef76523

Merge branch 'master' of https://github.com/microsoft/onnxruntime int…

1dd9c2e6

SherlockNoMad commented on 2021-11-11

SherlockNoMad dismissed these changes on 2021-11-11

remove cudnn path/add python tests

8017dd87

pengwa dismissed their stale review via 8017dd87 4 years ago

pengwa requested a review from

baijumeswani 4 years ago

pengwa requested a review from

liqunfu 4 years ago

pengwa requested a review from

thiagocrepaldi 4 years ago

pengwa requested a review from

tlh20 4 years ago

pengwa requested a review from

xadupre 4 years ago

SherlockNoMad approved these changes on 2021-11-12

pengwa merged 6e09fc51 into master 4 years ago

pengwa deleted the pengwa/softmax_large_dim branch 4 years ago

Reviewers

SherlockNoMad

mindest

weixingzhang

baijumeswani

liqunfu

thiagocrepaldi

tlh20

xadupre

Assignees

No one assigned

Labels

training

Milestone

No milestone

onnxruntime Implement block wise softmax for reduction dimention > 1024 cases. #9696 Merged

Implement block wise softmax for reduction dimention > 1024 cases. #9696

onnxruntime
Implement block wise softmax for reduction dimention > 1024 cases.
#9696

Merged