FNO-Speed: Accelerate Fourier Neural Operators Training for Solving PDEs
FNO-Speed, an integrated solution incorporating the multi-level parallel FNO-aware mapping and tiling GEMM optimization strategy and the custom-sized high-frequency signal filtering scheme, is proposed, fully demonstrating the effectiveness of the FNO-Speed optimization strategy in improving FNO performance.