commit 19e9142604edce6efbf117181578f8f408008134
author zhenglin <zhenglin.li@arm.com> Wed Jan 03 12:14:13 2018 +0800
committer Anthony Barbier <anthony.barbier@arm.com> Fri Nov 02 16:42:33 2018 +0000
tree 14751c98d4f7bad673209ec709310bdcd5a3e064
parent fef6dae9c2cfe1003ab2abe3a41255e849b1b5eb

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>

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 404b46a..3fb0e23 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 @@
 #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 @@
 #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)