Skip to content

Commit 69d7c3c

Browse files
reporter status debug message error + shorter dse run
1 parent 033c202 commit 69d7c3c

File tree

2 files changed

+12
-4
lines changed

2 files changed

+12
-4
lines changed

conf/experimental/aiconfigurator/test/dse_aiconfigurator_disagg.toml

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -34,16 +34,16 @@ isl = 4000
3434
osl = 500
3535

3636
[cmd_args.disagg]
37-
p_tp = [1, 2]
37+
p_tp = [1]
3838
p_pp = [1]
3939
p_dp = [1]
4040
p_bs = 1
4141
p_workers = [1, 2]
4242

43-
d_tp = [1, 2]
43+
d_tp = [1]
4444
d_pp = [1]
4545
d_dp = [1]
46-
d_bs = [8, 16]
46+
d_bs = [8]
4747
d_workers = [1, 2]
4848

4949
gemm_quant_mode = "fp8_block"

src/cloudai/workloads/aiconfig/standalone_command_gen_strategy.py

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,10 @@
2222
from pathlib import Path
2323
from typing import cast
2424

25+
import toml
26+
2527
from cloudai.core import CommandGenStrategy
28+
from cloudai.models.scenario import TestRunDetails
2629

2730
from .aiconfigurator import Agg, AiconfiguratorCmdArgs, AiconfiguratorTestDefinition, Disagg
2831

@@ -31,9 +34,14 @@ class AiconfiguratorStandaloneCommandGenStrategy(CommandGenStrategy):
3134
"""Generate a standalone command that invokes the Aiconfigurator predictor and writes JSON output."""
3235

3336
def store_test_run(self) -> None:
34-
return
37+
test_cmd, full_cmd = ("", "n/a")
38+
with (self.test_run.output_path / self.TEST_RUN_DUMP_FILE_NAME).open("w", encoding="utf-8") as f:
39+
trd = TestRunDetails.from_test_run(self.test_run, test_cmd=test_cmd, full_cmd=full_cmd)
40+
toml.dump(trd.model_dump(), f)
3541

3642
def gen_exec_command(self) -> str:
43+
self.store_test_run()
44+
3745
tdef: AiconfiguratorTestDefinition = cast(AiconfiguratorTestDefinition, self.test_run.test)
3846
args: AiconfiguratorCmdArgs = tdef.cmd_args
3947
out_dir = Path(self.test_run.output_path).resolve()

0 commit comments

Comments
 (0)