1
0
mirror of https://github.com/opencv/opencv.git synced 2026-07-29 15:23:05 +04:00

Fixed OpenCL PF16 fallback in Einsum layer.

This commit is contained in:
Alexander Smorkalov
2023-09-29 15:52:23 +03:00
parent c7ec0d599a
commit 5caee5cc64
+9 -1
View File
@@ -38,7 +38,6 @@ Mat batchwiseMatMul(
const Mat& input2,
const MatShape& input2ShapeOverride)
{
// Sanity checks before the actual MatMul
//input_1.DataType() == input_2.DataType(), "Data types of the inputs must match for MatMul");
@@ -391,6 +390,15 @@ public:
OutputArrayOfArrays outputs_arr,
OutputArrayOfArrays internals_arr) CV_OVERRIDE
{
CV_TRACE_FUNCTION();
CV_TRACE_ARG_VALUE(name, "name", name.c_str());
if (inputs_arr.depth() == CV_16S)
{
forward_fallback(inputs_arr, outputs_arr, internals_arr);
return;
}
// homogenize inputs
preProcessInputs(inputs_arr);