| # LLM model | |
| api_key: "" | |
| model_id: http://localhost:8040/v1 | |
| temperature: 1.0 | |
| # dataloader | |
| statis_path: "/wekafs/zihao/exp/agent/jianghui_triton/TB-eval/tb_eval/data/TritonBench/data/TritonBench_G_comp_alpac_v1_fixed_with_difficulty.json" | |
| py_folder: "/wekafs/zihao/exp/agent/jianghui_triton/TB-eval/tb_eval/data/TritonBench/data/TritonBench_G_v1" | |
| instruction_path: "/wekafs/zihao/exp/agent/jianghui_triton/TB-eval/tb_eval/data/TritonBench/data/TritonBench_G_comp_alpac_v1_fixed_with_difficulty.json" | |
| corpus_path: "/wekafs/zihao/exp/agent/jianghui_triton/GEAK-agent/src/dataloaders/TB_eval/train_crawl.json" | |
| golden_metrics: "/wekafs/zihao/exp/agent/jianghui_triton/TB-eval/tb_eval/data/TritonBench/performance_metrics/perf_G/golden_metrics" | |
| perf_ref_folder: null | |
| perf_G_path: "" | |
| py_interpreter: "python" | |
| # agent | |
| result_path: null #Resuming from Checkpoints | |
| mem_file: null #Resuming from Checkpoints | |
| start_iter: 0 | |
| start_idx: 0 | |
| datalen: 4 | |
| output_path: "/wekafs/zihao/exp/agent/jianghui_triton/GEAK-agent/outputs/outputs/1107/triton_1110_debug.jsonl" | |
| max_iteration: 10 | |
| # multi_thread: true | |
| multi_thread: false | |
| ancestor_num: 5 #ancestor number of perf_candidates used to generate descendant | |
| descendant_num: 2 #offspring | |
| gpu_id: 7 | |
| descendant_debug: 1 # if descendant_debug of descendant pass execution test, then generate descendant_num of new codes. | |
| target_gpu: "MI200" | |
| profiling: false # use profiling info. | |