dspm_mult_ex_f32_ansi.c File Reference

#include "dspm_mult.h"

Include dependency graph for dspm_mult_ex_f32_ansi.c:

Go to the source code of this file.

Functions
esp_err_t	dspm_mult_ex_f32_ansi (const float *A, const float *B, float *C, int A_rows, int A_cols, int B_cols, int A_padding, int B_padding, int C_padding)
	Matrix subset multiplication.

Function Documentation

dspm_mult_ex_f32_ansi()

esp_err_t dspm_mult_ex_f32_ansi	(	const float *	A,
		const float *	B,
		float *	C,
		int	m,
		int	n,
		int	k,
		int	A_padd,
		int	B_padd,
		int	C_padd )

Matrix subset multiplication.

One or all of the matrices are matrix subsets, described with pointers and strides Matrix multiplication for two floating point matrices: C[m][k] = A[m][n] * B[n][k] The extension (_ansi) use ANSI C and could be compiled and run on any platform. The extension (_ae32) is optimized for ESP32 chip.

Parameters

[in]	A	input matrix A[m][n]
[in]	B	input matrix B[n][k]
[out]	C	result matrix C[m][k]
[in]	m	matrix dimension
[in]	n	matrix dimension
[in]	k	matrix dimension
[in]	A_padd	input matrix A padding
[in]	B_padd	input matrix B padding
[in]	C_padd	result matrix C padding

Returns

• ESP_OK on success
• One of the error codes from DSP library

Definition at line 12 of file dspm_mult_ex_f32_ansi.c.

{
    if (NULL == A) {
        return ESP_ERR_DSP_PARAM_OUTOFRANGE;
    }
    if (NULL == B) {
        return ESP_ERR_DSP_PARAM_OUTOFRANGE;
    }
    if (NULL == C) {
        return ESP_ERR_DSP_PARAM_OUTOFRANGE;
    }
 
    if (A_rows <= 0) {
        return ESP_ERR_DSP_PARAM_OUTOFRANGE;
    }
    if (A_cols <= 0) {
        return ESP_ERR_DSP_PARAM_OUTOFRANGE;
    }
    if (B_cols <= 0) {
        return ESP_ERR_DSP_PARAM_OUTOFRANGE;
    }
 
    if (A_padding < 0) {
        return ESP_ERR_DSP_PARAM_OUTOFRANGE;
    }
    if (B_padding < 0) {
        return ESP_ERR_DSP_PARAM_OUTOFRANGE;
    }
    if (C_padding < 0) {
        return ESP_ERR_DSP_PARAM_OUTOFRANGE;
    }
 
    const int A_step = A_cols + A_padding;
    const int B_step = B_cols + B_padding;
    const int C_step = B_cols + C_padding;
 
    for (int i = 0; i < A_rows; i++) {
        for (int j = 0; j < B_cols; j++) {
            C[i * C_step + j] = A[i * A_step] * B[j];
            for (int s = 1; s < A_cols; s++) {
                C[i * C_step + j] += A[i * A_step + s] * B[s * B_step + j];
            }
        }
    }
    return ESP_OK;
}

References A, B, C, ESP_ERR_DSP_PARAM_OUTOFRANGE, and ESP_OK.