OpenMDAO: как использовать параллельную обработку для конкретной подсистемы, в то время как другие работают последовательно?

Я выполняю статический аэроупругий анализ, используя скрипт на основе OpenMDAO. Я пытаюсь адаптировать свой скрипт для запуска на суперкомпьютере.

Цикл MDA включает в себя несколько компонентов, в том числе решатель CFD, который обычно запускается в терминале с помощью mpirun -np 24 python runScript.py, и структурный решатель, который обычно запускается в терминале с помощью nast20212 nastran_static_alpha_c.bdf.

Чтобы запустить CFD в параллельном режиме в моем основном скрипте, мне нужно запустить его с помощью mpirun -np 24 python aeroelastic_mda.py.

Проблема в том, что когда я это делаю, скрипт пытается запустить все компоненты в параллельном режиме. Однако мне нужно было только запустить мой CFD-решатель для работы на нескольких узлах, в то время как другие компоненты работают в последовательном режиме.

Я просмотрел документацию OpenMDAO и другие сообщения, но не нашел информации по своей проблеме.

Ниже приведен краткий обзор моего сценария OpenMDAO, который я запускаю с mpirun -np 24 python aeroelastic_mda.py.

#Importing necessary modules
...  
if __name__ == "__main__":
    #Problem parameters
    Sw = 383.689555/2 #Wing reference surface
    V = 295.0 #Airspeed
    rho_a = 1.17 #Air density
    ... #Some other parameters
   
    root = Problem()

    #Add independent variables
    root.model.add_subsystem('wing_area', IndepVarComp('Sw', Sw), promotes=['*'])
    root.model.add_subsystem('airspeed', IndepVarComp('V', V), promotes=['*'])
    root.model.add_subsystem('air_density', IndepVarComp('rho_a', rho_a), promotes=['*'])
    ... #Some other independent variables

    mda = Group()

    #Add disciplines to the group

    # Displacement transfer: 
    # Inputs -> Displacements at structural nodes
    # Outputs -> Displacements to be applied to aerodynamic nodes 
    # Simple matrix product
    mda.add_subsystem('displacement_transfer', DisplacementTransfer(), promotes=['*'])

    # Aerodynamic solver
    mda.add_subsystem('aerodynamics', Dafoam(), promotes=['*'])

    # Load transfer
    # Inputs -> Forces at aerodynamic nodes
    # Outputs -> Forces to be applied to structural nodes 
    # Simple matrix product
    mda.add_subsystem('load_transfer', LoadTransfer(), promotes=['*'])
   
    # Structural solver
    mda.add_subsystem('structures', NastranStatic(), promotes=['*'])

    root.model.add_subsystem('mda_group', mda, promotes=['*'])

    mda.nonlinear_solver = NonlinearBlockGS()
    mda.nonlinear_solver.options['rtol'] = 1.e-3
    mda.nonlinear_solver.options['use_aitken'] = True

    mda.linear_solver = ScipyKrylov()
    root.linear_solver = ScipyKrylov()

    root.setup()
    root.run_model()

Также вот упрощенный вид моего аэродинамического компонента:

#Importing necessary modules
...
class Dafoam(ExplicitComponent):
    ... #Some constants
    
    def initialize(self):
        ... #Declaring options

    def setup(self):
        ... #Declaring inputs
        ... #Declaring outputs

    def compute(self, inputs, outputs):
        #Generate current deformed shape
        self.create_current_geom(inputs, outputs)

        #Generate the input file for Dafoam from current geometry and flow conditions
        self.create_input_file(inputs, outputs)

        # Run DAFoam
        p = Popen(['python', 'runScript.py'])
        p.wait()

        # Parse the output file from the external code    
        ...

    def create_current_geom(self, inputs, outputs):
    ...
    #Method that creates the Dafoam input file runScript.py
    def create_input_file(self, inputs, outputs):
    ...
    def get_forces(self, inputs, outputs):
    ...
    def get_aero_coeff(self, inputs, outputs):     
    ...

Также вот упрощенный вид моего структурного компонента:

#Importing necessary modules
...  
class NastranStatic(ExplicitComponent):
    ... #Some constants
    
    def initialize(self):
        ... #Declaring options

    def setup(self):
        ... #Declaring inputs
        ... #Declaring outputs

    def compute(self, inputs, outputs):
        # Generate the input file for Nastran
        self.create_input_file(inputs, outputs)

        # Run MSC Nastran
        p = Popen(['nast20212', 'nastran_static_alpha_c.bdf'])
        p.wait()

        # Parse the output file from the external code and set output values
        ...

    def create_input_file(self, inputs, outputs):
    ...
    def get_output_data(self, inputs, outputs):
    ...

Заранее большое спасибо за вашу помощь!