/github/workspace/src/MatrixFunctionsStride/mat_mult_cmplx_stride/kernels/plp_mat_mult_cmplx_stride_q32p_xpulpv2.c
Functions
Name | |
---|---|
void | plp_mat_mult_cmplx_stride_q32p_xpulpv2(void * args) parallel strided matrix matrix multiplication for complex 32-bit fix-point on XpulpV2 |
Defines
Name | |
---|---|
BASIC_VERSION |
Functions Documentation
function plp_mat_mult_cmplx_stride_q32p_xpulpv2
void plp_mat_mult_cmplx_stride_q32p_xpulpv2(
void * args
)
parallel strided matrix matrix multiplication for complex 32-bit fix-point on XpulpV2
Parameters:
- args pointer to plp_mat_mat_mult_cmplx_instance_q32 struct initialized by plp_mat_mult_cmplx_stride_q32_parallel
Return: none
Par: Fix-Point
Fix-Point and Shifting The result will be shifted by the parameter shift
to the right (which corresponds to a multiplication by 2^-shift
). Assume that matrix A is represente as pSrcA * 2^-x
and matrix B as pSrcB * 2^-y
(which means that A has x
, and B has y
bits after the binary point). Then, the output matrix C is represented as pDstC * 2^-(x + y - shift)
. The output matrix is also stored with the same number of bits as the inputs. Set the shift
parameter such that no overflow occurrs.
Macros Documentation
define BASIC_VERSION
#define BASIC_VERSION
Source code
/* =====================================================================
* Project: PULP DSP Library
* Title: plp_mat_mult_cmplx_stride_q32p_xpulpv2.c
* Description: parallel 32-bit fix-point complex strided matrix matrix multiplication for XPULPV2
*
* $Date: complex strided matrix matrix multiplication
* $Revision: V0
*
* Target Processor: PULP cores
* ===================================================================== */
/*
* Copyright (C) 2020 ETH Zurich and Ubiversity of Bologna. All rights reserved.
*
* Author: Tibor Schneider, ETH Zurich
*
* SPDX-License-Identifier: Apache-2.0
*
* Licensed under the Apache License, Version 2.0 (the License); you may
* not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an AS IS BASIS, WITHOUT
* WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#include "plp_math.h"
void plp_mat_mult_cmplx_stride_q32p_xpulpv2(void *args) {
int core_id = hal_core_id();
plp_mat_mult_cmplx_stride_instance_q32 *a = (plp_mat_mult_cmplx_stride_instance_q32 *)args;
const int32_t *__restrict__ pSrcA = a->pSrcA;
const int32_t *__restrict__ pSrcB = a->pSrcB;
uint32_t M = a->M;
uint32_t N = a->N;
uint32_t O = a->O;
uint32_t strideA = a->strideA;
uint32_t strideB = a->strideB;
uint32_t strideC = a->strideC;
uint32_t shift = a->shift;
uint32_t nPE = a->nPE;
int32_t *__restrict__ pDstC = a->pDstC;
#define BASIC_VERSION // if used don't forget to also use the undefine at end of file
#ifdef BASIC_VERSION
for (int m = core_id; m < M; m += nPE) {
for (int o = 0; o < O; o++) {
int32_t sum_re = 0;
int32_t sum_im = 0;
for (int n = 0; n < N; n++) {
int32_t a_re = (int32_t)pSrcA[(m * strideA + n) * 2 + 0];
int32_t a_im = (int32_t)pSrcA[(m * strideA + n) * 2 + 1];
int32_t b_re = (int32_t)pSrcB[(n * strideB + o) * 2 + 0];
int32_t b_im = (int32_t)pSrcB[(n * strideB + o) * 2 + 1];
sum_re += __ROUNDNORM_REG(a_re * b_re - a_im * b_im, shift);
sum_im += __ROUNDNORM_REG(a_re * b_im + a_im * b_re, shift);
}
pDstC[(m * strideC + o) * 2 + 0] = (int32_t)sum_re;
pDstC[(m * strideC + o) * 2 + 1] = (int32_t)sum_im;
}
}
#else
// TODO: Hackathon
#endif
#undef BASIC_VERSION
}
Updated on 2023-03-01 at 16:16:33 +0000