|
@@ -1,870 +0,0 @@
|
|
-/* StarPU --- Runtime system for heterogeneous multicore architectures.
|
|
|
|
- *
|
|
|
|
- * Copyright (C) 2010, 2011 Université de Bordeaux 1
|
|
|
|
- * Copyright (C) 2010, 2012 Centre National de la Recherche Scientifique
|
|
|
|
- *
|
|
|
|
- * StarPU is free software; you can redistribute it and/or modify
|
|
|
|
- * it under the terms of the GNU Lesser General Public License as published by
|
|
|
|
- * the Free Software Foundation; either version 2.1 of the License, or (at
|
|
|
|
- * your option) any later version.
|
|
|
|
- *
|
|
|
|
- * StarPU is distributed in the hope that it will be useful, but
|
|
|
|
- * WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
|
|
- * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
|
|
|
|
- *
|
|
|
|
- * See the GNU Lesser General Public License in COPYING.LGPL for more details.
|
|
|
|
- */
|
|
|
|
-
|
|
|
|
-#include "pxlu.h"
|
|
|
|
-#include "pxlu_kernels.h"
|
|
|
|
-#include <sys/time.h>
|
|
|
|
-
|
|
|
|
-#define MPI_TAG11(k) ((1U << 16) | (k))
|
|
|
|
-#define MPI_TAG12(k, j) ((2U << 16) | (k)<<8 | (j))
|
|
|
|
-#define MPI_TAG21(k, i) ((3U << 16) | (i)<<8 | (k))
|
|
|
|
-
|
|
|
|
-// 11 21
|
|
|
|
-// 12 22
|
|
|
|
-
|
|
|
|
-#define TAG11(k) ((starpu_tag_t)( (1ULL<<50) | (unsigned long long)(k)))
|
|
|
|
-#define TAG12(k,j) ((starpu_tag_t)(((2ULL<<50) | (((unsigned long long)(k))<<32) \
|
|
|
|
- | (unsigned long long)(j))))
|
|
|
|
-#define TAG21(k,i) ((starpu_tag_t)(((3ULL<<50) | (((unsigned long long)(k))<<32) \
|
|
|
|
- | (unsigned long long)(i))))
|
|
|
|
-#define TAG22(k,i,j) ((starpu_tag_t)(((4ULL<<50) | ((unsigned long long)(k)<<32) \
|
|
|
|
- | ((unsigned long long)(i)<<16) \
|
|
|
|
- | (unsigned long long)(j))))
|
|
|
|
-#define TAG11_SAVE(k) ((starpu_tag_t)( (5ULL<<50) | (unsigned long long)(k)))
|
|
|
|
-#define TAG12_SAVE(k,j) ((starpu_tag_t)(((6ULL<<50) | (((unsigned long long)(k))<<32) \
|
|
|
|
- | (unsigned long long)(j))))
|
|
|
|
-#define TAG21_SAVE(k,i) ((starpu_tag_t)(((7ULL<<50) | (((unsigned long long)(k))<<32) \
|
|
|
|
- | (unsigned long long)(i))))
|
|
|
|
-
|
|
|
|
-#define TAG11_SAVE_PARTIAL(k) ((starpu_tag_t)( (8ULL<<50) | (unsigned long long)(k)))
|
|
|
|
-#define TAG12_SAVE_PARTIAL(k,j) ((starpu_tag_t)(((9ULL<<50) | (((unsigned long long)(k))<<32) \
|
|
|
|
- | (unsigned long long)(j))))
|
|
|
|
-#define TAG21_SAVE_PARTIAL(k,i) ((starpu_tag_t)(((10ULL<<50) | (((unsigned long long)(k))<<32) \
|
|
|
|
- | (unsigned long long)(i))))
|
|
|
|
-
|
|
|
|
-#define STARPU_TAG_INIT ((starpu_tag_t)(11ULL<<50))
|
|
|
|
-
|
|
|
|
-//#define VERBOSE_INIT 1
|
|
|
|
-
|
|
|
|
-//#define DEBUG 1
|
|
|
|
-
|
|
|
|
-static unsigned no_prio = 0;
|
|
|
|
-
|
|
|
|
-static unsigned nblocks = 0;
|
|
|
|
-static int rank = -1;
|
|
|
|
-static int world_size = -1;
|
|
|
|
-
|
|
|
|
-struct callback_arg {
|
|
|
|
- unsigned i, j, k;
|
|
|
|
-};
|
|
|
|
-
|
|
|
|
-/*
|
|
|
|
- * Various
|
|
|
|
- */
|
|
|
|
-
|
|
|
|
-static struct debug_info *create_debug_info(unsigned i, unsigned j, unsigned k)
|
|
|
|
-{
|
|
|
|
- struct debug_info *info = malloc(sizeof(struct debug_info));
|
|
|
|
-
|
|
|
|
- info->i = i;
|
|
|
|
- info->j = j;
|
|
|
|
- info->k = k;
|
|
|
|
-
|
|
|
|
- return info;
|
|
|
|
-}
|
|
|
|
-
|
|
|
|
-static struct starpu_task *create_task(starpu_tag_t id)
|
|
|
|
-{
|
|
|
|
- struct starpu_task *task = starpu_task_create();
|
|
|
|
- task->cl_arg = NULL;
|
|
|
|
-
|
|
|
|
- task->use_tag = 1;
|
|
|
|
- task->tag_id = id;
|
|
|
|
-
|
|
|
|
- return task;
|
|
|
|
-}
|
|
|
|
-
|
|
|
|
-/* Send handle to every node appearing in the mask, and unlock tag once the
|
|
|
|
- * transfers are done. */
|
|
|
|
-static void send_data_to_mask(starpu_data_handle_t handle, int *rank_mask, int mpi_tag, starpu_tag_t tag)
|
|
|
|
-{
|
|
|
|
- unsigned cnt = 0;
|
|
|
|
-
|
|
|
|
- STARPU_ASSERT(handle != STARPU_POISON_PTR);
|
|
|
|
-
|
|
|
|
- int rank_array[world_size];
|
|
|
|
- MPI_Comm comm_array[world_size];
|
|
|
|
- int mpi_tag_array[world_size];
|
|
|
|
- starpu_data_handle_t handle_array[world_size];
|
|
|
|
-
|
|
|
|
- unsigned r;
|
|
|
|
- for (r = 0; r < world_size; r++)
|
|
|
|
- {
|
|
|
|
- if (rank_mask[r]) {
|
|
|
|
- rank_array[cnt] = r;
|
|
|
|
-
|
|
|
|
- comm_array[cnt] = MPI_COMM_WORLD;
|
|
|
|
- mpi_tag_array[cnt] = mpi_tag;
|
|
|
|
- handle_array[cnt] = handle;
|
|
|
|
- cnt++;
|
|
|
|
- }
|
|
|
|
- }
|
|
|
|
-
|
|
|
|
- if (cnt == 0)
|
|
|
|
- {
|
|
|
|
- /* In case there is no message to send, we release the tag at
|
|
|
|
- * once */
|
|
|
|
- starpu_tag_notify_from_apps(tag);
|
|
|
|
- }
|
|
|
|
- else {
|
|
|
|
- starpu_mpi_isend_array_detached_unlock_tag(cnt, handle_array,
|
|
|
|
- rank_array, mpi_tag_array, comm_array, tag);
|
|
|
|
- }
|
|
|
|
-}
|
|
|
|
-
|
|
|
|
-/* Initiate a receive request once all dependencies are fulfilled and unlock
|
|
|
|
- * tag 'unlocked_tag' once it's done. */
|
|
|
|
-
|
|
|
|
-struct recv_when_done_callback_arg {
|
|
|
|
- int source;
|
|
|
|
- int mpi_tag;
|
|
|
|
- starpu_data_handle_t handle;
|
|
|
|
- starpu_tag_t unlocked_tag;
|
|
|
|
-};
|
|
|
|
-
|
|
|
|
-static void callback_receive_when_done(void *_arg)
|
|
|
|
-{
|
|
|
|
- struct recv_when_done_callback_arg *arg = _arg;
|
|
|
|
-
|
|
|
|
- starpu_mpi_irecv_detached_unlock_tag(arg->handle, arg->source,
|
|
|
|
- arg->mpi_tag, MPI_COMM_WORLD, arg->unlocked_tag);
|
|
|
|
-
|
|
|
|
- free(arg);
|
|
|
|
-}
|
|
|
|
-
|
|
|
|
-static void receive_when_deps_are_done(unsigned ndeps, starpu_tag_t *deps_tags,
|
|
|
|
- int source, int mpi_tag,
|
|
|
|
- starpu_data_handle_t handle,
|
|
|
|
- starpu_tag_t partial_tag,
|
|
|
|
- starpu_tag_t unlocked_tag)
|
|
|
|
-{
|
|
|
|
- STARPU_ASSERT(handle != STARPU_POISON_PTR);
|
|
|
|
-
|
|
|
|
- struct recv_when_done_callback_arg *arg =
|
|
|
|
- malloc(sizeof(struct recv_when_done_callback_arg));
|
|
|
|
-
|
|
|
|
- arg->source = source;
|
|
|
|
- arg->mpi_tag = mpi_tag;
|
|
|
|
- arg->handle = handle;
|
|
|
|
- arg->unlocked_tag = unlocked_tag;
|
|
|
|
-
|
|
|
|
- if (ndeps == 0)
|
|
|
|
- {
|
|
|
|
- callback_receive_when_done(arg);
|
|
|
|
- return;
|
|
|
|
- }
|
|
|
|
-
|
|
|
|
- starpu_create_sync_task(partial_tag, ndeps, deps_tags,
|
|
|
|
- callback_receive_when_done, arg);
|
|
|
|
-}
|
|
|
|
-
|
|
|
|
-/*
|
|
|
|
- * Task 11 (diagonal factorization)
|
|
|
|
- */
|
|
|
|
-
|
|
|
|
-static void create_task_11_recv(unsigned k)
|
|
|
|
-{
|
|
|
|
- /* The current node is not computing that task, so we receive the block
|
|
|
|
- * with MPI */
|
|
|
|
-
|
|
|
|
- /* We don't issue a MPI receive request until everyone using the
|
|
|
|
- * temporary buffer is done : 11_(k-1) can be used by 12_(k-1)j and
|
|
|
|
- * 21(k-1)i with i,j >= k */
|
|
|
|
- unsigned ndeps = 0;
|
|
|
|
- starpu_tag_t tag_array[2*nblocks];
|
|
|
|
-
|
|
|
|
-#ifdef SINGLE_TMP11
|
|
|
|
- unsigned i, j;
|
|
|
|
- if (k > 0)
|
|
|
|
- for (i = (k-1)+1; i < nblocks; i++)
|
|
|
|
- {
|
|
|
|
- if (rank == get_block_rank(i, k-1))
|
|
|
|
- tag_array[ndeps++] = TAG21(k-1, i);
|
|
|
|
- }
|
|
|
|
-
|
|
|
|
- if (k > 0)
|
|
|
|
- for (j = (k-1)+1; j < nblocks; j++)
|
|
|
|
- {
|
|
|
|
- if (rank == get_block_rank(k-1, j))
|
|
|
|
- tag_array[ndeps++] = TAG12(k-1, j);
|
|
|
|
- }
|
|
|
|
-#endif
|
|
|
|
-
|
|
|
|
- int source = get_block_rank(k, k);
|
|
|
|
-#ifdef SINGLE_TMP11
|
|
|
|
- starpu_data_handle_t block_handle = STARPU_PLU(get_tmp_11_block_handle)();
|
|
|
|
-#else
|
|
|
|
- starpu_data_handle_t block_handle = STARPU_PLU(get_tmp_11_block_handle)(k);
|
|
|
|
-#endif
|
|
|
|
- int mpi_tag = MPI_TAG11(k);
|
|
|
|
- starpu_tag_t partial_tag = TAG11_SAVE_PARTIAL(k);
|
|
|
|
- starpu_tag_t unlocked_tag = TAG11_SAVE(k);
|
|
|
|
-
|
|
|
|
-// fprintf(stderr, "NODE %d - 11 (%d) - recv when done ndeps %d - tag array %lx\n", rank, k, ndeps, tag_array[0]);
|
|
|
|
- receive_when_deps_are_done(ndeps, tag_array, source, mpi_tag, block_handle, partial_tag, unlocked_tag);
|
|
|
|
-}
|
|
|
|
-
|
|
|
|
-static void find_nodes_using_11(unsigned k, int *rank_mask)
|
|
|
|
-{
|
|
|
|
- memset(rank_mask, 0, world_size*sizeof(int));
|
|
|
|
-
|
|
|
|
- /* Block 11_k is used to compute 12_kj + 12ki with i,j > k */
|
|
|
|
- unsigned i;
|
|
|
|
- for (i = k+1; i < nblocks; i++)
|
|
|
|
- {
|
|
|
|
- int r = get_block_rank(i, k);
|
|
|
|
- rank_mask[r] = 1;
|
|
|
|
- }
|
|
|
|
-
|
|
|
|
- unsigned j;
|
|
|
|
- for (j = k+1; j < nblocks; j++)
|
|
|
|
- {
|
|
|
|
- int r = get_block_rank(k, j);
|
|
|
|
- rank_mask[r] = 1;
|
|
|
|
- }
|
|
|
|
-}
|
|
|
|
-
|
|
|
|
-static void callback_task_11_real(void *_arg)
|
|
|
|
-{
|
|
|
|
- struct callback_arg *arg = _arg;
|
|
|
|
-
|
|
|
|
- unsigned k = arg->k;
|
|
|
|
-
|
|
|
|
- /* Find all the nodes potentially requiring this block */
|
|
|
|
- int rank_mask[world_size];
|
|
|
|
- find_nodes_using_11(k, rank_mask);
|
|
|
|
- rank_mask[rank] = 0;
|
|
|
|
-
|
|
|
|
- /* Send the block to those nodes */
|
|
|
|
- starpu_data_handle_t block_handle = STARPU_PLU(get_block_handle)(k, k);
|
|
|
|
- starpu_tag_t tag = TAG11_SAVE(k);
|
|
|
|
- int mpi_tag = MPI_TAG11(k);
|
|
|
|
- send_data_to_mask(block_handle, rank_mask, mpi_tag, tag);
|
|
|
|
-
|
|
|
|
- free(arg);
|
|
|
|
-}
|
|
|
|
-
|
|
|
|
-static void create_task_11_real(unsigned k)
|
|
|
|
-{
|
|
|
|
- struct starpu_task *task = create_task(TAG11(k));
|
|
|
|
-
|
|
|
|
- task->cl = &STARPU_PLU(cl11);
|
|
|
|
-
|
|
|
|
- task->cl_arg = create_debug_info(k, k, k);
|
|
|
|
-
|
|
|
|
- /* which sub-data is manipulated ? */
|
|
|
|
- task->handles[0] = STARPU_PLU(get_block_handle)(k, k);
|
|
|
|
-
|
|
|
|
- struct callback_arg *arg = malloc(sizeof(struct callback_arg));
|
|
|
|
- arg->k = k;
|
|
|
|
-
|
|
|
|
- task->callback_func = callback_task_11_real;
|
|
|
|
- task->callback_arg = arg;
|
|
|
|
-
|
|
|
|
- /* this is an important task */
|
|
|
|
- if (!no_prio)
|
|
|
|
- task->priority = STARPU_MAX_PRIO;
|
|
|
|
-
|
|
|
|
- /* enforce dependencies ... */
|
|
|
|
- if (k > 0) {
|
|
|
|
- starpu_tag_declare_deps(TAG11(k), 1, TAG22(k-1, k, k));
|
|
|
|
- }
|
|
|
|
- else {
|
|
|
|
- starpu_tag_declare_deps(TAG11(k), 1, STARPU_TAG_INIT);
|
|
|
|
- }
|
|
|
|
-
|
|
|
|
- int ret = starpu_task_submit(task);
|
|
|
|
- STARPU_CHECK_RETURN_VALUE(ret, "starpu_task_submit");
|
|
|
|
-}
|
|
|
|
-
|
|
|
|
-static void create_task_11(unsigned k)
|
|
|
|
-{
|
|
|
|
- if (get_block_rank(k, k) == rank)
|
|
|
|
- {
|
|
|
|
-#ifdef VERBOSE_INIT
|
|
|
|
- fprintf(stderr, "CREATE real task 11(%d) (TAG11_SAVE(%d) = %lx) on node %d\n", k, k, TAG11_SAVE(k), rank);
|
|
|
|
-#endif
|
|
|
|
- create_task_11_real(k);
|
|
|
|
- }
|
|
|
|
- else {
|
|
|
|
- /* We don't handle the task, but perhaps we have to generate MPI transfers. */
|
|
|
|
- int rank_mask[world_size];
|
|
|
|
- find_nodes_using_11(k, rank_mask);
|
|
|
|
-
|
|
|
|
- if (rank_mask[rank])
|
|
|
|
- {
|
|
|
|
-#ifdef VERBOSE_INIT
|
|
|
|
- fprintf(stderr, "create RECV task 11(%d) on node %d\n", k, rank);
|
|
|
|
-#endif
|
|
|
|
- create_task_11_recv(k);
|
|
|
|
- }
|
|
|
|
- else {
|
|
|
|
-#ifdef VERBOSE_INIT
|
|
|
|
- fprintf(stderr, "Node %d needs not 11(%d)\n", rank, k);
|
|
|
|
-#endif
|
|
|
|
- }
|
|
|
|
- }
|
|
|
|
-}
|
|
|
|
-
|
|
|
|
-
|
|
|
|
-
|
|
|
|
-/*
|
|
|
|
- * Task 12 (Update lower left (TRSM))
|
|
|
|
- */
|
|
|
|
-
|
|
|
|
-static void create_task_12_recv(unsigned k, unsigned j)
|
|
|
|
-{
|
|
|
|
- unsigned i;
|
|
|
|
-
|
|
|
|
- /* The current node is not computing that task, so we receive the block
|
|
|
|
- * with MPI */
|
|
|
|
-
|
|
|
|
- /* We don't issue a MPI receive request until everyone using the
|
|
|
|
- * temporary buffer is done : 12_(k-1)j can be used by 22_(k-1)ij with
|
|
|
|
- * i >= k */
|
|
|
|
- unsigned ndeps = 0;
|
|
|
|
- starpu_tag_t tag_array[nblocks];
|
|
|
|
-
|
|
|
|
-#ifdef SINGLE_TMP1221
|
|
|
|
- if (k > 0)
|
|
|
|
- for (i = (k-1)+1; i < nblocks; i++)
|
|
|
|
-#else
|
|
|
|
- if (k > 1)
|
|
|
|
- for (i = (k-2)+1; i < nblocks; i++)
|
|
|
|
-#endif
|
|
|
|
- {
|
|
|
|
- if (rank == get_block_rank(i, j))
|
|
|
|
-#ifdef SINGLE_TMP1221
|
|
|
|
- tag_array[ndeps++] = TAG22(k-1, i, j);
|
|
|
|
-#else
|
|
|
|
- tag_array[ndeps++] = TAG22(k-2, i, j);
|
|
|
|
-#endif
|
|
|
|
- }
|
|
|
|
-
|
|
|
|
- int source = get_block_rank(k, j);
|
|
|
|
-#ifdef SINGLE_TMP1221
|
|
|
|
- starpu_data_handle_t block_handle = STARPU_PLU(get_tmp_12_block_handle)(j);
|
|
|
|
-#else
|
|
|
|
- starpu_data_handle_t block_handle = STARPU_PLU(get_tmp_12_block_handle)(j,k);
|
|
|
|
-#endif
|
|
|
|
- int mpi_tag = MPI_TAG12(k, j);
|
|
|
|
- starpu_tag_t partial_tag = TAG12_SAVE_PARTIAL(k, j);
|
|
|
|
- starpu_tag_t unlocked_tag = TAG12_SAVE(k, j);
|
|
|
|
-
|
|
|
|
- receive_when_deps_are_done(ndeps, tag_array, source, mpi_tag, block_handle, partial_tag, unlocked_tag);
|
|
|
|
-}
|
|
|
|
-
|
|
|
|
-static void find_nodes_using_12(unsigned k, unsigned j, int *rank_mask)
|
|
|
|
-{
|
|
|
|
- memset(rank_mask, 0, world_size*sizeof(int));
|
|
|
|
-
|
|
|
|
- /* Block 12_kj is used to compute 22_kij with i > k */
|
|
|
|
- unsigned i;
|
|
|
|
- for (i = k+1; i < nblocks; i++)
|
|
|
|
- {
|
|
|
|
- int r = get_block_rank(i, j);
|
|
|
|
- rank_mask[r] = 1;
|
|
|
|
- }
|
|
|
|
-}
|
|
|
|
-
|
|
|
|
-static void callback_task_12_real(void *_arg)
|
|
|
|
-{
|
|
|
|
- struct callback_arg *arg = _arg;
|
|
|
|
-
|
|
|
|
- unsigned k = arg->k;
|
|
|
|
- unsigned j = arg->j;
|
|
|
|
-
|
|
|
|
- /* Find all the nodes potentially requiring this block */
|
|
|
|
- int rank_mask[world_size];
|
|
|
|
- find_nodes_using_12(k, j, rank_mask);
|
|
|
|
- rank_mask[rank] = 0;
|
|
|
|
-
|
|
|
|
- /* Send the block to those nodes */
|
|
|
|
- starpu_data_handle_t block_handle = STARPU_PLU(get_block_handle)(k, j);
|
|
|
|
- starpu_tag_t tag = TAG12_SAVE(k, j);
|
|
|
|
- int mpi_tag = MPI_TAG12(k, j);
|
|
|
|
- send_data_to_mask(block_handle, rank_mask, mpi_tag, tag);
|
|
|
|
-
|
|
|
|
- free(arg);
|
|
|
|
-}
|
|
|
|
-
|
|
|
|
-static void create_task_12_real(unsigned k, unsigned j)
|
|
|
|
-{
|
|
|
|
- struct starpu_task *task = create_task(TAG12(k, j));
|
|
|
|
-
|
|
|
|
-#warning temporary fix :/
|
|
|
|
-// task->cl = &STARPU_PLU(cl12);
|
|
|
|
- task->cl = &STARPU_PLU(cl21);
|
|
|
|
-
|
|
|
|
- task->cl_arg = create_debug_info(j, j, k);
|
|
|
|
-
|
|
|
|
- unsigned diag_block_is_local = (get_block_rank(k, k) == rank);
|
|
|
|
-
|
|
|
|
- starpu_tag_t tag_11_dep;
|
|
|
|
-
|
|
|
|
- /* which sub-data is manipulated ? */
|
|
|
|
- starpu_data_handle_t diag_block;
|
|
|
|
- if (diag_block_is_local)
|
|
|
|
- {
|
|
|
|
- diag_block = STARPU_PLU(get_block_handle)(k, k);
|
|
|
|
- tag_11_dep = TAG11(k);
|
|
|
|
- }
|
|
|
|
- else
|
|
|
|
- {
|
|
|
|
-#ifdef SINGLE_TMP11
|
|
|
|
- diag_block = STARPU_PLU(get_tmp_11_block_handle)();
|
|
|
|
-#else
|
|
|
|
- diag_block = STARPU_PLU(get_tmp_11_block_handle)(k);
|
|
|
|
-#endif
|
|
|
|
- tag_11_dep = TAG11_SAVE(k);
|
|
|
|
- }
|
|
|
|
-
|
|
|
|
- task->handles[0] = diag_block;
|
|
|
|
- task->handles[1] = STARPU_PLU(get_block_handle)(k, j);
|
|
|
|
-
|
|
|
|
- STARPU_ASSERT(get_block_rank(k, j) == rank);
|
|
|
|
-
|
|
|
|
- STARPU_ASSERT(task->handles[0] != STARPU_POISON_PTR);
|
|
|
|
- STARPU_ASSERT(task->handles[1] != STARPU_POISON_PTR);
|
|
|
|
-
|
|
|
|
- struct callback_arg *arg = malloc(sizeof(struct callback_arg));
|
|
|
|
- arg->j = j;
|
|
|
|
- arg->k = k;
|
|
|
|
-
|
|
|
|
- task->callback_func = callback_task_12_real;
|
|
|
|
- task->callback_arg = arg;
|
|
|
|
-
|
|
|
|
- if (!no_prio && (j == k+1)) {
|
|
|
|
- task->priority = STARPU_MAX_PRIO;
|
|
|
|
- }
|
|
|
|
-
|
|
|
|
- /* enforce dependencies ... */
|
|
|
|
- if (k > 0) {
|
|
|
|
- starpu_tag_declare_deps(TAG12(k, j), 2, tag_11_dep, TAG22(k-1, k, j));
|
|
|
|
- }
|
|
|
|
- else {
|
|
|
|
- starpu_tag_declare_deps(TAG12(k, j), 1, tag_11_dep);
|
|
|
|
- }
|
|
|
|
-
|
|
|
|
- int ret = starpu_task_submit(task);
|
|
|
|
- STARPU_CHECK_RETURN_VALUE(ret, "starpu_task_submit");
|
|
|
|
-}
|
|
|
|
-
|
|
|
|
-static void create_task_12(unsigned k, unsigned j)
|
|
|
|
-{
|
|
|
|
- if (get_block_rank(k, j) == rank)
|
|
|
|
- {
|
|
|
|
-#ifdef VERBOSE_INIT
|
|
|
|
- fprintf(stderr, "CREATE real task 12(k = %d, j = %d) on node %d\n", k, j, rank);
|
|
|
|
-#endif
|
|
|
|
- create_task_12_real(k, j);
|
|
|
|
- }
|
|
|
|
- else {
|
|
|
|
- /* We don't handle the task, but perhaps we have to generate MPI transfers. */
|
|
|
|
- int rank_mask[world_size];
|
|
|
|
- find_nodes_using_12(k, j, rank_mask);
|
|
|
|
-
|
|
|
|
- if (rank_mask[rank])
|
|
|
|
- {
|
|
|
|
-#ifdef VERBOSE_INIT
|
|
|
|
- fprintf(stderr, "create RECV task 12(k = %d, j = %d) on node %d\n", k, j, rank);
|
|
|
|
-#endif
|
|
|
|
- create_task_12_recv(k, j);
|
|
|
|
- }
|
|
|
|
- else {
|
|
|
|
-#ifdef VERBOSE_INIT
|
|
|
|
- fprintf(stderr, "Node %d needs not 12(k=%d, i=%d)\n", rank, k, j);
|
|
|
|
-#endif
|
|
|
|
- }
|
|
|
|
- }
|
|
|
|
-}
|
|
|
|
-
|
|
|
|
-/*
|
|
|
|
- * Task 21 (Update upper right (TRSM))
|
|
|
|
- */
|
|
|
|
-
|
|
|
|
-static void create_task_21_recv(unsigned k, unsigned i)
|
|
|
|
-{
|
|
|
|
- unsigned j;
|
|
|
|
-
|
|
|
|
- /* The current node is not computing that task, so we receive the block
|
|
|
|
- * with MPI */
|
|
|
|
-
|
|
|
|
- /* We don't issue a MPI receive request until everyone using the
|
|
|
|
- * temporary buffer is done : 21_(k-1)i can be used by 22_(k-1)ij with
|
|
|
|
- * j >= k */
|
|
|
|
- unsigned ndeps = 0;
|
|
|
|
- starpu_tag_t tag_array[nblocks];
|
|
|
|
-
|
|
|
|
-#ifdef SINGLE_TMP1221
|
|
|
|
- if (k > 0)
|
|
|
|
- for (j = (k-1)+1; j < nblocks; j++)
|
|
|
|
-#else
|
|
|
|
- if (k > 1)
|
|
|
|
- for (j = (k-2)+1; j < nblocks; j++)
|
|
|
|
-#endif
|
|
|
|
- {
|
|
|
|
- if (rank == get_block_rank(i, j))
|
|
|
|
-#ifdef SINGLE_TMP1221
|
|
|
|
- tag_array[ndeps++] = TAG22(k-1, i, j);
|
|
|
|
-#else
|
|
|
|
- tag_array[ndeps++] = TAG22(k-2, i, j);
|
|
|
|
-#endif
|
|
|
|
- }
|
|
|
|
-
|
|
|
|
- int source = get_block_rank(i, k);
|
|
|
|
-#ifdef SINGLE_TMP1221
|
|
|
|
- starpu_data_handle_t block_handle = STARPU_PLU(get_tmp_21_block_handle)(i);
|
|
|
|
-#else
|
|
|
|
- starpu_data_handle_t block_handle = STARPU_PLU(get_tmp_21_block_handle)(i, k);
|
|
|
|
-#endif
|
|
|
|
- int mpi_tag = MPI_TAG21(k, i);
|
|
|
|
- starpu_tag_t partial_tag = TAG21_SAVE_PARTIAL(k, i);
|
|
|
|
- starpu_tag_t unlocked_tag = TAG21_SAVE(k, i);
|
|
|
|
-
|
|
|
|
-// fprintf(stderr, "NODE %d - 21 (%d, %d) - recv when done ndeps %d - tag array %lx\n", rank, k, i, ndeps, tag_array[0]);
|
|
|
|
- receive_when_deps_are_done(ndeps, tag_array, source, mpi_tag, block_handle, partial_tag, unlocked_tag);
|
|
|
|
-}
|
|
|
|
-
|
|
|
|
-static void find_nodes_using_21(unsigned k, unsigned i, int *rank_mask)
|
|
|
|
-{
|
|
|
|
- memset(rank_mask, 0, world_size*sizeof(int));
|
|
|
|
-
|
|
|
|
- /* Block 21_ki is used to compute 22_kij with j > k */
|
|
|
|
- unsigned j;
|
|
|
|
- for (j = k+1; j < nblocks; j++)
|
|
|
|
- {
|
|
|
|
- int r = get_block_rank(i, j);
|
|
|
|
- rank_mask[r] = 1;
|
|
|
|
- }
|
|
|
|
-}
|
|
|
|
-
|
|
|
|
-static void callback_task_21_real(void *_arg)
|
|
|
|
-{
|
|
|
|
- struct callback_arg *arg = _arg;
|
|
|
|
-
|
|
|
|
- unsigned k = arg->k;
|
|
|
|
- unsigned i = arg->i;
|
|
|
|
-
|
|
|
|
- /* Find all the nodes potentially requiring this block */
|
|
|
|
- int rank_mask[world_size];
|
|
|
|
- find_nodes_using_21(k, i, rank_mask);
|
|
|
|
- rank_mask[rank] = 0;
|
|
|
|
-
|
|
|
|
- /* Send the block to those nodes */
|
|
|
|
- starpu_data_handle_t block_handle = STARPU_PLU(get_block_handle)(i, k);
|
|
|
|
- starpu_tag_t tag = TAG21_SAVE(k, i);
|
|
|
|
- int mpi_tag = MPI_TAG21(k, i);
|
|
|
|
- send_data_to_mask(block_handle, rank_mask, mpi_tag, tag);
|
|
|
|
-
|
|
|
|
- free(arg);
|
|
|
|
-}
|
|
|
|
-
|
|
|
|
-static void create_task_21_real(unsigned k, unsigned i)
|
|
|
|
-{
|
|
|
|
- struct starpu_task *task = create_task(TAG21(k, i));
|
|
|
|
-
|
|
|
|
-#warning temporary fix
|
|
|
|
-// task->cl = &STARPU_PLU(cl21);
|
|
|
|
- task->cl = &STARPU_PLU(cl12);
|
|
|
|
-
|
|
|
|
- task->cl_arg = create_debug_info(i, i, k);
|
|
|
|
-
|
|
|
|
- unsigned diag_block_is_local = (get_block_rank(k, k) == rank);
|
|
|
|
-
|
|
|
|
- starpu_tag_t tag_11_dep;
|
|
|
|
-
|
|
|
|
- /* which sub-data is manipulated ? */
|
|
|
|
- starpu_data_handle_t diag_block;
|
|
|
|
- if (diag_block_is_local)
|
|
|
|
- {
|
|
|
|
- diag_block = STARPU_PLU(get_block_handle)(k, k);
|
|
|
|
- tag_11_dep = TAG11(k);
|
|
|
|
- }
|
|
|
|
- else
|
|
|
|
- {
|
|
|
|
-#ifdef SINGLE_TMP11
|
|
|
|
- diag_block = STARPU_PLU(get_tmp_11_block_handle)();
|
|
|
|
-#else
|
|
|
|
- diag_block = STARPU_PLU(get_tmp_11_block_handle)(k);
|
|
|
|
-#endif
|
|
|
|
- tag_11_dep = TAG11_SAVE(k);
|
|
|
|
- }
|
|
|
|
-
|
|
|
|
- task->handles[0] = diag_block;
|
|
|
|
- task->handles[1] = STARPU_PLU(get_block_handle)(i, k);
|
|
|
|
-
|
|
|
|
- STARPU_ASSERT(task->handles[0] != STARPU_POISON_PTR);
|
|
|
|
- STARPU_ASSERT(task->handles[1] != STARPU_POISON_PTR);
|
|
|
|
-
|
|
|
|
- struct callback_arg *arg = malloc(sizeof(struct callback_arg));
|
|
|
|
- arg->i = i;
|
|
|
|
- arg->k = k;
|
|
|
|
-
|
|
|
|
- task->callback_func = callback_task_21_real;
|
|
|
|
- task->callback_arg = arg;
|
|
|
|
-
|
|
|
|
- if (!no_prio && (i == k+1)) {
|
|
|
|
- task->priority = STARPU_MAX_PRIO;
|
|
|
|
- }
|
|
|
|
-
|
|
|
|
- /* enforce dependencies ... */
|
|
|
|
- if (k > 0) {
|
|
|
|
- starpu_tag_declare_deps(TAG21(k, i), 2, tag_11_dep, TAG22(k-1, i, k));
|
|
|
|
- }
|
|
|
|
- else {
|
|
|
|
- starpu_tag_declare_deps(TAG21(k, i), 1, tag_11_dep);
|
|
|
|
- }
|
|
|
|
-
|
|
|
|
- int ret = starpu_task_submit(task);
|
|
|
|
- STARPU_CHECK_RETURN_VALUE(ret, "starpu_task_submit");
|
|
|
|
-}
|
|
|
|
-
|
|
|
|
-static void create_task_21(unsigned k, unsigned i)
|
|
|
|
-{
|
|
|
|
- if (get_block_rank(i, k) == rank)
|
|
|
|
- {
|
|
|
|
-#ifdef VERBOSE_INIT
|
|
|
|
- fprintf(stderr, "CREATE real task 21(k = %d, i = %d) on node %d\n", k, i, rank);
|
|
|
|
-#endif
|
|
|
|
- create_task_21_real(k, i);
|
|
|
|
- }
|
|
|
|
- else {
|
|
|
|
- /* We don't handle the task, but perhaps we have to generate MPI transfers. */
|
|
|
|
- int rank_mask[world_size];
|
|
|
|
- find_nodes_using_21(k, i, rank_mask);
|
|
|
|
-
|
|
|
|
- if (rank_mask[rank])
|
|
|
|
- {
|
|
|
|
-#ifdef VERBOSE_INIT
|
|
|
|
- fprintf(stderr, "create RECV task 21(k = %d, i = %d) on node %d\n", k, i, rank);
|
|
|
|
-#endif
|
|
|
|
- create_task_21_recv(k, i);
|
|
|
|
- }
|
|
|
|
- else {
|
|
|
|
-#ifdef VERBOSE_INIT
|
|
|
|
- fprintf(stderr, "Node %d needs not 21(k=%d, i=%d)\n", rank, k,i);
|
|
|
|
-#endif
|
|
|
|
- }
|
|
|
|
- }
|
|
|
|
-}
|
|
|
|
-
|
|
|
|
-/*
|
|
|
|
- * Task 22 (GEMM)
|
|
|
|
- */
|
|
|
|
-
|
|
|
|
-static void create_task_22_real(unsigned k, unsigned i, unsigned j)
|
|
|
|
-{
|
|
|
|
-// printf("task 22 k,i,j = %d,%d,%d TAG = %llx\n", k,i,j, TAG22(k,i,j));
|
|
|
|
-
|
|
|
|
- struct starpu_task *task = create_task(TAG22(k, i, j));
|
|
|
|
-
|
|
|
|
- task->cl = &STARPU_PLU(cl22);
|
|
|
|
-
|
|
|
|
- task->cl_arg = create_debug_info(i, j, k);
|
|
|
|
-
|
|
|
|
- /* which sub-data is manipulated ? */
|
|
|
|
-
|
|
|
|
- /* produced by TAG21_SAVE(k, i) */
|
|
|
|
- unsigned block21_is_local = (get_block_rank(i, k) == rank);
|
|
|
|
- starpu_tag_t tag_21_dep;
|
|
|
|
-
|
|
|
|
- starpu_data_handle_t block21;
|
|
|
|
- if (block21_is_local)
|
|
|
|
- {
|
|
|
|
- block21 = STARPU_PLU(get_block_handle)(i, k);
|
|
|
|
- tag_21_dep = TAG21(k, i);
|
|
|
|
- }
|
|
|
|
- else
|
|
|
|
- {
|
|
|
|
-#ifdef SINGLE_TMP1221
|
|
|
|
- block21 = STARPU_PLU(get_tmp_21_block_handle)(i);
|
|
|
|
-#else
|
|
|
|
- block21 = STARPU_PLU(get_tmp_21_block_handle)(i, k);
|
|
|
|
-#endif
|
|
|
|
- tag_21_dep = TAG21_SAVE(k, i);
|
|
|
|
- }
|
|
|
|
-
|
|
|
|
- /* produced by TAG12_SAVE(k, j) */
|
|
|
|
- unsigned block12_is_local = (get_block_rank(k, j) == rank);
|
|
|
|
- starpu_tag_t tag_12_dep;
|
|
|
|
-
|
|
|
|
- starpu_data_handle_t block12;
|
|
|
|
- if (block12_is_local)
|
|
|
|
- {
|
|
|
|
- // block12 = STARPU_PLU(get_block_handle)(j, k);
|
|
|
|
- block12 = STARPU_PLU(get_block_handle)(k, j);
|
|
|
|
- tag_12_dep = TAG12(k, j);
|
|
|
|
- }
|
|
|
|
- else
|
|
|
|
- {
|
|
|
|
-#ifdef SINGLE_TMP1221
|
|
|
|
- block12 = STARPU_PLU(get_tmp_12_block_handle)(j);
|
|
|
|
-#else
|
|
|
|
- block12 = STARPU_PLU(get_tmp_12_block_handle)(j, k);
|
|
|
|
-#endif
|
|
|
|
- tag_12_dep = TAG12_SAVE(k, j);
|
|
|
|
- }
|
|
|
|
-
|
|
|
|
-
|
|
|
|
-
|
|
|
|
-#warning temporary fix :/
|
|
|
|
- //task->handles[0] = block21;
|
|
|
|
- task->handles[0] = block12;
|
|
|
|
-
|
|
|
|
- //task->handles[1] = block12;
|
|
|
|
- task->handles[1] = block21;
|
|
|
|
-
|
|
|
|
- /* produced by TAG22(k-1, i, j) */
|
|
|
|
- task->handles[2] = STARPU_PLU(get_block_handle)(i, j);
|
|
|
|
-
|
|
|
|
- STARPU_ASSERT(task->handles[0] != STARPU_POISON_PTR);
|
|
|
|
- STARPU_ASSERT(task->handles[1] != STARPU_POISON_PTR);
|
|
|
|
- STARPU_ASSERT(task->handles[2] != STARPU_POISON_PTR);
|
|
|
|
-
|
|
|
|
- if (!no_prio && (i == k + 1) && (j == k +1) ) {
|
|
|
|
- task->priority = STARPU_MAX_PRIO;
|
|
|
|
- }
|
|
|
|
-
|
|
|
|
- /* enforce dependencies ... */
|
|
|
|
- if (k > 0) {
|
|
|
|
- starpu_tag_declare_deps(TAG22(k, i, j), 3, TAG22(k-1, i, j), tag_12_dep, tag_21_dep);
|
|
|
|
- }
|
|
|
|
- else {
|
|
|
|
- starpu_tag_declare_deps(TAG22(k, i, j), 2, tag_12_dep, tag_21_dep);
|
|
|
|
- }
|
|
|
|
-
|
|
|
|
- int ret = starpu_task_submit(task);
|
|
|
|
- STARPU_CHECK_RETURN_VALUE(ret, "starpu_task_submit");
|
|
|
|
-}
|
|
|
|
-
|
|
|
|
-static void create_task_22(unsigned k, unsigned i, unsigned j)
|
|
|
|
-{
|
|
|
|
- if (get_block_rank(i, j) == rank)
|
|
|
|
- {
|
|
|
|
- // fprintf(stderr, "CREATE real task 22(k = %d, i = %d, j = %d) on node %d\n", k, i, j, rank);
|
|
|
|
- create_task_22_real(k, i, j);
|
|
|
|
- }
|
|
|
|
-// else {
|
|
|
|
-// fprintf(stderr, "Node %d needs not 22(k=%d, i=%d, j = %d)\n", rank, k,i,j);
|
|
|
|
-// }
|
|
|
|
-}
|
|
|
|
-
|
|
|
|
-static void wait_tag_and_fetch_handle(starpu_tag_t tag, starpu_data_handle_t handle)
|
|
|
|
-{
|
|
|
|
- STARPU_ASSERT(handle != STARPU_POISON_PTR);
|
|
|
|
-
|
|
|
|
- starpu_tag_wait(tag);
|
|
|
|
-// fprintf(stderr, "Rank %d : tag %lx is done\n", rank, tag);
|
|
|
|
-
|
|
|
|
- starpu_data_acquire(handle, STARPU_R);
|
|
|
|
-
|
|
|
|
-// starpu_data_unregister(handle);
|
|
|
|
-}
|
|
|
|
-
|
|
|
|
-static void wait_termination(void)
|
|
|
|
-{
|
|
|
|
- unsigned k, i, j;
|
|
|
|
- for (k = 0; k < nblocks; k++)
|
|
|
|
- {
|
|
|
|
- /* Wait task 11k if needed */
|
|
|
|
- if (get_block_rank(k, k) == rank)
|
|
|
|
- {
|
|
|
|
- starpu_data_handle_t diag_block = STARPU_PLU(get_block_handle)(k, k);
|
|
|
|
- wait_tag_and_fetch_handle(TAG11_SAVE(k), diag_block);
|
|
|
|
- }
|
|
|
|
-
|
|
|
|
-
|
|
|
|
- for (i = k + 1; i < nblocks; i++)
|
|
|
|
- {
|
|
|
|
- /* Wait task 21ki if needed */
|
|
|
|
- if (get_block_rank(i, k) == rank)
|
|
|
|
- {
|
|
|
|
- starpu_data_handle_t block21 = STARPU_PLU(get_block_handle)(i, k);
|
|
|
|
- //starpu_data_handle_t block21 = STARPU_PLU(get_block_handle)(k, i);
|
|
|
|
- //fprintf(stderr, "BLOCK21 i %d k %d -> handle %p\n", i, k, block21);
|
|
|
|
- wait_tag_and_fetch_handle(TAG21_SAVE(k, i), block21);
|
|
|
|
- }
|
|
|
|
- }
|
|
|
|
-
|
|
|
|
- for (j = k + 1; j < nblocks; j++)
|
|
|
|
- {
|
|
|
|
- /* Wait task 12kj if needed */
|
|
|
|
- if (get_block_rank(k, j) == rank)
|
|
|
|
- {
|
|
|
|
- //starpu_data_handle_t block12 = STARPU_PLU(get_block_handle)(j, k);
|
|
|
|
- starpu_data_handle_t block12 = STARPU_PLU(get_block_handle)(k, j);
|
|
|
|
- //fprintf(stderr, "BLOCK12 j %d k %d -> handle %p\n", j, k, block12);
|
|
|
|
- wait_tag_and_fetch_handle(TAG12_SAVE(k, j), block12);
|
|
|
|
- }
|
|
|
|
- }
|
|
|
|
- }
|
|
|
|
-}
|
|
|
|
-
|
|
|
|
-/*
|
|
|
|
- * code to bootstrap the factorization
|
|
|
|
- */
|
|
|
|
-
|
|
|
|
-double STARPU_PLU(plu_main)(unsigned _nblocks, int _rank, int _world_size)
|
|
|
|
-{
|
|
|
|
- struct timeval start;
|
|
|
|
- struct timeval end;
|
|
|
|
-
|
|
|
|
- nblocks = _nblocks;
|
|
|
|
- rank = _rank;
|
|
|
|
- world_size = _world_size;
|
|
|
|
-
|
|
|
|
- /* create all the DAG nodes */
|
|
|
|
- unsigned i,j,k;
|
|
|
|
-
|
|
|
|
- for (k = 0; k < nblocks; k++)
|
|
|
|
- {
|
|
|
|
- create_task_11(k);
|
|
|
|
-
|
|
|
|
- for (i = k+1; i<nblocks; i++)
|
|
|
|
- {
|
|
|
|
- create_task_12(k, i);
|
|
|
|
- create_task_21(k, i);
|
|
|
|
- }
|
|
|
|
-
|
|
|
|
- for (i = k+1; i<nblocks; i++)
|
|
|
|
- {
|
|
|
|
- for (j = k+1; j<nblocks; j++)
|
|
|
|
- {
|
|
|
|
- create_task_22(k, i, j);
|
|
|
|
- }
|
|
|
|
- }
|
|
|
|
- }
|
|
|
|
-
|
|
|
|
- int barrier_ret = starpu_mpi_barrier(MPI_COMM_WORLD);
|
|
|
|
- STARPU_ASSERT(barrier_ret == MPI_SUCCESS);
|
|
|
|
-
|
|
|
|
- /* schedule the codelet */
|
|
|
|
- gettimeofday(&start, NULL);
|
|
|
|
-
|
|
|
|
- starpu_tag_notify_from_apps(STARPU_TAG_INIT);
|
|
|
|
-
|
|
|
|
- wait_termination();
|
|
|
|
-
|
|
|
|
- gettimeofday(&end, NULL);
|
|
|
|
-
|
|
|
|
- double timing = (double)((end.tv_sec - start.tv_sec)*1000000 + (end.tv_usec - start.tv_usec));
|
|
|
|
-
|
|
|
|
-// fprintf(stderr, "RANK %d -> took %f ms\n", rank, timing/1000);
|
|
|
|
-
|
|
|
|
- return timing;
|
|
|
|
-}
|
|
|