dspm_mult_s16_ansi.c File Reference

#include "dsps_dotprod.h"
#include "dspm_mult.h"

Include dependency graph for dspm_mult_s16_ansi.c:

Go to the source code of this file.

Functions
esp_err_t	dspm_mult_s16_ansi (const int16_t *A, const int16_t *B, int16_t *C, int m, int n, int k, int shift)
	Matrix multiplication 16 bit signeg int.

Function Documentation

dspm_mult_s16_ansi()

esp_err_t dspm_mult_s16_ansi	(	const int16_t *	A,
		const int16_t *	B,
		int16_t *	C,
		int	m,
		int	n,
		int	k,
		int	shift )

Matrix multiplication 16 bit signeg int.

Matrix multiplication for two signed 16 bit fixed point matrices: C[m][k] = (A[m][n] * B[n][k]) >> (15- shift) 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]
	C	result matrix C[m][k]
[in]	m	matrix dimension
[in]	n	matrix dimension
[in]	k	matrix dimension
[in]	shift	every result will be shifted and stored as 16 bit signed value.

Returns

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

Definition at line 21 of file dspm_mult_s16_ansi.c.

{
    int final_shift = shift - 15;
    for (int i = 0 ; i < m ; i++) {
        for (int j = 0 ; j < k ; j++) {
            // This code also could be used
            //dsps_dotprode_f32_ae32(&A[i*n],&B[j],&C[i*k + j],n,1,n);
            long long acc = 0x7fff >> shift;
            for (int s = 0; s < n ; s++) {
                acc += (int32_t)A[i * n + s] * (int32_t)B[s * k + j];
            }
            if (final_shift > 0) {
                C[i * k + j] = (acc << final_shift);
            } else {
                C[i * k + j] = (acc >> (-final_shift));
            }
        }
    }
    return ESP_OK;
}

References A, B, C, ESP_OK, k, m, and n.