Skip to main content
Log in

A TensorFlow-based new high-performance computational framework for CFD

  • Article
  • Published:
Journal of Hydrodynamics Aims and scope Submit manuscript

Abstract

In this study, a computational framework in the field of artificial intelligence was applied in computational fluid dynamics (CFD) field. This Framework, which was initially proposed by Google AI department, is called “TensorFlow”. An improved CFD model based on this framework was developed with a high-order difference method, which is a constrained interpolation profile (CIP) scheme for the base flow solver of the advection term in the Navier-Stokes equations, and preconditioned conjugate gradient (PCG) method was implemented in the model to solve the Poisson equation. Some new features including the convolution, vectorization, and graphics processing unit (GPU) acceleration were implemented to raise the computational efficiency. The model was tested with several benchmark cases and shows good performance. Compared with our former CIP-based model, the present TensorFlow-based model also shows significantly higher computational efficiency in large-scale computation. The results indicate TensorFlow could be a promising framework for CFD models due to its ability in the computational acceleration and convenience for programming.

This is a preview of subscription content, log in via an institution to check access.

Access this article

Price excludes VAT (USA)
Tax calculation will be finalised during checkout.

Instant access to the full article PDF.

Similar content being viewed by others

References

  1. Sasan T., Patrick L. Celeris: A GPU-accelerated open source software with a Boussinesq-type wave solver for real-time interactive simulation and visualization [J]. Computer Physics Communications, 2017, 217: 117–127.

    Article  MATH  Google Scholar 

  2. Xu C., Deng X., "Zhang L. et al. Collaborating CPU and GPU for large-scale high-order CFD simulations with complex grids on the Tianhe-1A supercomputer [J]. Journal of Computational Physics, 2014, 278: 275–297.

    Article  MATH  Google Scholar 

  3. Thibault J. C., Senocak I. CUDA implementation of a Navier-Stokes solver on multi-GPU desktop platforms for incompressible flows [C]. Proceedings of the 47th AIAA Aerospace Sciences Meeting, Florida, USA, 2009.

    Google Scholar 

  4. Farshid M., Riccardo R., Pooyan D. et al. OpenCL-based implementation of an unstructured edge-based finite element convection-diffusion solver on graphics hardware [J]. International Journal for Numerical Methods in Engineering, 2012, 89(13): 1635–1651.

    Article  Google Scholar 

  5. Abadi M. Tensorflow: Learning functions at scale [C]. Proceedings of the 21st ACM SIGPLAN International Conference on Functional Programming, New York, USA, 2016.

    Google Scholar 

  6. Aminian J. A. Scale adaptive simulation of vortex structures past a square cylinder [J]. Journal of Hydrodynamics, 2018, 30(4): 657–671.

    Article  Google Scholar 

  7. Wang D. X., Sun J. W., Gui J. S. et al. A numerical piston-type wave-maker toolbox for the open-source library OpenFOAM [J]. Journal of Hydrodynamics, 2019, 31(4): 800–813.

    Article  Google Scholar 

  8. Zhao X. J., Zong Z., Jiang Y. C. et al., Numerical simulation of micro-bubble drag reduction of an axisymmetric body using OpenFOAM [J]. Journal of Hydrodynamics, 2019, 31(5): 900–910.

    Article  Google Scholar 

  9. Liu H. L., Ren Y., Wang K. et al. Research of inner flow in a double blades pump based on OpenFOAM [J]. Journal of Hydrodynamics, 2012, 24(2): 226–234.

    Article  Google Scholar 

  10. Wang J. H., Zhao W. W., Wan D. C. Development of naoe-FOAM-SJTU solver based on OpenFOAM for marine hydrodynamics [J]. Journal of Hydrodynamics, 2019, 31(1): 1–20.

    Article  Google Scholar 

  11. Hans B., Arun K. Combined level set/ghost cell immersed boundary representation for floating body simulations [J]. International Journal For Numerical Methods In Fluids, 2017, 83: 905–916.

    Article  MathSciNet  Google Scholar 

  12. Yabe T., Aoki T., Sakaguchi G. et al. The compact CIP (cubic-interpolated pseudo-particle) method as a general hyperbolic solver [J]. Computers and Fluids, 1991, 19(3-4): 421–431.

    Article  MathSciNet  MATH  Google Scholar 

  13. Ye Z., Zhao X. Investigation of water-water interface in dam break flow with a wet bed [J]. Journal of Hydrology, 2017, 548: 104–120.

    Article  Google Scholar 

  14. Li M., Zhao X., Ye Z. et al. Generation of regular and focused waves by using an internal wave maker in a CIP-based model [J]. Ocean Engineering, 2018, 167: 334–347.

    Article  Google Scholar 

  15. Fu Y. N., Zhao X. Z., Cao F. F. et al. Numerical simulation of viscous flow past an oscillating square cylinder using a CIP-based model [J]. Journal of Hydrodynamics, 2017, 29(1): 96–108.

    Article  Google Scholar 

  16. Reid J. K. On the method of conjugate gradients for the solution of large sparse systems of linear equations [C]. Proceedings of the Conference on Large Sparse Sets of Linear Equations, 1971.

    Google Scholar 

  17. Guermond J., Minev P., Shen J. An overview of projection methods for incompressible flows [J]. Computer Methods in Applied Mechanics and Engineering, 2006, 195(44-47): 6011–6045.

    Article  MathSciNet  MATH  Google Scholar 

  18. Peskin C. S. Numerical analysis of blood flow in the heart [J]. Journal of Computational Physics, 1977, 25(3): 220–252.

    Article  MathSciNet  MATH  Google Scholar 

  19. Fujimatsu N., Suzuki K. New interpolation technique for the CIP method on curvilinear coordinates [J]. Journal of Computational Physics, 2010, 229(16): 5573–5596.

    Article  MATH  Google Scholar 

  20. Jeffrey B., Ian F., Eitan G. et al. Sparse matrix solvers on the GPU: Conjugate gradients and multigrid [J]. ACM Transactions Graphics, 2003, 22(3): 917–924.

    Article  Google Scholar 

  21. Zalesak S. T. Fully multidimensional flux-corrected transport algorithms for fluids [J]. Journal of Computational Physics, 1979, 31(3): 335–362.

    Article  MathSciNet  MATH  Google Scholar 

  22. Fukumitsu K., Yabe T., Ogata Y. et al. A new directionalsplitting CIP interpolation with high accuracy and low memory consumption [J]. Journal of Computational Physics, 2015, 286: 62–69.

    Article  MathSciNet  MATH  Google Scholar 

  23. Bruneau C. H., Saad M. The 2D lid-driven cavity problem revisited [J]. Computers and Fluids, 2006, 35(3): 326–348.

    Article  MATH  Google Scholar 

  24. Ghia U., Ghia K., Shin C. High-Re solutions for incompressible flow using the Navier-Stokes equations and a multigrid method [J]. Journal of Computational Physics, 1982, 48(3): 387–411.

    Article  MATH  Google Scholar 

  25. Wang S., Zhang X. An immersed boundary method based on discrete stream function formulation for two- and three-dimensional incompressible flows [J]. Journal of Computational Physics, 2011, 230(9): 3479–3499.

    Article  MathSciNet  MATH  Google Scholar 

  26. Tseng Y. H., Ferziger J. H. A ghost-cell immersed boundary method for flow in complex geometry [J]. Journal of Computational Physics, 2003, 192(2): 593–623.

    Article  MathSciNet  MATH  Google Scholar 

  27. Kim J., Kim D., Choi H. An immersed-boundary finitevolume method for simulations of flow in complex geometries [J]. Journal of Computational Physics, 2001, 171(1): 132–150.

    Article  MathSciNet  MATH  Google Scholar 

  28. Lai M., Peskin C. An immersed boundary method with formal second order accuracy and reduced numerical viscosity [J]. Journal of Computational Physics, 2000, 160(2): 705–719.

    Article  MathSciNet  MATH  Google Scholar 

  29. Rajani B., Kandasamy A., Majumdar S. Numerical simulation of laminar flow past a circular cylinder [J]. Applied Mathematical Modelling, 2009, 33(3): 1228–1247.

    Article  MathSciNet  MATH  Google Scholar 

Download references

Acknowledgements

This work was supported by the Natural Science Foundation of Zhejiang Provincial (Grant No. LR16E090002), the Fundamental Research Funds for the Central Universities (Grant No. 2018QNA4041), Blue Bay Renovation Project of Pingtan Comprehensive Pilot Zone, the Bureau of Science and Technology of Zhoushan (Grant No. 2018C81040), the HPC Center OF ZJU (Zhoushan Campus) and the Tang scholar.

Author information

Authors and Affiliations

Authors

Corresponding author

Correspondence to Wei-jie Liu.

Additional information

Project supported by the National Natural Science Foundation of China (Grant No. 51679212, 51979245).

Biography: Xi-zeng Zhao (1979-), Male, Ph. D., Professor

Rights and permissions

Reprints and permissions

About this article

Check for updates. Verify currency and authenticity via CrossMark

Cite this article

Zhao, Xz., Xu, Ty., Ye, Zt. et al. A TensorFlow-based new high-performance computational framework for CFD. J Hydrodyn 32, 735–746 (2020). https://doi.org/10.1007/s42241-020-0050-0

Download citation

  • Received:

  • Revised:

  • Accepted:

  • Published:

  • Issue Date:

  • DOI: https://doi.org/10.1007/s42241-020-0050-0

Key words

Navigation