diff options
| author | Georgios Pinitas <georgios.pinitas@arm.com> | 2018-08-28 16:24:56 +0100 |
|---|---|---|
| committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:54:54 +0000 |
| commit | ea9e0dc18c408fecb6dc482b774bd900dd321610 (patch) | |
| tree | b6e67a6559b53b5d4d97f77251d83ac73a6e55a5 /src/runtime/NEON/functions/NEGEMM.cpp | |
| parent | 84797636b0ad44c16838df4177cf5a05aa929781 (diff) | |
| download | ComputeLibrary-ea9e0dc18c408fecb6dc482b774bd900dd321610.tar.gz | |
COMPMID-1469: Add validate in NEGEMMMatrixAdditionKernel
Change-Id: I228e2503eb40c12869fbd7e834ac1309aa613480
Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/145878
Reviewed-by: Giorgio Arena <giorgio.arena@arm.com>
Tested-by: Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'src/runtime/NEON/functions/NEGEMM.cpp')
| -rw-r--r-- | src/runtime/NEON/functions/NEGEMM.cpp | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/src/runtime/NEON/functions/NEGEMM.cpp b/src/runtime/NEON/functions/NEGEMM.cpp index de51266267..321ecf85d8 100644 --- a/src/runtime/NEON/functions/NEGEMM.cpp +++ b/src/runtime/NEON/functions/NEGEMM.cpp @@ -218,6 +218,12 @@ Status NEGEMM::validate(const ITensorInfo *a, const ITensorInfo *b, const ITenso ARM_COMPUTE_RETURN_ON_ERROR(NEGEMMMatrixMultiplyKernel::validate(matrix_a_info, matrix_b_info, &tmp_output_info, alpha, run_interleave_transpose, reshape_info)); } + // Validate matrix addition kernel + if(beta != 0 && c != nullptr) + { + ARM_COMPUTE_RETURN_ON_ERROR(NEGEMMMatrixAdditionKernel::validate(c, output, beta)); + } + return Status{}; } |