Line data    Source code

       1             : /**********************************************************************/
       2             : /*                    DO NOT MODIFY THIS HEADER                       */
       3             : /*             Swift, a Fourier spectral solver for MOOSE             */
       4             : /*                                                                    */
       5             : /*            Copyright 2024 Battelle Energy Alliance, LLC            */
       6             : /*                        ALL RIGHTS RESERVED                         */
       7             : /**********************************************************************/
       8             : 
       9             : #include "SwiftInit.h"
      10             : #include "SwiftApp.h"
      11             : 
      12           2 : SwiftInit::SwiftInit(int argc, char * argv[], MPI_Comm COMM_WORLD_IN)
      13           2 :   : MooseInit(argc, argv, COMM_WORLD_IN)
      14             : {
      15           2 :   for (const auto i : make_range(argc - 1))
      16           0 :     if (std::string(argv[i]) == "--libtorch-device")
      17           0 :       SwiftApp::setTorchDeviceStatic(argv[i + 1], {});
      18           2 : }