diff options
author | Cedric Nugteren <web@cedricnugteren.nl> | 2018-09-07 22:02:44 +0200 |
---|---|---|
committer | Cedric Nugteren <web@cedricnugteren.nl> | 2018-09-07 22:02:44 +0200 |
commit | c788e040f7f4e46d9f03644cadb65788fe42571e (patch) | |
tree | 46cfc164d5e0104174ee1b7ff1489eee2b23688d /src/kernels | |
parent | 2dd539f911dc9e53f188ed404ba95a795ee56fb6 (diff) |
Added xCONVGEMM as im2col plus a batched GEMM kernel
Diffstat (limited to 'src/kernels')
-rw-r--r-- | src/kernels/levelx/xconvgemm_part1.opencl | 1 | ||||
-rw-r--r-- | src/kernels/levelx/xconvgemm_part2.opencl | 1 |
2 files changed, 2 insertions, 0 deletions
diff --git a/src/kernels/levelx/xconvgemm_part1.opencl b/src/kernels/levelx/xconvgemm_part1.opencl index 6f870ec0..abdb5324 100644 --- a/src/kernels/levelx/xconvgemm_part1.opencl +++ b/src/kernels/levelx/xconvgemm_part1.opencl @@ -11,6 +11,7 @@ // uses parameters from the direct GEMM kernel. This is the part with the loads from memory (1/2). // This uses "CONVGEMM_WITH_IM2COL" as a switch to select between direct convgemm or first running // the im2col kernel to create a 'col' temporary matrix. +// TODO: Currently only works with 'CONVGEMM_WITH_IM2COL' set // // ================================================================================================= diff --git a/src/kernels/levelx/xconvgemm_part2.opencl b/src/kernels/levelx/xconvgemm_part2.opencl index 46a72711..e0ac24a0 100644 --- a/src/kernels/levelx/xconvgemm_part2.opencl +++ b/src/kernels/levelx/xconvgemm_part2.opencl @@ -11,6 +11,7 @@ // uses parameters from the direct GEMM kernel. This part contains the main kernel (2/2). // This uses "CONVGEMM_WITH_IM2COL" as a switch to select between direct convgemm or first running // the im2col kernel to create a 'col' temporary matrix. +// TODO: Currently only works with 'CONVGEMM_WITH_IM2COL' set // // ================================================================================================= |