sergiotarxz/nuttx
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

math/mpi: add mpi driver in math

Browse Source 
add interface for MPI(Multiple Precision Integer) registration in /math/mpi
Signed-off-by: makejian <makejian@xiaomi.com>
This commit is contained in:
makejian 2023-09-04 10:35:11 +08:00 committed by Xiang Xiao
parent b1c8c84e81
commit be72a6e26f
7 changed files with 217 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
drivers/math/CMakeLists.txt
View File

@ -18,6 +18,8 @@
#
# ##############################################################################
if(CONFIG_MATH_CORDIC OR CONFIG_MATH_FFT)
if(CONFIG_MATH_CORDIC
OR CONFIG_MATH_FFT
OR CONFIG_MATH_MPI)
target_sources(drivers PRIVATE math.c)
endif()

7
drivers/math/Kconfig
View File

@ -32,4 +32,11 @@ config MATH_FFT
This selection enables building of the "upper-half" FFT driver.
See include/nuttx/math/fft.h for further FFT driver information.
config MATH_MPI
bool "MATH MPI support"
default n
---help---
This selection enables building of the "upper-half" MPI driver.
See include/nuttx/math/mpi.h for further MPI driver information.
endmenu # MATH Acceleration Information

2
drivers/math/Make.defs
View File

@ -20,7 +20,7 @@
# Include math acceleration support
ifneq ($(CONFIG_MATH_CORDIC)$(CONFIG_MATH_FFT),)
ifneq ($(CONFIG_MATH_CORDIC)$(CONFIG_MATH_FFT)$(CONFIG_MATH_MPI),)
CSRCS += math.c
endif

36
drivers/math/math.c
View File

@ -156,6 +156,21 @@ static int math_ioctl(FAR struct file *filep, int cmd, unsigned long arg)
break;
}
#endif
#ifdef CONFIG_MATH_MPI
case MATHIOC_MPI_CALC:
{
FAR struct mpi_calc_s *calc =
(FAR struct mpi_calc_s *)((uintptr_t)arg);
if (upper->mpi != NULL)
{
ret = upper->mpi->ops->calc(upper->mpi, calc);
}
break;
}
#endif
}
leave_critical_section(flags);
@ -252,3 +267,24 @@ int fft_register(FAR const char *path,
return math_register(path, &config);
}
#endif
/****************************************************************************
* Name: mpi_register
*
* Description:
* Register a MPI driver.
*
****************************************************************************/
#ifdef CONFIG_MATH_MPI
int mpi_register(FAR const char *path,
FAR struct mpi_lowerhalf_s *lower)
{
struct math_config_s config;
memset(&config, 0, sizeof(config));
config.mpi = lower;
return math_register(path, &config);
}
#endif

6
include/nuttx/math/math.h
View File

@ -31,6 +31,9 @@
#ifdef CONFIG_MATH_FFT
#include <nuttx/math/fft.h>
#endif
#ifdef CONFIG_MATH_MPI
#include <nuttx/math/mpi.h>
#endif
/****************************************************************************
* Public Types
@ -49,6 +52,9 @@ struct math_config_s
#ifdef CONFIG_MATH_FFT
FAR struct fft_lowerhalf_s *fft;
#endif
#ifdef CONFIG_MATH_MPI
FAR struct mpi_lowerhalf_s *mpi;
#endif
};
/****************************************************************************

1
include/nuttx/math/math_ioctl.h
View File

@ -40,5 +40,6 @@
#define MATHIOC_CORDIC_CALC _MATHIOC(1)
#define MATHIOC_FFT_CALC _MATHIOC(2)
#define MATHIOC_MPI_CALC _MATHIOC(3)
#endif /* __INCLUDE_NUTTX_MATH_MATH_IOCTL_H */

163
include/nuttx/math/mpi.h Normal file
View File

@ -0,0 +1,163 @@
/****************************************************************************
* include/nuttx/math/mpi.h
*
* Licensed to the Apache Software Foundation (ASF) under one or more
* contributor license agreements. See the NOTICE file distributed with
* this work for additional information regarding copyright ownership. The
* ASF licenses this file to you 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
*
* http://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.
*
****************************************************************************/
#ifndef __INCLUDE_NUTTX_MATH_MPI_H
#define __INCLUDE_NUTTX_MATH_MPI_H
/****************************************************************************
* Included Files
****************************************************************************/
#include <stdint.h>
#include <sys/types.h>
/****************************************************************************
* Pre-processor Definitions
****************************************************************************/
#define MPI_MAXPARAM 4
/****************************************************************************
* Public Types
****************************************************************************/
enum mpi_calc_func_e
{
MPI_CALC_FUNC_INVAL = 0,
MPI_CALC_FUNC_ADD = 1, /* X = A + B
* first argument: A,
* second argument: B,
* third argument: X
*/
MPI_CALC_FUNC_SUB = 2, /* X = A - B
* first argument: A,
* second argument: B,
* third argument: X
*/
MPI_CALC_FUNC_MUL = 3, /* X = A * B,
* first argument: A,
* second argument: B,
* third argument: X
*/
MPI_CALC_FUNC_DIV = 4, /* A = Q * B + R
* first argument: A,
* second argument: B,
* third argument: Q,
* fourth argument: R
*/
MPI_CALC_FUNC_MOD = 5, /* R = A mod B
* first argument: A,
* second argument: B,
* third argument: R,
*/
MPI_CALC_FUNC_EXP_MOD = 6, /* X = A ^ E mod N
* first argument: A,
* second argument: E,
* third argument: N,
* fourth argument: X,
*/
MPI_CALC_FUNC_INV_MOD = 7, /* X = A ^ -1 mod N
* first argument: A,
* second argument: N,
* third argument: X,
*/
MPI_CALC_FUNC_GCD = 8, /* G = gcd(A, B)
* first argument: A,
* second argument: B,
* third argument: G,
*/
MPI_CALC_FUNC_LAST
};
/* MPI structure
* s: -1 if the mpi is negative, 1 otherwise.
* n: total number of bits in p.
* p: pointer to mpi.
*/
struct mpiparam
{
int s;
size_t n;
FAR uint8_t *p;
};
struct mpi_calc_s
{
uint16_t op;
struct mpiparam param[MPI_MAXPARAM];
};
/* This structure provides the "lower-half" driver operations available to
* the "upper-half" driver.
*/
struct mpi_lowerhalf_s;
struct mpi_ops_s
{
/* MPI calculate request */
CODE int (*calc)(FAR struct mpi_lowerhalf_s *lower,
FAR struct mpi_calc_s *calc);
};
/* This structure provides the publicly visible representation of the
* "lower-half" driver state structure. "lower half" drivers will have an
* internal structure definition that will be cast-compatible with this
* structure definitions.
*/
struct mpi_lowerhalf_s
{
/* Publicly visible portion of the "lower-half" driver state structure. */
FAR const struct mpi_ops_s *ops; /* Lower half operations */
};
/****************************************************************************
* Public Function Prototypes
****************************************************************************/
#undef EXTERN
#if defined(__cplusplus)
#define EXTERN extern "C"
extern "C"
{
#else
#define EXTERN extern
#endif
/****************************************************************************
* Name: mpi_register
*
* Description:
* Register a MPI driver.
*
****************************************************************************/
int mpi_register(FAR const char *path,
FAR struct mpi_lowerhalf_s *lower);
#undef EXTERN
#if defined(__cplusplus)
}
#endif
#endif /* __INCLUDE_NUTTX_MATH_MPI_H */