squall/cutlass
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity
dd571f0edb
cutlass/examples/30_wgrad_split_k
History
Masahiro Masuda 0e71d9b450
Transposed conv2d and wgrad split k examples (#413) 
* add split k wgrad example

* wgrad done

* begin transposed conv2d example

* update transposed conv2d example and add ref check

* update doc for conv2d transpose example

* add license

* add wgrad doc

* more clarification on GEMM output type

* typo fix

* clean up indent

* address comments

* rename example numbers to 34 and 35

* GEMM -> Implicit GEMM

* Revert "rename example numbers to 34 and 35"

This reverts commit 551a808c227216e9e38d4472ba8ff020557b8500.

* transposed_conv2d is 34

* add compiler and device version check to exit gracefully

Co-authored-by: Haicheng Wu <haichengw@nvidia.com>
2022-03-23 14:52:54 -04:00
..
30_wgrad_split_k.cu Transposed conv2d and wgrad split k examples (#413) 2022-03-23 14:52:54 -04:00
CMakeLists.txt Transposed conv2d and wgrad split k examples (#413) 2022-03-23 14:52:54 -04:00