coosld/tig-monorepopool
1
0
Fork 0
mirror of https://github.com/tig-pool-nk/tig-monorepo.git synced 2026-02-21 17:27:21 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

improve error handling in tig-benchmarker @run_tig_runtime

Browse Source
This commit is contained in:
germ3n 2025-07-17 20:17:26 +01:00 committed by FiveMovesAhead
parent 6cbf3e6621
commit 78eae463a0
1 changed files with 15 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
tig-benchmarker/slave/main.py
View File

@ -80,15 +80,16 @@ def run_tig_runtime(nonce, batch, so_path, ptx_path, results_dir):
while True:
ret = process.poll()
if ret is not None:
# exit codes:
# 0 - success
# 82 - cuda out of memory
# 83 - host out of memory
# 84 - runtime error
# 85 - no solution
# 86 - invalid solution
# 87 - out of fuel
if (ret == 84 or ret == 87) and not os.path.exists(output_file):
exit_codes = {
0: "success",
82: "cuda out of memory",
83: "host out of memory",
84: "runtime error",
85: "no solution",
86: "invalid solution",
87: "out of fuel",
}
if (ret != 0 and ret in exit_codes) and not os.path.exists(output_file):
with open(output_file, "w") as f:
json.dump(dict(
nonce=nonce,
@ -98,8 +99,11 @@ def run_tig_runtime(nonce, batch, so_path, ptx_path, results_dir):
cpu_arch=CPU_ARCH
), f)
if ret not in {0, 84, 85, 86, 87}:
logger.error(f"batch {batch['id']}, nonce {nonce} failed with exit code {ret}: {process.stderr.read().decode()}")
if ret != 0:
if ret not in exit_codes:
logger.error(f"batch {batch['id']}, nonce {nonce} failed with exit code {ret}: {process.stderr.read().decode()}")
else:
logger.error(f"batch {batch['id']}, nonce {nonce} failed with exit code {ret}: {exit_codes[ret]}")
break