diff options
author | Giorgio Arena <giorgio.arena@arm.com> | 2021-10-13 11:13:04 +0100 |
---|---|---|
committer | Giorgio Arena <giorgio.arena@arm.com> | 2021-10-14 14:54:48 +0000 |
commit | 945ae9e4027655267170ecc56563c362d8110d1e (patch) | |
tree | d5c695462c57ca88bc628901e4b26b739d440651 /src/gpu/cl/ClKernelLibrary.cpp | |
parent | de23fc381aca403c94870d7f8bc78716eb350d53 (diff) | |
download | ComputeLibrary-945ae9e4027655267170ecc56563c362d8110d1e.tar.gz |
Implement CLDirectConv3D f32/f16
Resolve COMPMID-4660
Signed-off-by: Giorgio Arena <giorgio.arena@arm.com>
Change-Id: Ibd66ec1eb6faa60086981b1e3a9c12561df3445f
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/6420
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Gian Marco Iodice <gianmarco.iodice@arm.com>
Diffstat (limited to 'src/gpu/cl/ClKernelLibrary.cpp')
-rw-r--r-- | src/gpu/cl/ClKernelLibrary.cpp | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/src/gpu/cl/ClKernelLibrary.cpp b/src/gpu/cl/ClKernelLibrary.cpp index c05bb96753..4af42262b9 100644 --- a/src/gpu/cl/ClKernelLibrary.cpp +++ b/src/gpu/cl/ClKernelLibrary.cpp @@ -433,6 +433,7 @@ const std::map<std::string, std::string> ClKernelLibrary::_kernel_program_map = { "dwc_native_fp_nhwc", "nhwc/dwc_native_fp_nhwc.cl" }, { "dwc_native_quantized_nhwc", "nhwc/dwc_native_quantized_nhwc.cl" }, { "direct_convolution_nhwc", "nhwc/direct_convolution.cl" }, + { "direct_convolution3d_ndhwc", "nhwc/direct_convolution3d.cl" }, { "im2col3x3_nhwc", "nhwc/im2col.cl" }, { "im2col9x9_nhwc", "nhwc/im2col.cl" }, { "im2col_generic_nhwc", "nhwc/im2col.cl" }, @@ -858,6 +859,10 @@ const std::map<std::string, std::string> ClKernelLibrary::_program_source_map = #include "./cl_kernels/nhwc/direct_convolution.clembed" }, { + "nhwc/direct_convolution3d.cl", +#include "./cl_kernels/nhwc/direct_convolution3d.clembed" + }, + { "nhwc/dwc_native_fp_nhwc.cl", #include "./cl_kernels/nhwc/dwc_native_fp_nhwc.clembed" }, |