src/instr/instr_TI_trace.cpp

   1 /* Copyright (c) 2010-2015. 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 "src/instr/instr_private.h"
   8 #include "xbt/virtu.h" /* sg_cmdline */
   9 #include "xbt/xbt_os_time.h"
  10 #include "simgrid/sg_config.h"
  11
  12 #include <errno.h>
  13 #include <string.h>
  14 #include <sys/stat.h>
  15
  16 #ifdef WIN32
  17 #include <direct.h> // _mkdir
  18 #endif
  19
  20 XBT_LOG_NEW_DEFAULT_SUBCATEGORY(instr_TI_trace, instr_trace, "tracing event system");
  21
  22 extern FILE *tracing_file;
  23 double prefix=0.0;
  24
  25 xbt_dict_t tracing_files = NULL;
  26
  27 extern s_instr_trace_writer_t active_writer;
  28
  29 void TRACE_TI_init(void)
  30 {
  31   active_writer.print_PushState = print_TIPushState;
  32   active_writer.print_CreateContainer=print_TICreateContainer;
  33   active_writer.print_DestroyContainer=print_TIDestroyContainer;
  34 }
  35
  36 void TRACE_TI_start(void)
  37 {
  38   char *filename = TRACE_get_filename();
  39   tracing_file = fopen(filename, "w");
  40   if (tracing_file == NULL) {
  41     THROWF(system_error, 1, "Tracefile %s could not be opened for writing.", filename);
  42   }
  43
  44   XBT_DEBUG("Filename %s is open for writing", filename);
  45
  46   /* output one line comment */
  47   dump_comment(TRACE_get_comment());
  48
  49   /* output comment file */
  50   dump_comment_file(TRACE_get_comment_file());
  51 }
  52
  53 void TRACE_TI_end(void)
  54 {
  55   xbt_dict_free(&tracing_files);
  56   fclose(tracing_file);
  57   char *filename = TRACE_get_filename();
  58   XBT_DEBUG("Filename %s is closed", filename);
  59 }
  60
  61 void print_TICreateContainer(paje_event_t event)
  62 {
  63   //if we are in the mode with only one file
  64   static FILE *temp = NULL;
  65
  66   if (tracing_files == NULL) {
  67     tracing_files = xbt_dict_new_homogeneous(NULL);
  68     //generate unique run id with time
  69     prefix = xbt_os_time();
  70   }
  71
  72   if (!xbt_cfg_get_boolean(_sg_cfg_set, "tracing/smpi/format/ti_one_file") || temp == NULL) {
  73     char *folder_name = bprintf("%s_files", TRACE_get_filename());
  74     char *filename = bprintf("%s/%f_%s.txt", folder_name, prefix, ((createContainer_t) event->data)->container->name);
  75 #ifdef WIN32
  76     _mkdir(folder_name);
  77 #else
  78     mkdir(folder_name, S_IRWXU | S_IRWXG | S_IRWXO);
  79 #endif
  80     temp = fopen(filename, "w");
  81     if (temp == NULL)
  82       xbt_die("Tracefile %s could not be opened for writing: %s", filename, strerror(errno));
  83     fprintf(tracing_file, "%s\n", filename);
  84
  85     xbt_free(folder_name);
  86     xbt_free(filename);
  87   }
  88
  89   xbt_dict_set(tracing_files, ((createContainer_t) event->data)->container->name, (void *) temp, NULL);
  90 }
  91
  92 void print_TIDestroyContainer(paje_event_t event)
  93 {
  94   if (!xbt_cfg_get_boolean(_sg_cfg_set, "tracing/smpi/format/ti_one_file")|| xbt_dict_length(tracing_files) == 1) {
  95     FILE* f = (FILE*)xbt_dict_get_or_null(tracing_files, ((destroyContainer_t) event->data)->container->name);
  96     fclose(f);
  97   }
  98   xbt_dict_remove(tracing_files, ((destroyContainer_t) event->data)->container->name);
  99 }
 100
 101 void print_TIPushState(paje_event_t event)
 102 {
 103   int i;
 104
 105   //char* function=NULL;
 106   if (((pushState_t) event->data)->extra == NULL)
 107     return;
 108   instr_extra_data extra = (instr_extra_data) (((pushState_t) event->data)->extra);
 109
 110   char *process_id = NULL;
 111   //FIXME: dirty extract "rank-" from the name, as we want the bare process id here
 112   if (strstr(((pushState_t) event->data)->container->name, "rank-") == NULL)
 113     process_id = xbt_strdup(((pushState_t) event->data)->container->name);
 114   else
 115     process_id = xbt_strdup(((pushState_t) event->data)->container->name + 5);
 116
 117   FILE* trace_file =  (FILE* )xbt_dict_get(tracing_files, ((pushState_t) event->data)->container->name);
 118
 119   switch (extra->type) {
 120   case TRACING_INIT:
 121     fprintf(trace_file, "%s init\n", process_id);
 122     break;
 123   case TRACING_FINALIZE:
 124     fprintf(trace_file, "%s finalize\n", process_id);
 125     break;
 126   case TRACING_SEND:
 127     fprintf(trace_file, "%s send %d %d %s\n", process_id, extra->dst, extra->send_size, extra->datatype1);
 128     break;
 129   case TRACING_ISEND:
 130     fprintf(trace_file, "%s isend %d %d %s\n", process_id, extra->dst, extra->send_size, extra->datatype1);
 131     break;
 132   case TRACING_RECV:
 133     fprintf(trace_file, "%s recv %d %d %s\n", process_id, extra->src, extra->send_size, extra->datatype1);
 134     break;
 135   case TRACING_IRECV:
 136     fprintf(trace_file, "%s irecv %d %d %s\n", process_id, extra->src, extra->send_size, extra->datatype1);
 137     break;
 138   case TRACING_TEST:
 139     fprintf(trace_file, "%s test\n", process_id);
 140     break;
 141   case TRACING_WAIT:
 142     fprintf(trace_file, "%s wait\n", process_id);
 143     break;
 144   case TRACING_WAITALL:
 145     fprintf(trace_file, "%s waitall\n", process_id);
 146     break;
 147   case TRACING_BARRIER:
 148     fprintf(trace_file, "%s barrier\n", process_id);
 149     break;
 150   case TRACING_BCAST:          // rank bcast size (root) (datatype)
 151     fprintf(trace_file, "%s bcast %d ", process_id, extra->send_size);
 152     if (extra->root != 0 || (extra->datatype1 && strcmp(extra->datatype1, "")))
 153       fprintf(trace_file, "%d %s", extra->root, extra->datatype1);
 154     fprintf(trace_file, "\n");
 155     break;
 156   case TRACING_REDUCE:         // rank reduce comm_size comp_size (root) (datatype)
 157     fprintf(trace_file, "%s reduce %d %f ", process_id, extra->send_size, extra->comp_size);
 158     if (extra->root != 0 || (extra->datatype1 && strcmp(extra->datatype1, "")))
 159       fprintf(trace_file, "%d %s", extra->root, extra->datatype1);
 160     fprintf(trace_file, "\n");
 161     break;
 162   case TRACING_ALLREDUCE:      // rank allreduce comm_size comp_size (datatype)
 163     fprintf(trace_file, "%s allreduce %d %f %s\n", process_id, extra->send_size, extra->comp_size, extra->datatype1);
 164     break;
 165   case TRACING_ALLTOALL:       // rank alltoall send_size recv_size (sendtype) (recvtype)
 166     fprintf(trace_file, "%s alltoall %d %d %s %s\n", process_id, extra->send_size, extra->recv_size, extra->datatype1,
 167             extra->datatype2);
 168     break;
 169   case TRACING_ALLTOALLV:      // rank alltoallv send_size [sendcounts] recv_size [recvcounts] (sendtype) (recvtype)
 170     fprintf(trace_file, "%s alltoallv %d ", process_id, extra->send_size);
 171     for (i = 0; i < extra->num_processes; i++)
 172       fprintf(trace_file, "%d ", extra->sendcounts[i]);
 173     fprintf(trace_file, "%d ", extra->recv_size);
 174     for (i = 0; i < extra->num_processes; i++)
 175       fprintf(trace_file, "%d ", extra->recvcounts[i]);
 176     fprintf(trace_file, "%s %s \n", extra->datatype1, extra->datatype2);
 177     break;
 178   case TRACING_GATHER:         // rank gather send_size recv_size root (sendtype) (recvtype)
 179     fprintf(trace_file, "%s gather %d %d %d %s %s\n", process_id, extra->send_size, extra->recv_size, extra->root,
 180             extra->datatype1, extra->datatype2);
 181     break;
 182   case TRACING_ALLGATHERV:     // rank allgatherv send_size [recvcounts] (sendtype) (recvtype)
 183     fprintf(trace_file, "%s allgatherv %d ", process_id, extra->send_size);
 184     for (i = 0; i < extra->num_processes; i++)
 185       fprintf(trace_file, "%d ", extra->recvcounts[i]);
 186     fprintf(trace_file, "%s %s \n", extra->datatype1, extra->datatype2);
 187     break;
 188   case TRACING_REDUCE_SCATTER: // rank reducescatter [recvcounts] comp_size (sendtype)
 189     fprintf(trace_file, "%s reducescatter ", process_id);
 190     for (i = 0; i < extra->num_processes; i++)
 191       fprintf(trace_file, "%d ", extra->recvcounts[i]);
 192     fprintf(trace_file, "%f %s\n", extra->comp_size, extra->datatype1);
 193     break;
 194   case TRACING_COMPUTING:
 195     fprintf(trace_file, "%s compute %f\n", process_id, extra->comp_size);
 196     break;
 197   case TRACING_SLEEPING:
 198     fprintf(trace_file, "%s sleep %f\n", process_id, extra->sleep_duration);
 199     break;
 200   case TRACING_GATHERV: // rank gatherv send_size [recvcounts] root (sendtype) (recvtype)
 201     fprintf(trace_file, "%s gatherv %d ", process_id, extra->send_size);
 202     for (i = 0; i < extra->num_processes; i++)
 203       fprintf(trace_file, "%d ", extra->recvcounts[i]);
 204     fprintf(trace_file, "%d %s %s\n", extra->root, extra->datatype1, extra->datatype2);
 205     break;
 206   case TRACING_WAITANY:
 207   case TRACING_SENDRECV:
 208   case TRACING_SCATTER:
 209   case TRACING_SCATTERV:
 210   case TRACING_ALLGATHER:
 211   case TRACING_SCAN:
 212   case TRACING_EXSCAN:
 213   case TRACING_COMM_SIZE:
 214   case TRACING_COMM_SPLIT:
 215   case TRACING_COMM_DUP:
 216   case TRACING_SSEND:
 217   case TRACING_ISSEND:
 218   default:
 219     XBT_WARN ("Call from %s impossible to translate into replay command : Not implemented (yet)",
 220          ((pushState_t) event->data)->value->name);
 221     break;
 222   }
 223
 224   if (extra->recvcounts != NULL)
 225     xbt_free(extra->recvcounts);
 226   if (extra->sendcounts != NULL)
 227     xbt_free(extra->sendcounts);
 228   xbt_free(process_id);
 229   xbt_free(extra);
 230 }