24.02.1
|
#include "arm_compute/core/Types.h"
#include "arm_compute/core/utils/misc/ShapeCalculator.h"
#include "arm_compute/runtime/CL/CLTensor.h"
#include "arm_compute/runtime/CL/CLTensorAllocator.h"
#include "src/gpu/cl/kernels/ClGemmReshapeRhsMatrixKernel.h"
#include "tests/CL/CLAccessor.h"
#include "tests/CL/Helper.h"
#include "tests/PaddingCalculator.h"
#include "tests/datasets/ShapeDatasets.h"
#include "tests/framework/Asserts.h"
#include "tests/framework/Macros.h"
#include "tests/framework/datasets/Datasets.h"
#include "tests/validation/Validation.h"
#include "tests/validation/fixtures/GEMMReshapeRHSMatrixFixture.h"
Go to the source code of this file.
Namespaces | |
arm_compute | |
Copyright (c) 2017-2024 Arm Limited. | |
arm_compute::test | |
arm_compute::test::validation | |
Typedefs | |
template<typename T > | |
using | CLGEMMReshapeRHSMatrixFixture = GEMMReshapeRHSMatrixValidationFixture< CLTensor, CLAccessor, CLGEMMReshapeRHSMatrix, T > |
Functions | |
DATA_TEST_CASE (Validate, framework::DatasetMode::ALL, zip(zip(zip(zip(zip(framework::dataset::make("InputInfo", { TensorInfo(TensorShape(32U, 16U, 2U), 1, DataType::F32), TensorInfo(TensorShape(32U, 16U, 2U), 1, DataType::F32), TensorInfo(TensorShape(32U, 16U, 2U), 1, DataType::F32), TensorInfo(TensorShape(32U, 16U, 2U), 1, DataType::F32), TensorInfo(TensorShape(32U, 16U, 2U), 1, DataType::F32), TensorInfo(TensorShape(32U, 16U, 2U), 1, DataType::F32), TensorInfo(TensorShape(32U, 16U, 2U), 1, DataType::F32), TensorInfo(TensorShape(32U, 16U, 2U), 1, DataType::F32), }), framework::dataset::make("OutputInfo",{ TensorInfo(TensorShape(64U, 2U, 2U), 1, DataType::F32), TensorInfo(TensorShape(32U, 2U, 2U), 1, DataType::F16), TensorInfo(TensorShape(32U, 2U, 2U), 1, DataType::F32), TensorInfo(TensorShape(32U, 2U, 2U), 1, DataType::F32), TensorInfo(TensorShape(32U, 2U, 2U), 1, DataType::F32), TensorInfo(TensorShape(32U, 2U, 2U), 1, DataType::F32), TensorInfo(TensorShape(32U, 2U, 2U), 1, DataType::F32), TensorInfo(TensorShape(32U, 2U, 2U), 1, DataType::F32), })), framework::dataset::make("N0",{ 4, 0, 4, 4, 4, 17, 4, 4 })), framework::dataset::make("K0",{ 4, 4, 0, 4, 4, 4, 17, 1 })), framework::dataset::make("H0",{ 4, 4, 4, 0, 4, 4, 4, 4 })), framework::dataset::make("Expected", { false, false, false, false, false, false, false})), input_info, output_info, n0, k0, h0, expected) | |
DATA_TEST_CASE (ValidatePadding, framework::DatasetMode::ALL, combine(combine(combine(combine(framework::dataset::make("InputShape", { TensorShape(32U, 16U, 1U), TensorShape(32U, 16U, 2U) }), framework::dataset::make("N0",{ 4 })), framework::dataset::make("K0",{ 4, 8, 16 })), framework::dataset::make("H0",{ 1, 2, 4 })), framework::dataset::make("DataType",{ DataType::F32, DataType::F16 })), input_shape, n0, k0, h0, data_type) | |
FIXTURE_DATA_TEST_CASE (S32_NT, CLGEMMReshapeRHSMatrixFixture< int >, framework::DatasetMode::ALL, combine(combine(combine(combine(combine(combine(combine(datasets::SmallGEMMReshape2DShapes(), b_values), framework::dataset::make("DataType", DataType::S32)), n0_values_nt_s32), k0_values_nt_s32), h0_values), i_values), framework::dataset::make("transpose", false))) | |
FIXTURE_DATA_TEST_CASE (S32_T, CLGEMMReshapeRHSMatrixFixture< int >, framework::DatasetMode::ALL, combine(combine(combine(combine(combine(combine(combine(datasets::SmallGEMMReshape2DShapes(), b_values), framework::dataset::make("DataType", DataType::S32)), n0_values_t_s32), k0_values_t_s32), h0_values), i_values), framework::dataset::make("transpose", true))) | |
FIXTURE_DATA_TEST_CASE (S16_NT, CLGEMMReshapeRHSMatrixFixture< short >, framework::DatasetMode::ALL, combine(combine(combine(combine(combine(combine(combine(datasets::SmallGEMMReshape2DShapes(), b_values), framework::dataset::make("DataType", DataType::S16)), n0_values_nt_s16), k0_values_nt_s16), h0_values), i_values), framework::dataset::make("transpose", false))) | |
FIXTURE_DATA_TEST_CASE (S16_T, CLGEMMReshapeRHSMatrixFixture< short >, framework::DatasetMode::ALL, combine(combine(combine(combine(combine(combine(combine(datasets::SmallGEMMReshape2DShapes(), b_values), framework::dataset::make("DataType", DataType::S16)), n0_values_t_s16), k0_values_t_s16), h0_values), i_values), framework::dataset::make("transpose", true))) | |
FIXTURE_DATA_TEST_CASE (S8_NT, CLGEMMReshapeRHSMatrixFixture< char >, framework::DatasetMode::ALL, combine(combine(combine(combine(combine(combine(combine(datasets::SmallGEMMReshape2DShapes(), b_values), framework::dataset::make("DataType", DataType::S8)), n0_values_nt_s8), k0_values_nt_s8), h0_values), i_values), framework::dataset::make("transpose", false))) | |
FIXTURE_DATA_TEST_CASE (S8_T, CLGEMMReshapeRHSMatrixFixture< char >, framework::DatasetMode::ALL, combine(combine(combine(combine(combine(combine(combine(datasets::SmallGEMMReshape2DShapes(), b_values), framework::dataset::make("DataType", DataType::S8)), n0_values_t_s8), k0_values_t_s8), h0_values), i_values), framework::dataset::make("transpose", true))) | |