examples/simdag/sd_typed_tasks_test.c

   1 /* Copyright (c) 2006-2010, 2012-2016. The SimGrid Team.
   2  * All rights reserved.                                                     */
   3
   4 /* This program is free software; you can redistribute it and/or modify it
   5  * under the terms of the license (GNU LGPL) which comes with this package. */
   6
   7 #include "simgrid/simdag.h"
   8 #include "xbt/log.h"
   9
  10 XBT_LOG_NEW_DEFAULT_CATEGORY(sd_typed_tasks_test, "Logging specific to this SimDag example");
  11
  12 int main(int argc, char **argv)
  13 {
  14   int i;
  15   unsigned int ctr;
  16   const sg_host_t *hosts;
  17   SD_task_t task, seq_comp1, e2e_comm, seq_comp2;
  18   SD_task_t par_comp1, redist, par_comp2, par_comp3;
  19   xbt_dynar_t changed_tasks;
  20
  21   double computation_amount[4];
  22   double communication_amount[16] = { 0 };
  23   sg_host_t host_list[4];
  24
  25   SD_init(&argc, argv);
  26
  27   xbt_assert(argc > 1, "Usage: %s platform_file\n\nExample: %s two_clusters.xml", argv[0], argv[0]);
  28   SD_create_environment(argv[1]);
  29
  30   hosts = sg_host_list();
  31
  32   /* creation of some typed tasks and their dependencies */
  33   seq_comp1 = SD_task_create_comp_seq("Seq. comp. 1", NULL, 1e9);
  34   e2e_comm = SD_task_create_comm_e2e("E2E comm.", NULL, 1e7);
  35   seq_comp2 = SD_task_create_comp_seq("Seq. comp 2.", NULL, 1e9);
  36   par_comp1 = SD_task_create_comp_par_amdahl("Par. Comp. 1", NULL, 1e9, 0.2);
  37   redist = SD_task_create_comm_par_mxn_1d_block("MxN redist", NULL, 1.2e8);
  38   par_comp2 = SD_task_create_comp_par_amdahl("Par. Comp. 2", NULL, 3e8, 0.5);
  39
  40   par_comp3 = SD_task_create("Par. Comp. 3", NULL, 1e9);
  41
  42   SD_task_dependency_add(NULL, NULL, seq_comp1, e2e_comm);
  43   SD_task_dependency_add(NULL, NULL, e2e_comm, seq_comp2);
  44
  45   SD_task_dependency_add(NULL, NULL, par_comp1, redist);
  46   SD_task_dependency_add(NULL, NULL, redist, par_comp2);
  47
  48   SD_task_schedulel(seq_comp1, 1, hosts[8]);
  49   SD_task_schedulel(seq_comp2, 1, hosts[9]);
  50
  51   SD_task_schedulev(par_comp1, 4, hosts);
  52   SD_task_schedulev(par_comp2, 3, hosts);
  53
  54   /* Let's unschedule these tasks and test the auto-scheduling in the opposite way. */
  55   SD_task_unschedule(par_comp1);
  56   SD_task_unschedule(par_comp2);
  57   SD_task_unschedule(redist); /* yes, it was scheduled too */
  58
  59   SD_task_schedulev(par_comp2, 3, hosts);
  60   SD_task_schedulev(par_comp1, 4, hosts);
  61
  62   for (i=0;i<4;i++){
  63     host_list[i]=hosts[i+4];
  64     /* Apply Amdahl's law manually assuming a 20% serial part */
  65     computation_amount[i]=(0.2 + (1 - 0.2)/4) * SD_task_get_amount(par_comp3);
  66   }
  67
  68   SD_task_schedule(par_comp3, 4, host_list, computation_amount, communication_amount, -1);
  69
  70   changed_tasks = SD_simulate(-1.0);
  71   xbt_dynar_foreach(changed_tasks, ctr, task) {
  72     XBT_INFO("Task '%s' start time: %f, finish time: %f", SD_task_get_name(task), SD_task_get_start_time(task),
  73              SD_task_get_finish_time(task));
  74   }
  75
  76   xbt_dynar_foreach(changed_tasks, ctr, task) {
  77     SD_task_destroy(task);
  78   }
  79   SD_exit();
  80   return 0;
  81 }