composable_kernel/client_example/06_softmax/softmax4d.cpp at 7613c1d9b9dc612a5de79ab968c534ea58e7cbe4

mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-12 09:16:52 +00:00

Files

Qianfeng 52abc2f371 Use double for all scaling values and float-point constant values at the Device Op API (#557 )

* Use double as alpha/beta values type in reduce device op api

* Use double as alpha/beta values type in softmax device op api

* Use double as alpha/beta values type in multiple-reduce device op api

* Use double as epsilon value type in normalization/elementwise-normalization device op api

2023-01-18 12:02:50 -06:00

5.3 KiB

Raw Blame History

View Raw

5.3 KiB Raw Blame History

5.3 KiB

Raw Blame History