client_example/06_softmax/softmax4d.cpp · 52abc2f37112d49f85f31aa343a14bd92a83b07c · gaoqiong / composable_kernel

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

Qianfeng authored Jan 19, 2023

* 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

52abc2f3

softmax4d.cpp 5.27 KB

Replace softmax4d.cpp