36 lines
862 B
Python
Executable File
36 lines
862 B
Python
Executable File
#!/usr/bin/env python
|
|
|
|
import copy
|
|
import json
|
|
|
|
from pathlib import Path
|
|
|
|
SIZE = [384, 384, 384]
|
|
|
|
with (Path(__file__).parent.parent / "templates" / "strong-cpu.json").open(encoding="utf8") as f:
|
|
template = json.load(f)
|
|
|
|
configs = [
|
|
[ 4, 4, 3],
|
|
[ 4, 4, 6],
|
|
[ 4, 4, 12],
|
|
[ 4, 8, 12],
|
|
[ 8, 8, 12],
|
|
[ 8, 8, 24],
|
|
[ 8, 16, 24],
|
|
[ 16, 16, 24],
|
|
[ 16, 16, 48]
|
|
]
|
|
|
|
out_path = Path(__file__).parent.parent / "generated" / "config"
|
|
|
|
for c in configs:
|
|
nc = copy.deepcopy(template)
|
|
nc["Geometry"]["blockcount"] = c
|
|
nc["Geometry"]["blocksize"] = [bs // bc for bc, bs in zip(c, SIZE)]
|
|
|
|
nc_out_path = out_path / f"strong-cpu-{c[0]:02}-{c[1]:02}-{c[2]:02}.json"
|
|
print(f"Dumping {(c[0] * c[1] * c[2]) // 48} to {nc_out_path}")
|
|
with nc_out_path.open("w", encoding="utf8") as f:
|
|
json.dump(nc, f)
|