zjing14 b53e9d08ed Batched GEMM for fp16 (#79) ...

* prepare host for batched_gemm

* init commit of batched kernels

* fixed

* refine transform with freeze

* m/n padding

* fixed a bug; clean

* add small tiles

* clean

* clean code

* clean code

* add nt, tn, tt layout

* add missing file

* use StaticBufferTupleOfVector instead

* add reference_batched_gemm

* fixed a macro

2022-02-11 09:36:52 -06:00

..

convolution_forward_specialization.hpp

Fusion Conv+Bias+ReLU(+Add) (#62)

2021-12-26 07:43:42 -07:00

device_base.hpp

Fusion Conv+Bias+ReLU(+Add) (#62)

2021-12-26 07:43:42 -07:00

device_batched_gemm_xdl.hpp

Batched GEMM for fp16 (#79)

2022-02-11 09:36:52 -06:00

device_conv2d_fwd_xdl_c_shuffle_bias_activation_add_nhwc_kyxc_nhwk.hpp

GEMM+Bias+ReLU+Add (#76)

2022-02-06 22:32:47 -06:00

device_conv2d_fwd_xdl_c_shuffle_bias_activation_nhwc_kyxc_nhwk.hpp

Fusion Conv+Bias+ReLU(+Add) (#62)

2021-12-26 07:43:42 -07:00

device_conv2d_fwd_xdl_c_shuffle_nhwc_kyxc_nhwk.hpp

Fusion Conv+Bias+ReLU(+Add) (#62)

2021-12-26 07:43:42 -07:00

device_conv2d_fwd_xdl_nhwc_kyxc_nhwk.hpp

Fusion Conv+Bias+ReLU(+Add) (#62)

2021-12-26 07:43:42 -07:00

device_conv_fwd_bias_activation_add.hpp

Fusion Conv+Bias+ReLU(+Add) (#62)

2021-12-26 07:43:42 -07:00

device_conv_fwd_bias_activation.hpp

Fusion Conv+Bias+ReLU(+Add) (#62)

2021-12-26 07:43:42 -07:00

device_conv_fwd.hpp

Fusion Conv+Bias+ReLU(+Add) (#62)

2021-12-26 07:43:42 -07:00

device_gemm_bias_activation_add.hpp

GEMM+Bias+ReLU+Add (#76)

2022-02-06 22:32:47 -06:00

device_gemm_bias_activation.hpp

GEMM+Bias+ReLU+Add (#76)

2022-02-06 22:32:47 -06:00

device_gemm_xdl_c_shuffle_bias_2d.hpp

Support alpha beta scaling for GEMM (#78)

2022-02-11 00:48:41 -06:00

device_gemm_xdl_c_shuffle_bias_activation_add.hpp

fix build breaks (#81)

2022-02-10 23:52:19 -06:00

device_gemm_xdl_c_shuffle_bias_activation.hpp

fix build breaks (#81)

2022-02-10 23:52:19 -06:00

device_gemm_xdl_c_shuffle.hpp

fix build breaks (#81)

2022-02-10 23:52:19 -06:00

device_gemm_xdl_splitk.hpp

add split-k GEMM (#59)

2022-02-02 22:47:27 -06:00

device_gemm_xdl.hpp

fix build breaks (#81)

2022-02-10 23:52:19 -06:00

device_gemm.hpp

Support alpha beta scaling for GEMM (#78)

2022-02-11 00:48:41 -06:00

device_operation_instance.hpp

Fusion Conv+Bias+ReLU(+Add) (#62)

2021-12-26 07:43:42 -07:00

tensor_layout.hpp

ckProfiler and device-level XDL GEMM operator (#48)

2021-11-14 11:28:32 -06:00