[trace] check action type before setting its tracing category

[simgrid.git] / src / simix / smx_host.c

diff --git a/src/simix/smx_host.c b/src/simix/smx_host.c
index 8bcbf47..6ab4350 100644 (file)
--- a/src/simix/smx_host.c
+++ b/src/simix/smx_host.c
@@ -395,7 +395,11 @@ void SIMIX_post_host_execute(smx_action_t action)
 #ifdef HAVE_TRACING
 void SIMIX_set_category(smx_action_t action, const char *category)
 {
 #ifdef HAVE_TRACING
 void SIMIX_set_category(smx_action_t action, const char *category)
 {

-  surf_workstation_model->set_category(action->execution.surf_exec, category);
+  if (action->type == SIMIX_ACTION_EXECUTE){
+    surf_workstation_model->set_category(action->execution.surf_exec, category);
+  }else if (action->type == SIMIX_ACTION_COMMUNICATE){
+    surf_workstation_model->set_category(action->comm.surf_comm, category);
+  }

 }
 #endif
 
 }
 #endif