From 446fcc3510b488fbd5cfc855f9e0a85d06d49d59 Mon Sep 17 00:00:00 2001 From: Jayesh Badwaik <j.badwaik@fz-juelich.de> Date: Sat, 7 Dec 2024 22:12:30 +0100 Subject: [PATCH] + energy measurement included --- simple.yml | 10 ++++++++++ src/value.py | 13 ++++++++++++- 2 files changed, 22 insertions(+), 1 deletion(-) diff --git a/simple.yml b/simple.yml index 95809a8..c507f5d 100644 --- a/simple.yml +++ b/simple.yml @@ -119,6 +119,12 @@ patternset: pattern: - name: success _: $jube_pat_wrd + - name: energy_pattern + pattern: + - name: energy.start + _: 'energy_measure_start:$jube_pat_bl$jube_pat_fp' + - name: energy.stop + _: 'energy_measure_stop:$jube_pat_bl$jube_pat_fp' analyser: @@ -132,6 +138,8 @@ analyser: _: run.jobid - use: success_pattern _: success + - use: energy_pattern + _: energy.times result: use: @@ -150,3 +158,5 @@ result: - threadspertask - runtime - { title: "success", _: success } + - energy.start + - energy.end diff --git a/src/value.py b/src/value.py index aa19ebb..5c33c4a 100644 --- a/src/value.py +++ b/src/value.py @@ -3,6 +3,8 @@ import argparse import random from mpi4py import MPI +import time + def parser(): parser = argparse.ArgumentParser(description="Value of a number") @@ -10,6 +12,7 @@ def parser(): parser.add_argument("--input", type=str, help="Input file") return parser + # This is the main function if __name__ == "__main__": comm = MPI.COMM_WORLD @@ -20,5 +23,13 @@ if __name__ == "__main__": problem_size = float(data) * float(args.number) * 12 runtime = problem_size / nproc * random.uniform(0.8, 1.2) string = " " * random.randint(1, 10) + time.sleep(random.randint(10, 15)) + + with open("energy.times", "w") as g: + g.write("energy_measure_start: " + str(time.time()) + "\n") + if comm.Get_rank() == 0: - print("Runtime={}{}".format(string,runtime)) + print("Runtime={}{}".format(string, runtime)) + + with open("energy.times", "w") as g: + g.write("energy_measure_stop: " + str(time.time()) + "\n") -- GitLab