riscv_matrix_example_f32.c

/* ----------------------------------------------------------------------
* Copyright (C) 2010-2012 ARM Limited. All rights reserved.
* Copyright (c) 2019 Nuclei Limited. All rights reserved.
*
* $Date:         17. January 2013
* $Revision:     V1.4.0
*
* Project:       NMSIS DSP Library
* Title:         riscv_matrix_example_f32.c
*
* Description:   Example code demonstrating least square fit to data
*                using matrix functions
*
* Target Processor: RISC-V Cores
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions
* are met:
*   - Redistributions of source code must retain the above copyright
*     notice, this list of conditions and the following disclaimer.
*   - Redistributions in binary form must reproduce the above copyright
*     notice, this list of conditions and the following disclaimer in
*     the documentation and/or other materials provided with the
*     distribution.
*   - Neither the name of ARM LIMITED nor the names of its contributors
*     may be used to endorse or promote products derived from this
*     software without specific prior written permission.
*
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
* "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
* LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
* FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
* COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
* INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
* BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
* LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
* CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
* LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
* ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
* POSSIBILITY OF SUCH DAMAGE.
 * -------------------------------------------------------------------- */
 
#include "riscv_math.h"
#include "math_helper.h"
#include <stdio.h>
 
#define SNR_THRESHOLD   77
 
/* --------------------------------------------------------------------------------
* Test input data(Cycles) taken from FIR Q15 module for differant cases of blockSize
* and tapSize
* --------------------------------------------------------------------------------- */
 
const float32_t B_f32[4] =
{
  782.0, 7577.0, 470.0, 4505.0
};
 
/* --------------------------------------------------------------------------------
* Formula to fit is  C1 + C2 * numTaps + C3 * blockSize + C4 * numTaps * blockSize
* -------------------------------------------------------------------------------- */
 
const float32_t A_f32[16] =
{
  /* Const,   numTaps,   blockSize,   numTaps*blockSize */
  1.0,     32.0,      4.0,     128.0,
  1.0,     32.0,     64.0,    2048.0,
  1.0,     16.0,      4.0,      64.0,
  1.0,     16.0,     64.0,    1024.0,
};
 
 
/* ----------------------------------------------------------------------
* Temporary buffers  for storing intermediate values
* ------------------------------------------------------------------- */
/* Transpose of A Buffer */
float32_t AT_f32[16];
/* (Transpose of A * A) Buffer */
float32_t ATMA_f32[16];
/* Inverse(Transpose of A * A)  Buffer */
float32_t ATMAI_f32[16];
/* Test Output Buffer */
float32_t X_f32[4];
 
/* ----------------------------------------------------------------------
* Reference ouput buffer C1, C2, C3 and C4 taken from MATLAB
* ------------------------------------------------------------------- */
const float32_t xRef_f32[4] = {73.0, 8.0, 21.25, 2.875};
 
float32_t snr;
 
 
/* ----------------------------------------------------------------------
* Max magnitude FFT Bin test
* ------------------------------------------------------------------- */
 
int32_t main(void)
{
/*  Enable Vector  */
#if (defined(__riscv_vector))
  __RV_CSR_SET(CSR_MSTATUS, 0x200);
#endif
 
  riscv_matrix_instance_f32 A;      /* Matrix A Instance */
  riscv_matrix_instance_f32 AT;     /* Matrix AT(A transpose) instance */
  riscv_matrix_instance_f32 ATMA;   /* Matrix ATMA( AT multiply with A) instance */
  riscv_matrix_instance_f32 ATMAI;  /* Matrix ATMAI(Inverse of ATMA) instance */
  riscv_matrix_instance_f32 B;      /* Matrix B instance */
  riscv_matrix_instance_f32 X;      /* Matrix X(Unknown Matrix) instance */
 
  uint32_t srcRows, srcColumns;  /* Temporary variables */
  riscv_status status;
 
  /* Initialise A Matrix Instance with numRows, numCols and data array(A_f32) */
  srcRows = 4;
  srcColumns = 4;
  riscv_mat_init_f32(&A, srcRows, srcColumns, (float32_t *)A_f32);
 
  /* Initialise Matrix Instance AT with numRows, numCols and data array(AT_f32) */
  srcRows = 4;
  srcColumns = 4;
  riscv_mat_init_f32(&AT, srcRows, srcColumns, AT_f32);
 
  /* calculation of A transpose */
  status = riscv_mat_trans_f32(&A, &AT);
 
 
  /* Initialise ATMA Matrix Instance with numRows, numCols and data array(ATMA_f32) */
  srcRows = 4;
  srcColumns = 4;
  riscv_mat_init_f32(&ATMA, srcRows, srcColumns, ATMA_f32);
 
  /* calculation of AT Multiply with A */
  status = riscv_mat_mult_f32(&AT, &A, &ATMA);
 
  /* Initialise ATMAI Matrix Instance with numRows, numCols and data array(ATMAI_f32) */
  srcRows = 4;
  srcColumns = 4;
  riscv_mat_init_f32(&ATMAI, srcRows, srcColumns, ATMAI_f32);
 
  /* calculation of Inverse((Transpose(A) * A) */
  status = riscv_mat_inverse_f32(&ATMA, &ATMAI);
 
  /* calculation of (Inverse((Transpose(A) * A)) *  Transpose(A)) */
  status = riscv_mat_mult_f32(&ATMAI, &AT, &ATMA);
 
  /* Initialise B Matrix Instance with numRows, numCols and data array(B_f32) */
  srcRows = 4;
  srcColumns = 1;
  riscv_mat_init_f32(&B, srcRows, srcColumns, (float32_t *)B_f32);
 
  /* Initialise X Matrix Instance with numRows, numCols and data array(X_f32) */
  srcRows = 4;
  srcColumns = 1;
  riscv_mat_init_f32(&X, srcRows, srcColumns, X_f32);
 
  /* calculation ((Inverse((Transpose(A) * A)) *  Transpose(A)) * B) */
  status = riscv_mat_mult_f32(&ATMA, &B, &X);
 
  /* Comparison of reference with test output */
  snr = riscv_snr_f32((float32_t *)xRef_f32, X_f32, 4);
 
  /*------------------------------------------------------------------------------
  *            Initialise status depending on SNR calculations
  *------------------------------------------------------------------------------*/
  if ( snr > SNR_THRESHOLD)
  {
    status = RISCV_MATH_SUCCESS;
    printf("passed\n");
    return 0;
  }
  else
  {
    status = RISCV_MATH_TEST_FAILURE;
    printf("failed\n");
    return 1;
  }
}