aboutsummaryrefslogtreecommitdiff
path: root/tests/validation/fixtures/GEMMFixture.h
diff options
context:
space:
mode:
authorAdnan AlSinan <adnan.alsinan@arm.com>2022-09-07 13:54:53 +0100
committerAdnan AlSinan <adnan.alsinan@arm.com>2022-09-12 13:12:38 +0000
commit26c9d1a787bccbe0b6c6749b80e2f5030395bda6 (patch)
treea3ec80da0862e616e1163ff92e8587c290848a2e /tests/validation/fixtures/GEMMFixture.h
parent4478e1cb2d7be9190147be597c3cfbf4c6f99f09 (diff)
downloadComputeLibrary-26c9d1a787bccbe0b6c6749b80e2f5030395bda6.tar.gz
Add test for NEGEMM to test a batched matrix multiplication with variable input tensors
- Add a test for CPU to batched matrix multiplication with variable input tensors - Disable assembly kernel when using _reshape_b_only_on_first_run flag Resolves COMPMID-5501 Signed-off-by: Adnan AlSinan <adnan.alsinan@arm.com> Change-Id: If96b182584617806a9dfe597dbfaf05241b123c2 Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/8234 Tested-by: Arm Jenkins <bsgcomp@arm.com> Reviewed-by: Gian Marco Iodice <gianmarco.iodice@arm.com> Comments-Addressed: Arm Jenkins <bsgcomp@arm.com> Benchmark: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'tests/validation/fixtures/GEMMFixture.h')
-rw-r--r--tests/validation/fixtures/GEMMFixture.h5
1 files changed, 2 insertions, 3 deletions
diff --git a/tests/validation/fixtures/GEMMFixture.h b/tests/validation/fixtures/GEMMFixture.h
index 0682337c82..5dc2711753 100644
--- a/tests/validation/fixtures/GEMMFixture.h
+++ b/tests/validation/fixtures/GEMMFixture.h
@@ -152,7 +152,6 @@ protected:
DataType data_type)
{
TensorShape shape_a_to_use = shape_a;
-
if(reinterpret_input_as_3d)
{
// Collapse the second and third dimension if the input is 3D
@@ -213,13 +212,13 @@ protected:
reference::gemm<T>((pretranspose_a) ? a_transposed : a, (pretranspose_b) ? b_transposed : b, c, alpha, disable_c ? 0.f : beta);
fill((pretranspose_a) ? a_transposed : a, 3);
fill((pretranspose_b) ? b_transposed : b, 4);
- fill(c , 5);
+ fill(c, 5);
}
// Setting beta to 0 will effectively disable C for the
// computation of the reference: alpha * A * B + 0 * C
// Use transposed tensors if boolean enabled else use original tensors
- auto r = reference::gemm<T>((pretranspose_a) ? a_transposed : a, (pretranspose_b) ? b_transposed : b, c, alpha, disable_c ? 0.f : beta);
+ auto r = reference::gemm<T>((pretranspose_a) ? a_transposed : a, (pretranspose_b) ? b_transposed : b, c, alpha, disable_c ? 0.f : beta);
return r;
}