fix: Missing output type validation in CpuGemmDirectConv2d

gunes-arm · gunes-arm · commit 7b9dbd81dd60 · 2025-10-09T16:11:50.000+01:00
CpuGemmDirectConv2d should validate the output data type.

Partially Resolves: ARMCL-1199

Change-Id: Icdab0856027f60ad50f9a87a4c82144de9d248ef
Signed-off-by: Gunes Bayir &lt;gunes.bayir@arm.com&gt;
diff --git a/src/cpu/operators/CpuGemmDirectConv2d.cpp b/src/cpu/operators/CpuGemmDirectConv2d.cpp
@@ -202,6 +202,8 @@ Status CpuGemmDirectConv2d::validate(const ITensorInfo *src,
         ARM_COMPUTE_RETURN_ERROR_ON(biases->num_dimensions() > 1);
     }
 
+    ARM_COMPUTE_RETURN_ERROR_ON_MISMATCHING_DATA_TYPES(src, dst);
+
     cpu::AsmGemmInfo asm_info = init_assembly_metadata(info, false);
     ARM_COMPUTE_RETURN_ON_ERROR(cpu::CpuGemmAssemblyDispatch::validate(src, weights, biases, dst, asm_info));
     return Status{};

Original file line number	Diff line number	Diff line change
`@@ -202,6 +202,8 @@ Status CpuGemmDirectConv2d::validate(const ITensorInfo *src,`
`202`	`202`	`ARM_COMPUTE_RETURN_ERROR_ON(biases->num_dimensions() > 1);`
`203`	`203`	`}`
`204`	`204`
	`205`	`+ ARM_COMPUTE_RETURN_ERROR_ON_MISMATCHING_DATA_TYPES(src, dst);`
	`206`	`+`
`205`	`207`	`cpu::AsmGemmInfo asm_info = init_assembly_metadata(info, false);`
`206`	`208`	`ARM_COMPUTE_RETURN_ON_ERROR(cpu::CpuGemmAssemblyDispatch::validate(src, weights, biases, dst, asm_info));`
`207`	`209`	`return Status{};`