diff options
| author | Georgios Pinitas <georgios.pinitas@arm.com> | 2019-01-23 11:24:50 +0000 |
|---|---|---|
| committer | Michalis Spyrou <michalis.spyrou@arm.com> | 2019-01-24 10:19:46 +0000 |
| commit | 1d480652b820317fc97ccbc3cb517e3b9e8be197 (patch) | |
| tree | b3c845ec02cccf89430b95186ed3e3f2ae65e2bd /src/core/NEON/kernels/arm_gemm/gemm_interleaved.hpp | |
| parent | 20b527a7029d02d0edda78fd92002cbc430dbe05 (diff) | |
| download | ComputeLibrary-1d480652b820317fc97ccbc3cb517e3b9e8be197.tar.gz | |
COMPMID-1867: Add u8 and s8 hybrid assembly kernels.
Change-Id: Ifeb005f9d18d19feff11949474cce84d9e03749c
Reviewed-on: https://review.mlplatform.org/565
Reviewed-by: Michalis Spyrou <michalis.spyrou@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'src/core/NEON/kernels/arm_gemm/gemm_interleaved.hpp')
| -rw-r--r-- | src/core/NEON/kernels/arm_gemm/gemm_interleaved.hpp | 13 |
1 files changed, 7 insertions, 6 deletions
diff --git a/src/core/NEON/kernels/arm_gemm/gemm_interleaved.hpp b/src/core/NEON/kernels/arm_gemm/gemm_interleaved.hpp index 436438f351..b83ccd3407 100644 --- a/src/core/NEON/kernels/arm_gemm/gemm_interleaved.hpp +++ b/src/core/NEON/kernels/arm_gemm/gemm_interleaved.hpp @@ -340,7 +340,7 @@ public: _k_block = std::max(_k_block, 1U) * strategy::k_unroll(); // Now tune to presented problem size; this is how many blocks we need. - int num_k_blocks = iceildiv(_Ksize, _k_block); + unsigned int num_k_blocks = iceildiv(_Ksize, _k_block); // So divide the space equally into that many blocks. _k_block = iceildiv(_Ksize, num_k_blocks); @@ -363,7 +363,7 @@ public: _x_block = std::max(_x_block, 1U) * strategy::out_width(); // And tune to the presented problem size. - int num_x_blocks = iceildiv(_Nsize, _x_block); + unsigned int num_x_blocks = iceildiv(_Nsize, _x_block); _x_block = iceildiv(_Nsize, num_x_blocks); _x_block = iceildiv(_x_block, strategy::out_width()); @@ -464,8 +464,8 @@ public: do { /* Figure out the size of each block. */ - size_t x_size = (current.xmax() - current.x0()); - size_t k_size = (current.kmax() - current.k0()); + unsigned int x_size = (current.xmax() - current.x0()); + unsigned int k_size = (current.kmax() - current.k0()); /* Round sizes up as needed. */ x_size = iceildiv(x_size, strategy::out_width()); @@ -480,6 +480,7 @@ public: return total; } + using GemmCommon<To, Tr>::pretranspose_B_array; void pretranspose_B_array(void *in_buffer, const To *B, const int ldb, const int B_multi_stride) override { blockwalker current(*this); Toi *buffer = reinterpret_cast<Toi *>(in_buffer); @@ -488,8 +489,8 @@ public: do { /* Figure out the size of each block. */ - size_t x_size = (current.xmax() - current.x0()); - size_t k_size = (current.kmax() - current.k0()); + unsigned int x_size = (current.xmax() - current.x0()); + unsigned int k_size = (current.kmax() - current.k0()); /* Round sizes up as needed. */ x_size = iceildiv(x_size, strategy::out_width()); |