Bing Xu d0d941efc7 [hardswish] correct implmentation (#403) ...

* [hardswish] correct implmentation

* seems working

* hardswish fp32/fp16x2 optimization

* [relu] half2 support

* add relu0; add multiply_add_relu0;

* cleanup

Co-authored-by: Bing Xu <bingxu@fb.com>
Co-authored-by: Haicheng Wu <haichengw@nvidia.com>

2022-02-09 14:28:53 -05:00

..

default_epilogue_complex_tensor_op.h

Cutlass 2.6 Update 1 (#301)

2021-07-27 17:58:30 -07:00

default_epilogue_direct_store.h

CUTLASS 2.8 (#363)

2021-11-19 13:26:35 -08:00

default_epilogue_planar_complex.h

Cutlass 2.6 Update 1 (#301)

2021-07-27 17:58:30 -07:00

default_epilogue_simt.h

Cutlass 2.6 Update 1 (#301)

2021-07-27 17:58:30 -07:00

default_epilogue_tensor_op.h

[hardswish] correct implmentation (#403)

2022-02-09 14:28:53 -05:00

default_epilogue_volta_tensor_op.h

Cutlass 2.6 Update 1 (#301)

2021-07-27 17:58:30 -07:00

default_epilogue_with_broadcast.h

Cutlass 2.6 Update 1 (#301)

2021-07-27 17:58:30 -07:00

default_epilogue_with_reduction.h

Cutlass 2.6 Update 1 (#301)

2021-07-27 17:58:30 -07:00

default_epilogue_wmma_tensor_op.h

Cutlass 2.6 Update 1 (#301)

2021-07-27 17:58:30 -07:00

default_thread_map_simt.h

Cutlass 2.6 Update 1 (#301)

2021-07-27 17:58:30 -07:00

default_thread_map_tensor_op.h

Cutlass 2.6 Update 1 (#301)

2021-07-27 17:58:30 -07:00

default_thread_map_volta_tensor_op.h

Cutlass 2.6 Update 1 (#301)

2021-07-27 17:58:30 -07:00

default_thread_map_wmma_tensor_op.h

Cutlass 2.6 Update 1 (#301)

2021-07-27 17:58:30 -07:00

direct_store_epilogue_iterator.h

CUTLASS 2.8 (#363)

2021-11-19 13:26:35 -08:00

epilogue_base.h

Cutlass 2.6 Update 1 (#301)

2021-07-27 17:58:30 -07:00

epilogue_direct_store.h

CUTLASS 2.8 (#363)

2021-11-19 13:26:35 -08:00

epilogue_gemm_k_reduction.h

Cutlass 2.6 Update 1 (#301)

2021-07-27 17:58:30 -07:00

epilogue_planar_complex.h

Cutlass 2.6 Update 1 (#301)

2021-07-27 17:58:30 -07:00

epilogue_smem_accumulator.h

CUTLASS 2.8 (#363)

2021-11-19 13:26:35 -08:00

epilogue_with_broadcast.h

Updates to fused epilogue (#383)

2021-12-17 16:04:43 -05:00

epilogue_with_reduction.h

Updates to fused epilogue (#383)

2021-12-17 16:04:43 -05:00

epilogue_workspace.h

Cutlass 2.6 Update 1 (#301)

2021-07-27 17:58:30 -07:00

epilogue.h

Cutlass 2.6 Update 1 (#301)

2021-07-27 17:58:30 -07:00

interleaved_epilogue.h

Cutlass 2.6 Update 1 (#301)

2021-07-27 17:58:30 -07:00

output_iterator_parameter.h

CUTLASS 2.4 (Implicit GEMM convolution) (#147)

2020-11-19 21:25:25 -08:00

output_tile_thread_map.h

Cutlass 2.6 Update 1 (#301)

2021-07-27 17:58:30 -07:00

predicated_tile_iterator_affine.h

Cutlass 2.6 Update 1 (#301)

2021-07-27 17:58:30 -07:00

predicated_tile_iterator_params.h

Cutlass 2.6 Update 1 (#301)

2021-07-27 17:58:30 -07:00

predicated_tile_iterator_predicates.h

Cutlass 2.6 Update 1 (#301)

2021-07-27 17:58:30 -07:00

predicated_tile_iterator_strided_dgrad.h

CUTLASS 2.6.1 - functional and performance enhancements to strided DGRAD, fixes, and tuning

2021-09-03 10:26:15 -07:00

predicated_tile_iterator.h

CUTLASS 2.6.1 - functional and performance enhancements to strided DGRAD, fixes, and tuning

2021-09-03 10:26:15 -07:00

shared_load_iterator_mixed.h

Cutlass 2.6 Update 1 (#301)

2021-07-27 17:58:30 -07:00

shared_load_iterator.h

Cutlass 2.6 Update 1 (#301)

2021-07-27 17:58:30 -07:00