onnxruntime
Enable add + softmax fusion for Rocm platform
#6259
Merged

Enable add + softmax fusion for Rocm platform #6259

suffiank merged 11 commits into master from sukha/biassoftmax
suffiank
add bias softmax; tests appear to pass
aac6214c
check fusion occurs for rocm as well
ed735b14
suffiank suffiank requested a review 5 years ago
check for rocm provider compatible as well
f55d2de2
suffiank suffiank requested a review from weixingzhang weixingzhang 5 years ago
build for cpu scenario as well
f8c574ff
try again; broader cope
1a893d94
proper scope on kGpuExecutionProvider
3f1ada2f
been editing wrong file
4b64b5f2
weixingzhang
weixingzhang commented on 2021-01-06
weixingzhang
weixingzhang dismissed these changes on 2021-01-06
remove commented #include lines
60f3facb
suffiank suffiank dismissed their stale review via 60f3facb 5 years ago
weixingzhang
weixingzhang dismissed these changes on 2021-01-11
try again due to mac os ci error
fd658248
suffiank suffiank dismissed their stale review via fd658248 5 years ago
try again
1b767669
test fusion both cuda and rocm to avoid mac ci error
68b47a3c
weixingzhang
weixingzhang approved these changes on 2021-01-13
suffiank suffiank merged 62e40459 into master 5 years ago
suffiank suffiank deleted the sukha/biassoftmax branch 5 years ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone