diff options
author | Adnan AlSinan <adnan.alsinan@arm.com> | 2023-10-24 12:03:21 +0100 |
---|---|---|
committer | Adnan AlSinan <adnan.alsinan@arm.com> | 2023-10-31 11:00:45 +0000 |
commit | fde45d836cf753a94915ac42d8a13da7edc52221 (patch) | |
tree | 6ed787749aa3caec13a0b3c2c64ea591b423089c /compute_kernel_writer/prototype/src | |
parent | 5ef0bdd53dd2ce6bc7ad28077ffac3bf9e939b5f (diff) | |
download | ComputeLibrary-fde45d836cf753a94915ac42d8a13da7edc52221.tar.gz |
Extend CKW MatMul with nt_t
- Add the kernel variant: (nt_t) to GpuCKWMatMul.
- Extend CKW MatMul validation test with nt_t.
- Fixes a bug in CKW where z-dim = 1.
Resolves: COMPMID-6435
Signed-off-by: Adnan AlSinan <adnan.alsinan@arm.com>
Change-Id: I4c5e8791e55f21ffff3c11eca7802c51a4259977
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/10525
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Gian Marco Iodice <gianmarco.iodice@arm.com>
Benchmark: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'compute_kernel_writer/prototype/src')
-rw-r--r-- | compute_kernel_writer/prototype/src/Prototype.h | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/compute_kernel_writer/prototype/src/Prototype.h b/compute_kernel_writer/prototype/src/Prototype.h index 433eef9e7b..b392fe2651 100644 --- a/compute_kernel_writer/prototype/src/Prototype.h +++ b/compute_kernel_writer/prototype/src/Prototype.h @@ -3050,7 +3050,7 @@ private: address += " * "; address += stride_y; } - if (z != "0" && (_mapper.is_one_component_z() != true)) + if (z != "0") { const std::string stride_z = _mapper.tensor_component_stride_z(); address += " + ("; |