File size: 611 Bytes
47709ed | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 | import json
import pathlib
from test_remote_quantum_gpu_v3 import benchmark
def main():
results = []
for batch in [4, 8, 16, 32, 64, 128]:
results.append(benchmark(batch=batch, n_qubits=8, depth=2, repeats=50))
payload = {
"experiment": "EXP-001-batch-sweep",
"results": results,
}
output = pathlib.Path(
"./artifacts/EXP-001-batch-sweep.json"
)
output.parent.mkdir(parents=True, exist_ok=True)
output.write_text(json.dumps(payload, indent=2), encoding="utf-8")
print(json.dumps(payload, indent=2))
if __name__ == "__main__":
main()
|