Skip to content

Commit ebba6e6

Browse files
committed
fix test failures
Signed-off-by: Will Guo <willg@nvidia.com>
1 parent a13c957 commit ebba6e6

File tree

3 files changed

+576
-41
lines changed

3 files changed

+576
-41
lines changed

modelopt/onnx/quantization/autotune/autotuner.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -50,12 +50,12 @@
5050
_MUTATION_SPECS = [
5151
("node_inputs", "node input points", lambda p: (p.node_index, p.input_index)),
5252
(
53-
"region_composite_inputs",
53+
"child_region_inputs",
5454
"region composite points",
5555
lambda p: (p.region_index, p.input_index),
5656
),
5757
(
58-
"region_output_points",
58+
"region_outputs",
5959
"region output points",
6060
lambda p: (p.region_index, p.node_index, p.output_index),
6161
),
@@ -1224,7 +1224,7 @@ def initialize(
12241224
self._search_regions()
12251225

12261226
@staticmethod
1227-
def _visit_region_recursively(self, region: Region) -> list[Region]:
1227+
def _visit_region_recursively(region: Region) -> list[Region]:
12281228
"""Recursively traverse region hierarchy and collect all regions.
12291229
12301230
Performs depth-first traversal of the region tree starting from a given
@@ -1240,7 +1240,7 @@ def _visit_region_recursively(self, region: Region) -> list[Region]:
12401240
regions = [region]
12411241

12421242
for child in region.get_children():
1243-
regions.extend(self._visit_region_recursively(child))
1243+
regions.extend(QDQAutotuner._visit_region_recursively(child))
12441244

12451245
return regions
12461246

@@ -1291,7 +1291,7 @@ def _search_regions(self) -> None:
12911291
# Flatten the hierarchy into a list of all regions
12921292
all_regions = []
12931293
for region in self.regions:
1294-
all_regions.extend(self._visit_region_recursively(region))
1294+
all_regions.extend(QDQAutotuner._visit_region_recursively(region))
12951295

12961296
all_regions.sort(key=lambda r: r.type != RegionType.LEAF)
12971297
self.regions = all_regions

0 commit comments

Comments
 (0)