diff options
author | zhenglin <zhenglin.li@arm.com> | 2018-01-03 12:14:13 +0800 |
---|---|---|
committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:42:33 +0000 |
commit | 19e9142604edce6efbf117181578f8f408008134 (patch) | |
tree | 14751c98d4f7bad673209ec709310bdcd5a3e064 /src/core/GLES_COMPUTE/cs_shaders/helpers_cs.h | |
parent | fef6dae9c2cfe1003ab2abe3a41255e849b1b5eb (diff) | |
download | ComputeLibrary-19e9142604edce6efbf117181578f8f408008134.tar.gz |
APPBROWSER-369: Rewrite the gemm.cs with the new common code
Change-Id: I9db00c846fa7fc223a22ab775025dfdea587ade8
Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/114957
Reviewed-by: Joel Liang <joel.liang@arm.com>
Reviewed-by: Pablo Tello <pablo.tello@arm.com>
Tested-by: Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'src/core/GLES_COMPUTE/cs_shaders/helpers_cs.h')
-rwxr-xr-x | src/core/GLES_COMPUTE/cs_shaders/helpers_cs.h | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/src/core/GLES_COMPUTE/cs_shaders/helpers_cs.h b/src/core/GLES_COMPUTE/cs_shaders/helpers_cs.h index 404b46a89d..3fb0e238ae 100755 --- a/src/core/GLES_COMPUTE/cs_shaders/helpers_cs.h +++ b/src/core/GLES_COMPUTE/cs_shaders/helpers_cs.h @@ -222,6 +222,9 @@ ImageIterator update_image_from_tensor3D_iter_offset(uint element_shift, uint of #define TENSOR3D_OFFSET(tensor_iter, x, y, z) \ uint(tensor3D_offset_in_bytes(tensor_iter, int(x), int(y), int(z)) >> tensor_iter.element_shift) +#define TENSOR_OFFSET_ADVANCE(tensor_iter, n) \ + uint((tensor_iter.current_offset_in_bytes >> tensor_iter.element_shift) + int(n)) + #define TENSOR_OFFSET_ADVANCE_IN_BYTES(tensor_iter, n) \ uint((tensor_iter.current_offset_in_bytes + int(n)) >> tensor_iter.element_shift) @@ -231,6 +234,9 @@ ImageIterator update_image_from_tensor3D_iter_offset(uint element_shift, uint of #define CURRENT_ITEM_OFFSET_IN_BYTES(tensor_iter) \ uint(tensor_iter.current_offset_in_bytes) +#define TENSOR_ITERATOR_ADVANCE(tensor_iter, n) \ + tensor_iter.current_offset_in_bytes += (int(n) << tensor_iter.element_shift) + #define TENSOR_ITERATOR_ADVANCE_IN_BYTES(tensor_iter, n) \ tensor_iter.current_offset_in_bytes += int(n) |