你的浏览器版本过低,可能导致网站不能正常访问!
为了你能正常使用网站功能,请使用这些浏览器。

详细讲解FFT浮点计算能力及速度对比

[复制链接]
gaosmile 发布时间:2020-3-4 17:13
今天测试的主要对象是NUCLEO-G431RB 的浮点计算能力,通过进行FFT侧面验证Nucleo-G431RB的计算能力。

% \: v/ z" Y" h) i. I! j

4 l" g: s+ z3 K9 z5 A9 l; J2 u
一、背景知识

! O8 e4 R4 {* g: _
FFT,是快速傅里叶变换(Fast Fourier Transform)的简称,即利用计算机计算离散傅里叶变换(DFT) 的高效、快速计算方法的统称;快速傅里叶变换是1965年由J.W.库利和T.W.图基提出的。采用这种算法能使计算机计算离散傅里叶变换所需要的乘法次数大为减少,特别是被变换的抽样点数N越多,FFT算法计算量的节省就越显著。$ Y$ `9 s" a4 Z) g
9 A7 K' l( V1 ]* R$ f* |6 t
微信图片_20200304170554.png

. U0 d% T& Z: r& S& d; N
上图是有限长序列可以通过离散傅里叶变换(DFT)将其频域也离散化成有限长序列。但其计算量太大,很难实时地处理问题,因此引出了快速傅里叶变换(FFT)。1965年,Cooley和Tukey提出了计算离散傅里叶变换(DFT)的快速算法,将DFT的运算量减少了几个数量级。数字信号处理这门新兴学科也随FFT的出现和发展而迅速发展。根据对序列分解与选取方法的不同而产生了FFT的多种算法,基本算法是基2DIT和基2DIF。FFT在离散傅里叶反变换、线性卷积和线性相关等方面也有重要应用。
/ Z: S/ o* G; N0 P. P

! P7 N. d( C, m  H1 \2 e% |
二、工程建立与代码移植
/ L1 ?: _+ A" ], }8 g2 w- Q/ o
ST为此提供了一套静态库,分别为Keil、IAR、GCC这三个开发环境提供相关的数学运算库。; l7 Q4 t2 h1 J$ O8 K
微信图片_20200304170601.png
% u# P: k0 _! W1 W5 \
引用头文件位置在:G4固件库STM32Cube_FW_G4_Vx.x.x)\Drviers\CMSIS\DSP\Include 目录中;! _' [6 S, [2 Y- _: t& E0 x1 x2 W
静态库位置在:G4固件库(STM32Cube_FW_G4_Vx.x.x)\Drviers\CMSIS\DSP\Lib)目录中,根据开发环境选择不同的库文件,包括大小端系统环境选择,是否浮点的库。
微信图片_20200304170608.png

/ C% F) ?" A' e6 \3 C5 T3 s9 t# Y" D
其中arm_cortexM4b_math.lib代表设备为大端模式;其中arm_cortexM4bf_math.lib代表设备为浮点大端模式;其中arm_cortexM4l_math.lib代表设备为小端模式;其中arm_cortexM4lf_math.lib代表设备为浮点小端模式;
微信图片_20200304170614.png
! k1 }; z/ `/ n* W
其中libarm_cortexM4l_math.a 代表设备为小端模式;其中libarm_cortexM4lf_math.a 代表设备为浮点小端模式;
微信图片_20200304170621.png
% F4 y/ h& ^. c) x. }$ o3 s) f* P
其中iar_cortexM4b_math.a代表设备为大端模式;其中iar_cortexM4bf_math.a代表设备为浮点大端模式;其中iar_cortexM4l_math.a代表设备为小端模式;其中iar_cortexM4lf_math.a代表设备为浮点小端模式; 此次体验期间适合STM32G431RB的免费的开发环境为STM32CubeIDE,该工具是在Eclipse 环境下使用的编译器为GCC。因此,适合搭建FFT测试内容是带浮点的库文件为:libarm_cortexM4lf_math.a
微信图片_20200304170627.png
在STM32CubeIDE新建一个工程FFT_G431RB,打开FFT_G431RB.ioc后在CubeMX中设置主频170MHz,添加lpuart1设备使能,设置频率为115200,lpuart1用于连接STLINK的VCP作为计算结果输出。项目在CubeMX中构建好后生成相关代码。在项目中创建DSP\Include和Lib\GCC目录,将开发包文件中的.h头文件和.a静态库文件移植到项目中,并设置相关编译环境如下图:
微信图片_20200304170634.png
添加编译静态库文件位置
微信图片_20200304170641.png
添加文件路径
* b5 V5 E7 B% c/ x! ~6 j
微信图片_20200304170647.png
添加编译宏定义
在Main.c文件头部添加以下定义代码:
/* USER CODE BEGIN Includes */
#include <stdio.h>
#include <stdlib.h>
#include <inttypes.h>
#include "arm_math.h"
#include "arm_const_structs.h"
/* USER CODE END Includes */
……
/* USER CODE BEGIN PTD */
typedef unsigned int ee_u32;
typedef clock_t CORE_TICKS;
typedef ee_u32 secs_ret;
/* USER CODE END PTD */
……
4 h/ u: A# U1 K; s
* g0 V! ~5 i( {

% _& U7 J3 k7 A# ]% t" D7 G

( F% x) m3 e9 O5 h+ q" @' L
/* USER CODE BEGIN PD */
#define EE_TICKS_PER_SEC 1000
//#define NPT 2048//4096
#define SysTick_Counter_Disable ((uint32_t)0xFFFFFFFE)
#define SysTick_Counter_Enable ((uint32_t)0x00000001)
#define SysTick_Counter_Clear ((uint32_t)0x00000000)
#define FFT_LENGTH        1024         //FFT长度,默认是1024点FFT
. x& h( j9 U' X3 G: f8 w" f" M! u
volatile __IO uint32_t Tick;
uint32_t ifftFlag = 0;
uint32_t doBitReverse = 1;
6 [4 x- ?% h% C0 u7 q# I3 {
float32_t fft_x[FFT_LENGTH * 2];
float32_t fft_input[FFT_LENGTH * 2];
float32_t fft_output[FFT_LENGTH]; //NPT/2
1 Y+ f4 X4 t' U: S, P: Q5 \
CORE_TICKS total_time;

0 e# W2 T, x# q) f2 M) x: `
/* USER CODE END PD */
……
' x! \9 c; [" ~( ?
/* USER CODE BEGIN PFP */
void fft_test();
void start_time(void);
void stop_time(void);
CORE_TICKS get_time(void);
secs_ret time_in_secs(CORE_TICKS ticks);
secs_ret time_in_subsecs(CORE_TICKS ticks);
/* USER CODE END PFP */
……
" M/ `: O! F2 h' P4 v0 `' B

7 o$ \8 E/ T2 m
在main()函数中增加以下代码
……
  /* USER CODE BEGIN 2 */
  printf("NucleoG431RB Test FFT Speed!\n");
  printf("FFT Point Number:%d\n",FFT_LENGTH);
  printf("CLKFreq:%d\n", HAL_RCC_GetHCLKFreq());
  fft_test();
  /* USER CODE END 2 */
……

. m8 r# o/ e+ X% w9 L
添加相关函数定义
4 m9 m, h$ a; k9 }1 D$ x$ U

3 D* T6 c9 T- @2 t( r

1 S% H4 R' G  \, [$ Y  K  k
……
/* USER CODE BEGIN 4 */
void start_time(void) {
  //GETMYTIME(&start_time_val );
  Tick = 0;
  SysTick_Config(HAL_RCC_GetHCLKFreq() / 1000);
}

4 I6 [! v; F% @9 d+ p- W3 a
void stop_time(void) {
  //GETMYTIME(&stop_time_val );
  /* Stop the Timer and get the encoding time */
  SysTick->CTRL &= SysTick_Counter_Disable;
  /* Clear the SysTick Counter */
  SysTick->VAL = SysTick_Counter_Clear;
}

9 Q" F) k  ~& i' o% f' A
CORE_TICKS get_time(void) {
  //CORE_TICKS elapsed = (CORE_TICKS)(MYTIMEDIFF(stop_time_val, start_time_val));
  CORE_TICKS elapsed = (CORE_TICKS) Tick;
  return elapsed;
}

( f2 c% Y# r2 h& A9 M
secs_ret time_in_secs(CORE_TICKS ticks) {
  secs_ret retval = ((secs_ret) ticks) / (secs_ret) EE_TICKS_PER_SEC;
  return retval;
}
secs_ret time_in_subsecs(CORE_TICKS ticks) {
  secs_ret retval = ((secs_ret) ticks);
  return retval;
}
……
FFT主函数

& i1 P, F3 a7 V: a
……
void fft_test() {
  int i, j = 0;
5 c$ m: {2 }$ _9 l! m6 V2 ^# Y
  int Adc = 3; //直流分量幅度
  float32_t A1 = 7.0; //频率F1 信号的幅度
  float32_t A2 = 1.5; //频率F2信号的幅度
  float32_t A3 = 5.1; //频率F3信号的幅度
  float32_t F1 = 180.0; //信号1频率(Hz)
  float32_t F2 = 390.0; //信号2频率(Hz)
  float32_t F3 = 600.0; //信号3频率(Hz)
  float32_t P1 = 0.0;     //信号1相位(度)
  float32_t P2 = 0.0;     //信号2相位(度)
  float32_t P3 = 0.0;     //信号3相位(度)
  //float32_t Fs = (float32_t)FFT_LENGTH;  //采样频率(Hz)
  //float32_t N  = (float32_t)FFT_LENGTH*2;  //采样点数
  //生成信号序列
  for (i = 0; i < FFT_LENGTH; i++) {
    //采样时刻 i/Fs
    fft_x[2 * i] = 10
        + A1 * arm_sin_f32(2 * PI * i * F1 / FFT_LENGTH + PI * P1 / 180)
        + A2 * arm_sin_f32(2 * PI * i * F2 / FFT_LENGTH + PI * P2 / 180)
        + A3 * arm_sin_f32(2 * PI * i * F3 / FFT_LENGTH + PI * P3 / 180);
    //虚部全部为0
    fft_x[2 * i + 1] = 0;
  }
  start_time();
  for (j = 0; j < 1000; j++) {
    //arm_cfft_sR_f32_len1024,该变量即为"arm_const_structs.h"提供的配置变量,包含头文件后,直接调用即可。
    //memset(fft_output, 0x00, sizeof(float32_t)*FFT_LENGTH);
    memcpy(fft_input, fft_x, sizeof(float32_t)*FFT_LENGTH * 2);
    arm_cfft_f32(&arm_cfft_sR_f32_len1024, fft_input, ifftFlag,  doBitReverse);
    //把运算结果复数求模得幅值
    arm_cmplx_mag_f32(fft_input, fft_output, FFT_LENGTH);
  }
  stop_time();
  total_time = get_time();
  printf("Total time (sub_secs): %d\n", time_in_subsecs(total_time));
  for (i = 0; i < FFT_LENGTH; i++) {
    printf("%f\r\n", i, fft_output);
  }
}……
4 q2 U7 [9 m$ H( Y# m* H! P" _
系统printf输出串口定义
: K$ m# @' i" d4 a1 P
#ifdef __GNUC__
/* With GCC/RAISONANCE, small printf (option LD Linker->Libraries->Small printf
* set to 'Yes') calls __io_putchar() */
#define PUTCHAR_PROTOTYPE int __io_putchar(int ch)
#else
#define PUTCHAR_PROTOTYPE int fputc(int ch, FILE *f)
#endif /* __GNUC__ */

+ U, w' `9 n' T+ ~# U1 `2 I. L
/**
* @brief Retargets the C library printf function to the USART.
* @param None
* @retval None
*/
PUTCHAR_PROTOTYPE {
  /* Place your implementation of fputc here */
  /* e.g. write a character to the EVAL_COM1 and Loop until the end of transmission */
  HAL_UART_Transmit(&hlpuart1, (uint8_t*) &ch, 1, 0xFFFF);
  return ch;
}

* K& @3 E9 h/ }) w1 ?) S
/* USER CODE END 4 */
▲ 左右移动,查看完整代码
7 D4 y. n9 w! D: Y: g
编译后下载即可实现FFT相关计算。
, L2 \' G# y. m0 T4 Z0 Q6 g' {
三、FFT计算能力对比及数值校验

4 `9 ^) b- K- c; |% }7 D1 g
因计算单次FFT速度过快,Tick时钟无法记录,改用计算1000次的1024个点的FFT计算汇总时间,以便对比计算速度。; Q8 Y" A: e% d- W8 I9 g# f' t! L
同样的代码移植到Nucleo-F446ZE的项目中,这里有一个小插曲,本想用Nucleo-F301R8的板子对比速度,但因F301R8的Flash、RAM比较小,无法支持带DSP静态库的应用代码,改用速度相近的F4板子Nucleo-F446进行功能验证和对比,以下是对比结果:: I2 U. l+ k7 A9 H2 {" Z
微信图片_20200304170654.jpg

7 u, H. T" Y" R0 @
这两个个有FFT代码的Nucleo板输出的数值几乎一样,差别只在小数点后第4位、5位几乎可以忽略。其余的内容可以看出STM32G431RB在170MHz主频下浮点处理能力略低于180MHz主频下的STM32F446。不过G431RB具备数学运算加速器,在个别计算领域上可以向其靠拢。不过这也比F3系列的芯片有了很大的提高。$ ~+ l) j5 e. d. ^5 w
数值图形展示及验证:
微信图片_20200304170700.png

; Z" d, {5 [, T) P
将输出的数据通过Excel绘图的方式展现全频域信息,符合预期。: Z7 @. n8 j% v1 q; {  e, P
相关频点和幅值计算的结果如下图:

1 e# d' l/ A; j0 H
微信图片_20200304170710.png
0频率,约10。 微信图片_20200304170717.png : v+ U: J4 f$ L/ p5 `3 y
180频率,约7。 微信图片_20200304170724.png
- d& D" ^9 M; ], ?6 Z) D
390频率,约1.5。 微信图片_20200304170730.png
% t% A4 G0 G* Q: H
600频率,约5.1。
与代码中生成信号序列的公式相关参数相近。# L( U4 g' z- \, O* U
FFT计算及验证正确,单次FFT计算速度接近1毫秒,可用于DSP某些领域的数据处理。另外,相关数值计算还可通过上位机的Matelab、Python等工具进行计算并展示相关图形信息,用于验证FFT的处理结果对比。
& Q' }/ J8 v: p$ K: h
1583388085(1).png

点评

如果我用m3的stm103跑这个是不是主频不够,跑起来更不精确啊  发表于 2024-6-23 23:15
收藏 2 评论9 发布时间:2020-3-4 17:13

举报

9个回答
cyk128 回答时间:2020-3-4 17:26:20
很有用的分享
网络孤客 回答时间:2020-3-5 10:22:21
如果代码格式稍稍调整会更好些。
gaosmile 回答时间:2020-3-5 15:12:00
cyk128 发表于 2020-3-4 17:26: r% t1 K( [: W2 F) m' c4 n9 k+ S
很有用的分享
& S1 ^8 U, D( x( W9 c! Q
已修改
gaosmile 回答时间:2020-3-5 15:13:00
ldptest 发表于 2020-3-5 10:228 X" H' s. A( N1 ^
如果代码格式稍稍调整会更好些。

  O& j- V2 ^+ X9 L' v1 \好的,谢谢,已经调整
网络孤客 回答时间:2020-3-6 08:46:48
gaosmile 发表于 2020-3-5 15:137 S0 m' s& t' ~& R1 q
好的,谢谢,已经调整

7 E4 \  A- X0 y/ ^) F谢谢!
慎微 回答时间:2020-3-9 13:47:41
good things
天臆弄人 回答时间:2020-3-9 17:24:21
平均 1.2ms ,计算 1024 点FFT?
laney603 回答时间:2022-8-10 13:03:36
楼主,向你请教一个问题,Keil和IAR编译浮点运算的时候,这2个编译器的编译时间差异大吗?我碰到一个奇怪的问题,我有一个文件需要使用大量的浮点运算,已经开启了FPU功能,但是在IAR编译的时候,编译很慢,如果不编译那个浮点运算的文件,只要几十秒就编译完了,但是加上需要大量计算浮点运算的那个文件,编译要20多分钟,编译时间相差10倍,但是在KEIL编译的时候,包括那个需要大量浮点计算的文件,整个工程编译也就几十秒,编译的很快,这个问题不知道楼主有没有碰到过?你在编译傅里叶计算的工程的时候,有没有感觉到这2个编译器编译时间存在差异的?有什么好的建议不?在线等,感谢感谢!!!!) O1 u+ h3 \' |6 V

所属标签

关于
我们是谁
投资者关系
意法半导体可持续发展举措
创新与技术
意法半导体官网
联系我们
联系ST分支机构
寻找销售人员和分销渠道
社区
媒体中心
活动与培训
隐私策略
隐私策略
Cookies管理
行使您的权利
官方最新发布
STM32N6 AI生态系统
STM32MCU,MPU高性能GUI
ST ACEPACK电源模块
意法半导体生物传感器
STM32Cube扩展软件包
关注我们
st-img 微信公众号
st-img 手机版