diff --git a/tests/python/meta_schedule/run_meta_schedule.sh b/tests/python/meta_schedule/run_meta_schedule.sh index 6afedc6a565af..fa0c7ca42562f 100644 --- a/tests/python/meta_schedule/run_meta_schedule.sh +++ b/tests/python/meta_schedule/run_meta_schedule.sh @@ -1,4 +1,4 @@ -set -euxo pipefail +# set -euxo pipefail RPC_HOST="192.168.6.66" RPC_PORT="4445" @@ -17,7 +17,7 @@ run () { --rpc-host "$RPC_HOST" \ --rpc-port "$RPC_PORT" \ --rpc-key "$RPC_KEY" \ - --num-trials 1500 \ + --num-trials 5000 \ 2>&1 | tee "$LOG_DIR/$name.log" } diff --git a/tests/python/meta_schedule/test_tune_te_cpu.py b/tests/python/meta_schedule/test_tune_te_cpu.py index 7e255f5baa341..b48fc4f9a04ce 100644 --- a/tests/python/meta_schedule/test_tune_te_cpu.py +++ b/tests/python/meta_schedule/test_tune_te_cpu.py @@ -87,6 +87,7 @@ def main(): num_trials_total=ARGS.num_trials, ), runner=runner, + task_name=ARGS.workload, ) if sch is None: print("No valid schedule found!")