1234567891011121314151617181920212223242526272829303132333435 |
- import os
- import subprocess
- SENDS = 2**20
- USERS = 2**13
- THREADS = 48
- D_MAPS = 15
- RUNS = 10
- WARMUP = 0
- DATA_DIR = os.path.join(os.getcwd(), "data", "d-scaling-store")
- os.makedirs(DATA_DIR, exist_ok=True)
- SPARTA_D_DIR = os.path.join(os.getcwd(), "sparta-d")
- SPARTA_D_FILE = os.path.join(
- DATA_DIR, f"sparta-d-{SENDS}-{USERS}-{THREADS}.csv")
- def sparta_d_cmd(maps):
- cmd = ["cargo", "run", "--release", "--",
- str(SENDS), str(USERS), str(THREADS), str(USERS), str(maps), "-r", str(RUNS), "-w", str(WARMUP)]
- result = subprocess.run(cmd, capture_output=True,
- text=True, cwd=SPARTA_D_DIR)
- return result.stdout
- for maps in range(1, 16):
- print(maps)
- with open(SPARTA_D_FILE, "a") as sparta_d_file:
- output = sparta_d_cmd(maps)
- print("\tsparta_d:", output)
- sparta_d_file.write(output)
|