Skip to content

Commit db3abb3

Browse files
committed
Merge with searchspace_experiments
1 parent 874c390 commit db3abb3

File tree

1 file changed

+15
-15
lines changed

1 file changed

+15
-15
lines changed

kernel_tuner/searchspace.py

Lines changed: 15 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -144,21 +144,21 @@ def __init__(
144144
else:
145145
raise ValueError(f"Invalid framework parameter {framework}")
146146

147-
# get the solver given the solver method argument
148-
solver = ""
149-
if solver_method.lower() == "pc_backtrackingsolver":
150-
solver = BacktrackingSolver()
151-
elif solver_method.lower() == "pc_optimizedbacktrackingsolver":
152-
solver = OptimizedBacktrackingSolver(forwardcheck=False)
153-
elif solver_method.lower() == "pc_parallelsolver":
154-
raise NotImplementedError("ParallelSolver is not yet implemented")
155-
# solver = ParallelSolver()
156-
elif solver_method.lower() == "pc_recursivebacktrackingsolver":
157-
solver = RecursiveBacktrackingSolver()
158-
elif solver_method.lower() == "pc_minconflictssolver":
159-
solver = MinConflictsSolver()
160-
else:
161-
raise ValueError(f"Solver method {solver_method} not recognized.")
147+
# get the solver given the solver method argument
148+
solver = ""
149+
if solver_method.lower() == "pc_backtrackingsolver":
150+
solver = BacktrackingSolver()
151+
elif solver_method.lower() == "pc_optimizedbacktrackingsolver":
152+
solver = OptimizedBacktrackingSolver(forwardcheck=False)
153+
elif solver_method.lower() == "pc_parallelsolver":
154+
raise NotImplementedError("ParallelSolver is not yet implemented")
155+
# solver = ParallelSolver()
156+
elif solver_method.lower() == "pc_recursivebacktrackingsolver":
157+
solver = RecursiveBacktrackingSolver()
158+
elif solver_method.lower() == "pc_minconflictssolver":
159+
solver = MinConflictsSolver()
160+
else:
161+
raise ValueError(f"Solver method {solver_method} not recognized.")
162162

163163
# build the search space
164164
self.list, self.__dict, self.size = searchspace_builder(block_size_names, max_threads, solver)

0 commit comments

Comments
 (0)