The application relates to a multi-dimensional
fast Fourier transform acceleration method based on matrix operation in the field of computing
power optimization. The method comprises the following steps: S1, applying for a computing resource,
copying computing data to device global memory on the host side, generating a
DFT matrix and an input
data matrix; S2, reading the
DFT matrix data and the input
data matrix from the global memory to a register and a
shared memory; S3, using multi-thread parallelism to complete the calculation task of
matrix element multiplication of the input
data matrix and a
rotation factor matrix, and obtaining intermediate calculation result data; S4, calling a matrix operation unit to perform
matrix multiplication on the intermediate calculation result data obtained in the step S3 and the
DFT matrix, and outputting final calculation result data; and S5,
copying the final calculation result data from the device global memory to the
host memory on the host side, and releasing the computing resource. The application adopts matrix operation to calculate multi-dimensional FFT, and uses a high-performance matrix operation unit of a GPU to accelerate the calculation, so that the FFT calculation efficiency on the GPU is improved.