import pathlib,subprocess,os,concurrent.futures,shlex
root=pathlib.Path('artifacts/regional-planning-20260909')
def run(batch,blocks):
 name=f'canonical-mlp-b{batch}-r{blocks}'
 cmd=['target/release/ipu-trivial-test','c600-init.ipucfg','--sdk','/srv/home/gc-sdk/poplar_sdk-ubuntu_20_04-3.4.0+1507-69d9d03fd8/poplar-ubuntu_20_04-3.4.0+73-aa67dd6164',
 '--workload','siglip-mlp-benchmark','--mlp-batch',str(batch),'--mlp-blocks',str(blocks),'--fp8-scale=-4','--regional-planning','--regional-evaluations','2',
 '--planning-region','0:3' if blocks==1 else '0:1','--benchmark-selection',str(root/(name+'.json'))]
 (root/(name+'.command')).write_text(shlex.join(cmd)+'\n')
 with (root/(name+'.log')).open('w') as f:p=subprocess.run(cmd,env=dict(os.environ,RUST_LOG='info'),stdout=f,stderr=f)
 print(name,p.returncode,flush=True)
with concurrent.futures.ThreadPoolExecutor(2) as pool:list(pool.map(lambda a:run(*a),[(1,1),(2,3)]))