Commit 308899c4 authored by David Trudgian's avatar David Trudgian
Browse files

WIP: Working single task srun with spearmint

parent ef02c888
......@@ -53,6 +53,8 @@ class SrunExecutor(BaseExecutor):
p.close()
p.join()
return self.task_summaries[self.task_count - 1]
def _run_cmd(self, cmd, stderr_file, stdout_file):
"""SLURM execution of a command using srun.
......@@ -70,6 +72,8 @@ class SrunExecutor(BaseExecutor):
srun_cmd = "srun --exclusive -N1 -n1 --cpus-per-task=%d --distribution=cyclic %s" % (
cpus_per_task, cmd['__command']['value'])
logging.debug("srun command for task is: %s", srun_cmd)
ret = call(srun_cmd, shell=True, stderr=stderr_file,
stdout=stdout_file, env=os.environ)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment