diff options
author | Tracy Narine <tracy.narine@arm.com> | 2024-02-06 15:22:41 +0000 |
---|---|---|
committer | TeresaARM <teresa.charlinreyes@arm.com> | 2024-02-07 21:04:02 +0000 |
commit | bc5a5d59e07d2365c2c88b3430b4a311d186ebbc (patch) | |
tree | 345691d4b83824e13b99348617dd68866e5f530f /src/backends/gpuFsa/test/GpuFsaEndToEndTests.cpp | |
parent | 7bcae3c835468d9b0770514dc7127f02d47cec5f (diff) | |
download | armnn-bc5a5d59e07d2365c2c88b3430b4a311d186ebbc.tar.gz |
IVGCVSW-7622 GpuFsa Op: Add Cast operator
* Added cast operator support
Signed-off-by: Tracy Narine <tracy.narine@arm.com>
Change-Id: Ie12cb1559a7a059ff35e1c395bc77243499243cd
Diffstat (limited to 'src/backends/gpuFsa/test/GpuFsaEndToEndTests.cpp')
-rw-r--r-- | src/backends/gpuFsa/test/GpuFsaEndToEndTests.cpp | 19 |
1 files changed, 19 insertions, 0 deletions
diff --git a/src/backends/gpuFsa/test/GpuFsaEndToEndTests.cpp b/src/backends/gpuFsa/test/GpuFsaEndToEndTests.cpp index 93a4a81f9d..ccab0bf4f9 100644 --- a/src/backends/gpuFsa/test/GpuFsaEndToEndTests.cpp +++ b/src/backends/gpuFsa/test/GpuFsaEndToEndTests.cpp @@ -6,6 +6,7 @@ #include "backendsCommon/test/EndToEndTestImpl.hpp" #include "backendsCommon/test/Convolution2dEndToEndTestImpl.hpp" +#include "backendsCommon/test/layerTests/CastTestImpl.hpp" #include "backendsCommon/test/DepthwiseConvolution2dEndToEndTests.hpp" #include "backendsCommon/test/ElementwiseBinaryEndToEndTestImpl.hpp" @@ -19,6 +20,24 @@ TEST_SUITE("GpuFsaEndToEnd") std::vector<BackendId> gpuFsaDefaultBackends = {"GpuFsa"}; +TEST_CASE("GpuFsaCastEndtoEndTestFloat32ToFloat16") +{ + using namespace half_float::literal; + + std::vector<unsigned int> inputShape { 2, 2, 2 }; + + std::vector<float> inputValues { -3.5f, -1.2f, -8.6f, -2.0f, -1.5f, -1.3f, -0.5f, -0.4f }; + + std::vector<armnn::Half> outputValues { -3.50_h, -1.20_h, -8.6_h, -2._h, -1.50_h, -1.30_h, -0.50_h, -0.40_h }; + + CastSimpleTest<DataType::Float32, DataType::Float16, float, armnn::Half>(gpuFsaDefaultBackends, + inputShape, + inputValues, + outputValues, + 1.0f, + 0); +} + // Conv2d TEST_CASE("GpuFsaConv2dEndtoEndTestFloat32") { |