onnxruntime/onnxruntime
Suffian Khan 62e404591a
Enable add + softmax fusion for Rocm platform (#6259)
* add bias softmax; tests appear to pass

* check fusion occurs for rocm as well

* check for rocm provider compatible as well

* build for cpu scenario as well

* try again; broader cope

* proper scope on kGpuExecutionProvider

* been editing wrong file

* remove commented #include lines

* try again due to mac os ci error

* try again

* test fusion both cuda and rocm to avoid mac ci error
2021-01-13 14:09:09 -08:00
..
contrib_ops Enable add + softmax fusion for Rocm platform (#6259) 2021-01-13 14:09:09 -08:00
core Enable add + softmax fusion for Rocm platform (#6259) 2021-01-13 14:09:09 -08:00
featurizers_ops/cpu
gsl
python Delete float16.py (#6336) 2021-01-13 13:41:06 -08:00
test Enable add + softmax fusion for Rocm platform (#6259) 2021-01-13 14:09:09 -08:00
tool/etw
.style.yapf
__init__.py Deprecate Python global configuration functions [Part 2] (#6171) 2021-01-07 10:10:55 -08:00
ReformatSource.ps1
ReformatSourcePython.bat
VSCodeCoverage.runsettings