diff options
| author | Gian Marco Iodice <gianmarco.iodice@arm.com> | 2019-09-27 09:23:15 +0100 |
|---|---|---|
| committer | Gian Marco Iodice <gianmarco.iodice@arm.com> | 2019-09-30 08:28:43 +0000 |
| commit | 0c17aa25a4f7bc812707150b91930f0cf8e75294 (patch) | |
| tree | 29088e00bd7ba443dc122ad3436b0a4ef369a102 /tests/validation/reference/GEMM.h | |
| parent | 40958adf8bad8fd9fefe591ee55a381f7bbb6fea (diff) | |
| download | ComputeLibrary-0c17aa25a4f7bc812707150b91930f0cf8e75294.tar.gz | |
COMPMID-2571: Add mixed-precision support in CLGEMMReshaped for FP16
Change-Id: I5ba90d4de4594ed784c7230aa6b10503be67c001
Signed-off-by: Gian Marco Iodice <gianmarco.iodice@arm.com>
Reviewed-on: https://review.mlplatform.org/c/1991
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'tests/validation/reference/GEMM.h')
| -rw-r--r-- | tests/validation/reference/GEMM.h | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/tests/validation/reference/GEMM.h b/tests/validation/reference/GEMM.h index 39007c60bc..9bcd640770 100644 --- a/tests/validation/reference/GEMM.h +++ b/tests/validation/reference/GEMM.h @@ -1,5 +1,5 @@ /* - * Copyright (c) 2017-2018 ARM Limited. + * Copyright (c) 2017-2019 ARM Limited. * * SPDX-License-Identifier: MIT * @@ -38,6 +38,9 @@ namespace reference template <typename T, typename std::enable_if<is_floating_point<T>::value, int>::type = 0> SimpleTensor<T> gemm(const SimpleTensor<T> &a, const SimpleTensor<T> &b, const SimpleTensor<T> &c, float alpha, float beta); +template <typename T, typename std::enable_if<is_floating_point<T>::value, int>::type = 0> +SimpleTensor<T> gemm_mixed_precision(const SimpleTensor<T> &a, const SimpleTensor<T> &b, const SimpleTensor<T> &c, float alpha, float beta); + } // namespace reference } // namespace validation } // namespace test |