ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-12 01:10:17 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
b8b2d0a6d1f6342686ee890eac64a1506d865452
composable_kernel/driver/include
History
Chao Liu b8b2d0a6d1 DL GEMM fp32/fp16/int8 (#41) 
* add threadwise copy the copy a tensor in one copy, added kpack to DL GEMM

* add kpack into fwd v4r5 nchw fp32
2021-07-04 22:50:29 -05:00
..
conv_common.hpp
Restructure gridwise and blockwise GEMM, add tensor contraction and FWD-v4r5 (#36)
2021-06-09 23:53:08 -05:00
conv_tunables.hpp
Add online compilation for dynamic kernels (#37)
2021-06-24 08:34:19 -05:00
device_convolution_backward_data_implicit_gemm_v1r1_nchw_kcyx_nkhw.hpp
Dynamic tensor descriptor (#24)
2021-03-25 13:51:11 -05:00
device_convolution_backward_data_implicit_gemm_v1r2_nchw_kcyx_nkhw.hpp
Code clean up (#20)
2020-06-23 20:31:27 -05:00
device_convolution_backward_data_implicit_gemm_v4r1_nchw_kcyx_nkhw.hpp
Dynamic tensor descriptor (#24)
2021-03-25 13:51:11 -05:00
device_convolution_backward_data_implicit_gemm_v5r1_nhwc_kyxc_nhwk.hpp
Dynamic tensor descriptor (#24)
2021-03-25 13:51:11 -05:00
device_convolution_forward_implicit_gemm_v4r1_nchw_kcyx_nkhw.hpp
Dynamic tensor descriptor (#24)
2021-03-25 13:51:11 -05:00
device_convolution_forward_implicit_gemm_v4r4_nchw_kcyx_nkhw.hpp
Dynamic tensor descriptor (#24)
2021-03-25 13:51:11 -05:00
device_convolution_forward_implicit_gemm_v4r4_nhwc_kyxc_nhwk.hpp
Dynamic tensor descriptor (#24)
2021-03-25 13:51:11 -05:00
device_dynamic_convolution_backward_data_implicit_gemm_v4r1_xdlops_nhwc_kyxc_nhwk.hpp
xdlops_v4r4_fwd fp32/fp16 (#34)
2021-07-01 14:33:00 -05:00
device_dynamic_convolution_backward_data_implicit_gemm_v4r1r2_xdlops_nhwc_kyxc_nhwk.hpp
xdlops_v4r4_fwd fp32/fp16 (#34)
2021-07-01 14:33:00 -05:00
device_dynamic_convolution_forward_implicit_gemm_v4r4_nchw_kcyx_nkhw.hpp
Restructure gridwise and blockwise GEMM, add tensor contraction and FWD-v4r5 (#36)
2021-06-09 23:53:08 -05:00
device_dynamic_convolution_forward_implicit_gemm_v4r4_nhwc_kyxc_nhwk.hpp
Restructure gridwise and blockwise GEMM, add tensor contraction and FWD-v4r5 (#36)
2021-06-09 23:53:08 -05:00
device_dynamic_convolution_forward_implicit_gemm_v4r4_xdlops_nchw_kcyx_nkhw.hpp
xdlops_v4r4_fwd fp32/fp16 (#34)
2021-07-01 14:33:00 -05:00
device_dynamic_convolution_forward_implicit_gemm_v4r4r2_nhwc_kyxc_nhwk.hpp
DL GEMM fp32/fp16/int8 (#41)
2021-07-04 22:50:29 -05:00
device_dynamic_convolution_forward_implicit_gemm_v4r4r2_xdlops_nchw_kcyx_nkhw.hpp
xdlops_v4r4_fwd fp32/fp16 (#34)
2021-07-01 14:33:00 -05:00
device_dynamic_convolution_forward_implicit_gemm_v4r4r2_xdlops_nhwc_kyxc_nhwk.hpp
xdlops_v4r4_fwd fp32/fp16 (#34)
2021-07-01 14:33:00 -05:00
device_dynamic_convolution_forward_implicit_gemm_v4r4r3_xdlops_nhwc_kyxc_nhwk.hpp
xdlops_v4r4_fwd fp32/fp16 (#34)
2021-07-01 14:33:00 -05:00
device_dynamic_convolution_forward_implicit_gemm_v4r4r4_xdlops_nhwc_kyxc_nhwk.hpp
DL GEMM fp32/fp16/int8 (#41)
2021-07-04 22:50:29 -05:00
device_dynamic_convolution_forward_implicit_gemm_v4r5_nchw_kcyx_nkhw.hpp
Restructure gridwise and blockwise GEMM, add tensor contraction and FWD-v4r5 (#36)
2021-06-09 23:53:08 -05:00
device_dynamic_convolution_forward_implicit_gemm_v4r5r2_nchw_kcyx_nkhw.hpp
DL GEMM fp32/fp16/int8 (#41)
2021-07-04 22:50:29 -05:00
device_dynamic_convolution_forward_implicit_gemm_v5r1_nchw_kcyx_nkhw.hpp
Restructure gridwise and blockwise GEMM, add tensor contraction and FWD-v4r5 (#36)
2021-06-09 23:53:08 -05:00
device_tensor.hpp
Code clean up (#20)
2020-06-23 20:31:27 -05:00
device.hpp
DL GEMM fp32/fp16/int8 (#41)
2021-07-04 22:50:29 -05:00
host_conv_bwd_data.hpp
xdlops_v4r4_fwd fp32/fp16 (#34)
2021-07-01 14:33:00 -05:00
host_conv.hpp
xdlops_v4r4_fwd fp32/fp16 (#34)
2021-07-01 14:33:00 -05:00
host_tensor_generator.hpp
xdlops_v4r4_fwd fp32/fp16 (#34)
2021-07-01 14:33:00 -05:00
host_tensor.hpp
xdlops_v4r4_fwd fp32/fp16 (#34)
2021-07-01 14:33:00 -05:00
olc_device_dynamic_convolution_forward_implicit_gemm_v4r4_nchw_kcyx_nkhw.hpp
Add online compilation for dynamic kernels (#37)
2021-06-24 08:34:19 -05:00
olc_device_dynamic_convolution_forward_implicit_gemm_v4r5_nchw_kcyx_nkhw.hpp
Add online compilation for dynamic kernels (#37)
2021-06-24 08:34:19 -05:00
olc_driver_common.hpp
Add online compilation for dynamic kernels (#37)
2021-06-24 08:34:19 -05:00