ltqin 7a9b93f4b6 Example for conv2d backward weight fp16 (#106) ...

* add wrw reference

* start device

* raw not split version

* run simple example

* start to use atomic add

* simple transform result correct

* first version that can run

* fix atomic and set operator choice

* add check split-k

* format

* change input parameter

* add pad for t total

* rename example index

Co-authored-by: ltqin <letaoqin@amd.com>

2022-03-04 21:18:15 -06:00

..

include

Example for conv2d backward weight fp16 (#106)

2022-03-04 21:18:15 -06:00

src

[Bf16 & int8] [example & ckprofiler] (#100)

2022-03-04 15:56:44 -06:00

CMakeLists.txt

[Bf16 & int8] [example & ckprofiler] (#100)

2022-03-04 15:56:44 -06:00