File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change 2828from argparse import ArgumentParser
2929from ast import literal_eval
3030from datetime import datetime
31+ import os
3132from pathlib import Path
3233from time import perf_counter
3334from copy import deepcopy
6768)
6869from kernel_tuner .strategies .wrapper import OptAlgWrapper
6970
71+ environment_key_parallel = "KERNEL_TUNER_PARALLEL"
72+
7073strategy_map = {
7174 "brute_force" : brute_force ,
7275 "random_sample" : random_sample ,
@@ -585,7 +588,7 @@ def tune_kernel(
585588 strategy_options = None ,
586589 cache = None ,
587590 metrics = None ,
588- simulation_mode = False ,
591+ simulation_mode = None ,
589592 parallel = None ,
590593 observers = None ,
591594 objective = None ,
@@ -662,6 +665,12 @@ def tune_kernel(
662665 # TODO: we could use the "match case" syntax when removing support for 3.9
663666 tuning_options .simulated_time = 0
664667
668+ # Get runner from environment if possible
669+ if parallel is None :
670+ parallel = bool (os .environ .get (environment_key_parallel ))
671+
672+
673+ # Create runner
665674 if parallel and simulation_mode :
666675 raise ValueError ("Enabling `parallel` and `simulation_mode` together is not supported" )
667676 elif simulation_mode :
You can’t perform that action at this time.
0 commit comments