27 #include "tests/datasets/LargeMatMulDataset.h"
28 #include "tests/datasets/SmallMatMulDataset.h"
32 #include "tests/validation/fixtures/MatMulKernelFixture.h"
48 constexpr
float abs_tolerance_f16(
87 using MatMulConfigurationPair = std::pair<MatMulKernelInfo, bool>;
89 const std::vector<MatMulConfigurationPair> supported_block_sizes =
165 for(
auto &pair : supported_block_sizes)
170 if(!pair.first.export_rhs_to_cl_image || export_to_cl_image_supported)
182 constexpr
size_t pixel_size = 4;
186 using ShapeConfigurationTuple = std::tuple<TensorShape, TensorShape, bool, bool, bool>;
187 const std::vector<ShapeConfigurationTuple> shape_configurations =
210 for(
auto &tuple : shape_configurations)
218 const bool adj_lhs = std::get<2>(tuple);
219 const bool adj_rhs = std::get<3>(tuple);
224 adj_lhs, adj_rhs, 4, 4, 4,
true
230 const bool expected = std::get<4>(tuple);
239 using ShapeConfigurationTuple = std::tuple<TensorShape, TensorShape, TensorShape, bool>;
240 const std::vector<ShapeConfigurationTuple> shape_configurations =
247 {
TensorShape(5
U, 4
U, 3
U, 4
U, 5
U, 6
U),
TensorShape(2
U, 5
U, 3
U, 4
U, 5
U, 6
U),
TensorShape(2
U),
true },
248 {
TensorShape(5
U, 4
U, 3
U, 4
U, 5
U, 1
U),
TensorShape(2
U, 5
U, 3
U, 4
U, 5
U, 6
U),
TensorShape(2
U),
false },
249 {
TensorShape(5
U, 4
U, 3
U, 4
U, 9
U, 6
U),
TensorShape(2
U, 5
U, 3
U, 4
U, 5
U, 6
U),
TensorShape(2
U),
false },
255 for(
auto &tuple : shape_configurations)
257 const bool expected = std::get<3>(tuple);
300 using DataTypeConfigurationTuple = std::tuple<DataType, DataType, DataType, bool>;
301 const std::vector<DataTypeConfigurationTuple> data_type_configurations =
325 for(
auto &tuple : data_type_configurations)
327 const bool expected = std::get<3>(tuple);
344 framework::dataset::
make("TransposeA", {
false,
true })),
448 framework::dataset::
make("TransposeA", {
true,
false })),
457 if(_device_supports_export_to_cl_image)
473 if(_device_supports_export_to_cl_image)
489 if(_device_supports_export_to_cl_image)
505 if(_device_supports_export_to_cl_image)
516 framework::dataset::
make("TransposeA", {
false,
true })),
581 framework::dataset::
make("TransposeA", {
true,
false })),
590 if(_device_supports_export_to_cl_image)
606 if(_device_supports_export_to_cl_image)
622 if(_device_supports_export_to_cl_image)
638 if(_device_supports_export_to_cl_image)