Skip to content

Commit 56facd2

Browse files
committed
Always profile root kernel as baseline regardless of min_depth setting
1 parent da46f0b commit 56facd2

File tree

1 file changed

+5
-6
lines changed

1 file changed

+5
-6
lines changed

nkigym/src/nkigym/search/search.py

Lines changed: 5 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -279,12 +279,11 @@ def _run_search(
279279
last_progress_count = 0
280280

281281
root_node = next(iter(graph.nodes.values()))
282-
if root_node.depth >= min_depth:
283-
_assign_variant_idx(root_node, depth_counts)
284-
nki_path, error = _save_node(root_node, ctx, lowering_errors)
285-
_verify_node(root_node, ctx.sim_kwargs, ctx.expected)
286-
_submit_to_pool(nki_path, error, ctx)
287-
qualifying.append(root_node.kernel)
282+
_assign_variant_idx(root_node, depth_counts)
283+
nki_path, error = _save_node(root_node, ctx, lowering_errors)
284+
_verify_node(root_node, ctx.sim_kwargs, ctx.expected)
285+
_submit_to_pool(nki_path, error, ctx)
286+
qualifying.append(root_node.kernel)
288287

289288
_emit_progress(ctx.report, root_node, graph, len(qualifying), min_depth, depth_dist)
290289

0 commit comments

Comments
 (0)