src/bindings/lua/master_slave.lua

   1
   2 --Master Function
   3 function Master(...)
   4
   5 simgrid.info("Hello from lua, I'm the master")
   6 for i,v in ipairs(arg) do
   7     simgrid.info("Got "..v)
   8 end
   9
  10 nb_task = arg[1];
  11 comp_size = arg[2];
  12 comm_size = arg[3];
  13 slave_count = arg[4]
  14
  15
  16 simgrid.info("Argc="..(#arg).." (should be 4)")
  17
  18 -- Dispatch the tasks
  19
  20 for i=1,nb_task do
  21   tk = simgrid.Task.new("Task "..i,comp_size,comm_size);
  22   alias = "slave "..(i%slave_count);
  23   simgrid.info("Master sending  '" .. simgrid.Task.name(tk) .."' To '" .. alias .."'");
  24   simgrid.Task.send(tk,alias); -- C user data set to NULL
  25   simgrid.info("Master done sending '".. simgrid.Task.name(tk) .."' To '" .. alias .."'");
  26 end
  27
  28 -- Sending Finalize Message To Others
  29
  30 simgrid.info("Master: All tasks have been dispatched. Let's tell everybody the computation is over.");
  31 for i=0,slave_count-1 do
  32   alias = "slave "..i;
  33   simgrid.info("Master: sending finalize to "..alias);
  34   simgrid.Task.send(simgrid.Task.new("finalize",0,0),alias);
  35 end
  36 simgrid.info("Master: Everything's done.");
  37
  38
  39 end
  40
  41 -- Slave Function ---------------------------------------------------------
  42 function Slave(...)
  43
  44 my_mailbox="slave "..arg[1]
  45 simgrid.info("Hello from lua, I'm a poor slave with mbox: "..my_mailbox)
  46
  47
  48 while true do
  49 --  tk = simgrid.Task.new("",0,0); --??
  50 --  simgrid.Task.recv2(tk,my_mailbox);
  51   tk = simgrid.Task.recv(my_mailbox);
  52
  53   tk_name = simgrid.Task.name(tk)
  54
  55   if (tk_name == "finalize") then
  56     simgrid.info("Slave '" ..my_mailbox.."' got finalize msg");
  57     break
  58   end
  59
  60   simgrid.info("Slave '" ..my_mailbox.."' processing "..simgrid.Task.name(tk))
  61   simgrid.Task.execute(tk);
  62
  63   simgrid.info("Slave '" ..my_mailbox.."': task "..simgrid.Task.name(tk) .. " done")
  64 end -- while
  65
  66 simgrid.info("Slave '" ..my_mailbox.."': I'm Done . See You !!");
  67
  68 end -- function ----------------------------------------------------------
  69 --]]
  70
  71 function doyield()
  72     coroutine.yield()
  73 end
  74
  75 require "simgrid"
  76
  77 simgrid.platform("../../../examples/msg/small_platform.xml")
  78 simgrid.application("../ruby/deploy.xml")
  79 simgrid.run()
  80 simgrid.info("Simulation's over. See you.")
  81 simgrid.clean()