Ejemplos de _get_mpi_implementation en Python

Lenguaje de programación: Python

Namespace/Package Name: horovod.run.mpi_run

Método / Función: _get_mpi_implementation

Ejemplos en hotexamples.com: 2

Python _get_mpi_implementation - 2 ejemplos encontrados. Estos son los ejemplos en Python del mundo real mejor valorados de horovod.run.mpi_run._get_mpi_implementation extraídos de proyectos de código abierto. Puedes valorar ejemplos para ayudarnos a mejorar la calidad de los ejemplos.

Ejemplo n.º 1

Mostrar archivo

Archivo: test_run.py Proyecto: pu55yf3r/horovod

 def test(output, expected, exit_code=0):
     ret = (output, exit_code) if output is not None else None
     env = {'VAR': 'val'}
     with mock.patch("horovod.run.mpi_run.tiny_shell_exec.execute",
                     return_value=ret) as m:
         implementation = _get_mpi_implementation(env)
         self.assertEqual(expected, implementation)
         m.assert_called_once_with('mpirun --version', env)

Ejemplo n.º 2

Mostrar archivo

 def test(output, expected, exit_code=0):
     ret = (output, exit_code) if output is not None else None
     with mock.patch("horovod.run.mpi_run.tiny_shell_exec.execute",
                     return_value=ret):
         implementation = _get_mpi_implementation()
         self.assertEqual(expected, implementation)