squall/cutlass
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity
dd571f0edb
cutlass/tools/library/src/reference
History
Fujun Han 1e4703cbab
Support parallel split K mode for porfiling (#277) 
* Support parallel split K mode for porfiling

Signed-off-by: Peter Han <fujun.han@iluvatar.ai>

* Parallel Split K support

  1. find gemm kernel by preference key
  2. switch m n for redution kernel

Signed-off-by: Peter Han <fujun.han@iluvatar.ai>

* parallel splitk for fp16 gemm

* add one missing file

Co-authored-by: Haicheng Wu <haichengw@nvidia.com>
2022-01-27 10:37:37 -05:00
..
conv2d.cu Cutlass 2.6 Update 1 (#301) 2021-07-27 17:58:30 -07:00
conv3d.cu Cutlass 2.6 Update 1 (#301) 2021-07-27 17:58:30 -07:00
conv_reference_operation.h Support parallel split K mode for porfiling (#277) 2022-01-27 10:37:37 -05:00
gemm_reference_operation.h Support parallel split K mode for porfiling (#277) 2022-01-27 10:37:37 -05:00
gemm.cu Cutlass 2.6 Update 1 (#301) 2021-07-27 17:58:30 -07:00
initialize_reference_operations.cu Cutlass 2.6 Update 1 (#301) 2021-07-27 17:58:30 -07:00