80 votes

La résolution de embarassingly parallèle des problèmes à l'aide de Python multitraitement

Comment utiliser le multitraitement de s'attaquer parallèle gênant problèmes?

Embarassingly parallèle problèmes sont généralement constitués de trois éléments de base:

  1. Lire les données d'entrée (à partir d'un fichier, base de données, de connexion tcp, etc.).
  2. Exécuter des calculs sur les données d'entrée, où chaque calcul est indépendant de tout autre calcul.
  3. Écrire les résultats de calculs (pour un fichier, base de données, de connexion tcp, etc.).

On peut paralléliser le programme en deux dimensions:

  • La partie 2 peut s'exécuter sur plusieurs cœurs, puisque chaque calcul est indépendant; ordre de traitement n'a pas d'importance.
  • Chaque partie peut fonctionner de façon indépendante. La partie 1 peut placer des données sur l'entrée de la file d'attente, la partie 2 peut tirer des données de la file d'attente d'entrée et de mettre les résultats sur une file d'attente de sortie, et la partie 3 peut tirer des résultats hors de la file d'attente de sortie et de les écrire.

Cela semble des plus rythme de base de la programmation simultanée, mais je suis toujours perdu dans la tente de le résoudre, de sorte que nous allons écrire un exemple canonique pour illustrer comment cela se fait à l'aide de multitraitement.

Voici l'exemple de problème: étant Donné un fichier CSV avec des rangées de nombres entiers en entrée, calculer leurs sommes. Séparer le problème en trois parties, qui peuvent s'exécuter en parallèle:

  1. Traiter le fichier d'entrée en données brutes (listes/iterables de nombres entiers)
  2. Calculer la somme des données, en parallèle
  3. Sortie le sommes

Ci-dessous est traditionnel, d'un processus unique lié programme en Python qui résout ces trois tâches:

#!/usr/bin/env python
# -*- coding: UTF-8 -*-
# basicsums.py
"""A program that reads integer values from a CSV file and writes out their
sums to another CSV file.
"""

import csv
import optparse
import sys

def make_cli_parser():
    """Make the command line interface parser."""
    usage = "\n\n".join(["python %prog INPUT_CSV OUTPUT_CSV",
            __doc__,
            """
ARGUMENTS:
    INPUT_CSV: an input CSV file with rows of numbers
    OUTPUT_CSV: an output file that will contain the sums\
"""])
    cli_parser = optparse.OptionParser(usage)
    return cli_parser


def parse_input_csv(csvfile):
    """Parses the input CSV and yields tuples with the index of the row
    as the first element, and the integers of the row as the second
    element.

    The index is zero-index based.

    :Parameters:
    - `csvfile`: a `csv.reader` instance

    """
    for i, row in enumerate(csvfile):
        row = [int(entry) for entry in row]
        yield i, row


def sum_rows(rows):
    """Yields a tuple with the index of each input list of integers
    as the first element, and the sum of the list of integers as the
    second element.

    The index is zero-index based.

    :Parameters:
    - `rows`: an iterable of tuples, with the index of the original row
      as the first element, and a list of integers as the second element

    """
    for i, row in rows:
        yield i, sum(row)


def write_results(csvfile, results):
    """Writes a series of results to an outfile, where the first column
    is the index of the original row of data, and the second column is
    the result of the calculation.

    The index is zero-index based.

    :Parameters:
    - `csvfile`: a `csv.writer` instance to which to write results
    - `results`: an iterable of tuples, with the index (zero-based) of
      the original row as the first element, and the calculated result
      from that row as the second element

    """
    for result_row in results:
        csvfile.writerow(result_row)


def main(argv):
    cli_parser = make_cli_parser()
    opts, args = cli_parser.parse_args(argv)
    if len(args) != 2:
        cli_parser.error("Please provide an input file and output file.")
    infile = open(args[0])
    in_csvfile = csv.reader(infile)
    outfile = open(args[1], 'w')
    out_csvfile = csv.writer(outfile)
    # gets an iterable of rows that's not yet evaluated
    input_rows = parse_input_csv(in_csvfile)
    # sends the rows iterable to sum_rows() for results iterable, but
    # still not evaluated
    result_rows = sum_rows(input_rows)
    # finally evaluation takes place as a chain in write_results()
    write_results(out_csvfile, result_rows)
    infile.close()
    outfile.close()


if __name__ == '__main__':
    main(sys.argv[1:])

Prenons ce programme et de le réécrire le multitraitement de paralléliser les trois parties décrites ci-dessus. Ci-dessous est un squelette de cette nouvelle, parallélisée programme, qui doit être étoffé à l'adresse de pièces dans les commentaires:

#!/usr/bin/env python
# -*- coding: UTF-8 -*-
# multiproc_sums.py
"""A program that reads integer values from a CSV file and writes out their
sums to another CSV file, using multiple processes if desired.
"""

import csv
import multiprocessing
import optparse
import sys

NUM_PROCS = multiprocessing.cpu_count()

def make_cli_parser():
    """Make the command line interface parser."""
    usage = "\n\n".join(["python %prog INPUT_CSV OUTPUT_CSV",
            __doc__,
            """
ARGUMENTS:
    INPUT_CSV: an input CSV file with rows of numbers
    OUTPUT_CSV: an output file that will contain the sums\
"""])
    cli_parser = optparse.OptionParser(usage)
    cli_parser.add_option('-n', '--numprocs', type='int',
            default=NUM_PROCS,
            help="Number of processes to launch [DEFAULT: %default]")
    return cli_parser


def main(argv):
    cli_parser = make_cli_parser()
    opts, args = cli_parser.parse_args(argv)
    if len(args) != 2:
        cli_parser.error("Please provide an input file and output file.")
    infile = open(args[0])
    in_csvfile = csv.reader(infile)
    outfile = open(args[1], 'w')
    out_csvfile = csv.writer(outfile)

    # Parse the input file and add the parsed data to a queue for
    # processing, possibly chunking to decrease communication between
    # processes.

    # Process the parsed data as soon as any (chunks) appear on the
    # queue, using as many processes as allotted by the user
    # (opts.numprocs); place results on a queue for output.
    #
    # Terminate processes when the parser stops putting data in the
    # input queue.

    # Write the results to disk as soon as they appear on the output
    # queue.

    # Ensure all child processes have terminated.

    # Clean up files.
    infile.close()
    outfile.close()


if __name__ == '__main__':
    main(sys.argv[1:])

Ces morceaux de code, ainsi qu' un autre morceau de code qui peut générer de l'exemple des fichiers CSV pour des fins de test, peut être trouvé sur github.

J'apprécierais toute réflexion ici à la façon dont vous la simultanéité des gourous de l'approche de ce problème.


Voici quelques questions que j'avais lorsque l'on pense à propos de ce problème. Les points de Bonus pour le traitement de toute/tous:

  • Dois-je demander à l'enfant de processus de lecture des données et de la placer dans la file d'attente, ou peut la faire sans blocage jusqu'à ce que toutes les entrées sont lues?
  • De même, si j'ai un enfant procédé pour écrire les résultats de la transformation de la file d'attente, ou peut la faire sans avoir à attendre pour les résultats?
  • Dois-je utiliser un processus de piscine pour la somme des opérations?
    • Si oui, quelle méthode dois-je appeler à la piscine pour il pour commencer à traiter les résultats à venir dans l'entrée de la file d'attente, sans bloquer l'entrée et la sortie des processus, trop? apply_async()? map_async()? imap()? imap_unordered()?
  • Supposons que nous n'avons pas besoin de siphonner de l'entrée et la sortie des files d'attente comme les données saisies, mais pourrait attendre jusqu'à ce que tous les commentaires ont été analysés et tous les résultats ont été calculées (par exemple, parce que nous le savons tous, l'entrée et la sortie de l'ajustement de la mémoire système). Devrions-nous modifier l'algorithme de toute manière (par exemple, de ne pas exécuter n'importe quel processus simultanément avec I/O)?

67voto

hbar Points 1255

Ma solution a une cloche et le sifflet à assurez-vous que l'ordre de sortie est le même que l'ordre de l'entrée. J'utilise le multitraitement.la file d'attente pour l'envoi de données entre threads, l'envoi de messages stop afin de threads savoir à cesser de vérifier les files d'attente. Je pense que les commentaires dans le code source doit faire clairement comprendre ce qu'il se passe, mais si non faites le moi savoir.

#!/usr/bin/env python
# -*- coding: UTF-8 -*-
# multiproc_sums.py
"""A program that reads integer values from a CSV file and writes out their
sums to another CSV file, using multiple processes if desired.
"""

import csv
import multiprocessing
import optparse
import sys

NUM_PROCS = multiprocessing.cpu_count()

def make_cli_parser():
    """Make the command line interface parser."""
    usage = "\n\n".join(["python %prog INPUT_CSV OUTPUT_CSV",
            __doc__,
            """
ARGUMENTS:
    INPUT_CSV: an input CSV file with rows of numbers
    OUTPUT_CSV: an output file that will contain the sums\
"""])
    cli_parser = optparse.OptionParser(usage)
    cli_parser.add_option('-n', '--numprocs', type='int',
            default=NUM_PROCS,
            help="Number of processes to launch [DEFAULT: %default]")
    return cli_parser

class CSVWorker(object):
    def __init__(self, numprocs, infile, outfile):
        self.numprocs = numprocs
        self.infile = open(infile)
        self.outfile = outfile
        self.in_csvfile = csv.reader(self.infile)
        self.inq = multiprocessing.Queue()
        self.outq = multiprocessing.Queue()

        self.pin = multiprocessing.Process(target=self.parse_input_csv, args=())
        self.pout = multiprocessing.Process(target=self.write_output_csv, args=())
        self.ps = [ multiprocessing.Process(target=self.sum_row, args=())
                        for i in range(self.numprocs)]

        self.pin.start()
        self.pout.start()
        for p in self.ps:
            p.start()

        self.pin.join()
        i = 0
        for p in self.ps:
            p.join()
            print "Done", i
            i += 1

        self.pout.join()
        self.infile.close()

    def parse_input_csv(self):
            """Parses the input CSV and yields tuples with the index of the row
            as the first element, and the integers of the row as the second
            element.

            The index is zero-index based.

            The data is then sent over inqueue for the workers to do their
            thing.  At the end the input thread sends a 'STOP' message for each
            worker.
            """
            for i, row in enumerate(self.in_csvfile):
                row = [ int(entry) for entry in row ]
                self.inq.put( (i, row) )

            for i in range(self.numprocs):
                self.inq.put("STOP")

    def sum_row(self):
        """
        Workers. Consume inq and produce answers on outq
        """
        tot = 0
        for i, row in iter(self.inq.get, "STOP"):
                self.outq.put( (i, sum(row)) )
        self.outq.put("STOP")

    def write_output_csv(self):
        """
        Open outgoing csv file then start reading outq for answers
        Since I chose to make sure output was synchronized to the input there
        is some extra goodies to do that.

        Obviously your input has the original row number so this is not
        required.
        """
        cur = 0
        stop = 0
        buffer = {}
        # For some reason csv.writer works badly across threads so open/close
        # and use it all in the same thread or else you'll have the last
        # several rows missing
        outfile = open(self.outfile, "w")
        self.out_csvfile = csv.writer(outfile)

        #Keep running until we see numprocs STOP messages
        for works in range(self.numprocs):
            for i, val in iter(self.outq.get, "STOP"):
                # verify rows are in order, if not save in buffer
                if i != cur:
                    buffer[i] = val
                else:
                    #if yes are write it out and make sure no waiting rows exist
                    self.out_csvfile.writerow( [i, val] )
                    cur += 1
                    while cur in buffer:
                        self.out_csvfile.writerow([ cur, buffer[cur] ])
                        del buffer[cur]
                        cur += 1

        outfile.close()

def main(argv):
    cli_parser = make_cli_parser()
    opts, args = cli_parser.parse_args(argv)
    if len(args) != 2:
        cli_parser.error("Please provide an input file and output file.")

    c = CSVWorker(opts.numprocs, args[0], args[1])

if __name__ == '__main__':
    main(sys.argv[1:])

6voto

Gael Varoquaux Points 862

Arrivé en retard à la fête...

joblib a une couche sur le dessus de multitraitement pour vous aider à faire en parallèle pour les boucles. Il vous donne des installations comme un paresseux à la distribution des emplois, et de mieux les rapports d'erreurs en plus de sa syntaxe simple.

Comme un avertissement, je suis l'auteur original de joblib.

4voto

S.Lott Points 207588

De La Vieille École.

p1.py

import csv
import pickle
import sys

with open( "someFile", "rb" ) as source:
    rdr = csv.reader( source )
    for line in eumerate( rdr ):
        pickle.dump( line, sys.stdout )

p2.py

import pickle
import sys

while True:
    try:
        i, row = pickle.load( sys.stdin )
    except EOFError:
        break
    pickle.dump( i, sum(row) )

p3.py

import pickle
import sys
while True:
    try:
        i, row = pickle.load( sys.stdin )
    except EOFError:
        break
    print i, row

Voici le multi-traitement de la structure finale.

python p1.py | python p2.py | python p3.py

Oui, la coquille a les assemble au niveau de l'OS. Il semble plus simple pour moi et ça fonctionne très bien.

Oui, il y a un peu plus de frais généraux en utilisant des cornichons (ou cPickle). La simplification, cependant, semble en vaut la peine.

Si vous voulez le nom du fichier à être un argument pour p1.py, c'est un changement facile.

Plus important encore, une fonction telle que la suivante est très pratique.

def get_stdin():
    while True:
        try:
            yield pickle.load( sys.stdin )
        except EOFError:
            return

Qui vous permet de le faire:

for item in get_stdin():
     process item

C'est très simple, mais il n'est pas facilement vous permettre de disposer de plusieurs copies de P2.py en cours d'exécution.

Vous avez deux problèmes: le ventilateur et ventilateur. L'P1.py doit en quelque sorte de ventilateur à plusieurs P2.py'. Et la P2.py s'doit en quelque sorte fusionner leurs résultats en une seule P3.py.

La vieille école de l'approche de fan-out est un "Push" de l'architecture, qui est très efficace.

Théoriquement, plusieurs P2.py s'en tirant à partir d'une file d'attente commune est l'allocation optimale des ressources. C'est souvent la solution idéale, mais c'est aussi une bonne quantité de programmation. Est la programmation vraiment nécessaire? Ou sera-round-robin de traitement d'être assez bon?

Dans la pratique, vous verrez que faire P1.py faire un simple "round robin" les rapports entre plusieurs P2.py's peut être tout à fait bon. Vous auriez P1.py configuré pour traiter de n copies de P2.py via des canaux nommés. L'P2.py's chaque lecture de leur canal approprié.

Que faire si l'un P2.py obtient tout le "pire des cas" des données et s'exécute en arrière? Oui, round-robin n'est pas parfait. Mais c'est mieux qu'un seul P2.py et vous pouvez répondre à ce biais avec un simple hasard.

Fan-in à partir de plusieurs P2.py pour une P3.py est un peu plus complexe encore. À ce stade, le vieux de la démarche de l'école cesse d'être avantageux. P3.py besoin de lire à partir de plusieurs canaux nommés à l'aide de l' select bibliothèque pour intercaler le lit.

2voto

snies Points 1983

0voto

Vatine Points 8884

Il est probablement possible d'introduire un peu de parallélisme dans la partie 1. Probablement pas un problème avec un format aussi simple que CSV, mais si le traitement des données d'entrée est sensiblement plus lente que la lecture des données, vous avez pu lire de plus gros morceaux, puis continuer à lire jusqu'à ce que vous trouver un "séparateur de ligne" (saut de ligne dans le fichier CSV cas, mais encore une fois cela dépend du format de lecture; ne fonctionne pas si le format est assez complexe).

Ces morceaux, chacun contenant probablement des entrées multiples, peuvent être cultivées à une foule de processus en parallèle de la lecture des missions à l'extérieur d'une file d'attente, où ils sont analysés et split, puis placé sur la file d'attente pour le stade 2.

Prograide.com

Prograide est une communauté de développeurs qui cherche à élargir la connaissance de la programmation au-delà de l'anglais.
Pour cela nous avons les plus grands doutes résolus en français et vous pouvez aussi poser vos propres questions ou résoudre celles des autres.

Powered by:

X