From 19e9142604edce6efbf117181578f8f408008134 Mon Sep 17 00:00:00 2001 From: zhenglin Date: Wed, 3 Jan 2018 12:14:13 +0800 Subject: 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 Reviewed-by: Pablo Tello Tested-by: Jenkins --- src/core/GLES_COMPUTE/cs_shaders/helpers_cs.h | 6 ++++++ 1 file changed, 6 insertions(+) (limited to 'src/core/GLES_COMPUTE/cs_shaders/helpers_cs.h') 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) -- cgit v1.2.1