Task based programming : #pragma omp task versus #pragma omp parallel for

前端 未结 1 1533
抹茶落季
抹茶落季 2021-02-07 17:41

Considering :

    void saxpy_worksharing(float* x, float* y, float a, int N) {
      #pragma omp parallel for
      for (int i = 0; i < N; i++) {
         y[i         


        
相关标签:
1条回答
  • 2021-02-07 18:32

    I would suggest that you have a look at the OpenMP tutorial from Lawrence Livermore National Laboratory, available here.

    Your particular example is one that should not be implemented using OpenMP tasks. The second code creates N times the number of threads tasks (because there is an error in the code beside the missing }; I would come back to it later), and each task is only performing a very simple computation. The overhead of tasks would be gigantic, as you can see in my answer to this question. Besides the second code is conceptually wrong. Since there is no worksharing directive, all threads would execute all iterations of the loop and instead of N tasks, N times the number of threads tasks would get created. It should be rewritten in one of the following ways:

    Single task producer - common pattern, NUMA unfriendly:

    void saxpy_tasks(float* x, float* y, float a, int N) {
       #pragma omp parallel
       {
          #pragma omp single
          {
             for (int i = 0; i < N; i++)
                #pragma omp task
                {
                   y[i] = y[i]+a*x[i];
                }
          }
       }
    }
    

    The single directive would make the loop run inside a single thread only. All other threads would skip it and hit the implicit barrier at the end of the single construct. As barriers contain implicit task scheduling points, the waiting threads will start processing tasks immediately as they become available.

    Parallel task producer - more NUMA friendly:

    void saxpy_tasks(float* x, float* y, float a, int N) {
       #pragma omp parallel
       {
          #pragma omp for
          for (int i = 0; i < N; i++)
             #pragma omp task
             {
                y[i] = y[i]+a*x[i];
             }
       }
    }
    

    In this case the task creation loop would be shared among the threads.

    If you do not know what NUMA is, ignore the comments about NUMA friendliness.

    0 讨论(0)
提交回复
热议问题