Shaojie WANG cd3fbda406 example for convnd bwd weight bf16 splitk (#265) ...

* add GetWorkSpaceSize to base arg and make an example on convnd_bwd_weight

* add bwd weight for bf16: init

* remove redundant compute

* use datatype and split k to check whether a workspace is used

* remove unused computation for work space size

* add some code for bfp16

* add device/grid unary op

* add unary type convert to bwd-weight example

* support bf16 splitk kernel for convnd bwd weight

* 1. remove comments. 2. add checkvalidity. 3. add gridsize computation

* add workspace size check

* fix format

* change function name

[ROCm/composable_kernel commit: 561ec12f4a]

2022-06-16 14:16:01 -05:00

..

ck

example for convnd bwd weight bf16 splitk (#265)

2022-06-16 14:16:01 -05:00