onnxruntime/onnxruntime/contrib_ops
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
..
cpu MLAS: add fallback implementation for quantized GEMM (#6335) 2021-01-13 10:53:47 -08:00
cuda Minor changes to AMD element-wise kernels to converge with CUDA element-wise kernels. 2020-12-15 08:46:36 -08:00
rocm Enable add + softmax fusion for Rocm platform (#6259) 2021-01-13 14:09:09 -08:00