[Meta Schedule] Tweak experiment scripts (apache#539)

zxybazh · Jan 7, 2022 · e840635 · e840635
1 parent d22405b
commit e840635
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git 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
@@ -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!")