Skip to content

Commit baa8d87

Browse files
kirklandsignfacebook-github-bot
authored andcommitted
Update optimization recipe (#220)
Summary: XNNPACK delegation out-of-box experience is ready. Pull Request resolved: #220 Reviewed By: digantdesai, guangy10 Differential Revision: D49022184 fbshipit-source-id: 3fa24fc6852ff09208f5202d9e8aa97eee05c5ef
1 parent e045952 commit baa8d87

File tree

2 files changed

+9
-7
lines changed

2 files changed

+9
-7
lines changed

.ci/scripts/gather_test_models.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,8 @@
2020
RUNNERS = {
2121
# This one runs OOM on smaller runner, the root cause is unclear (T163016365)
2222
"w2l": "linux.12xlarge",
23+
"ic4": "linux.12xlarge",
24+
"resnet50": "linux.12xlarge",
2325
# This one causes timeout on smaller runner, the root cause is unclear (T161064121)
2426
"emformer_join": "linux.12xlarge",
2527
}

examples/recipes/xnnpack_optimization/models.py

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -17,13 +17,13 @@ class OptimizationOptions(object):
1717
"linear": OptimizationOptions(True, True),
1818
"add": OptimizationOptions(True, True),
1919
"add_mul": OptimizationOptions(True, True),
20-
"mv2": OptimizationOptions(True, True),
21-
"mv3": OptimizationOptions(False, True),
20+
"dl3": OptimizationOptions(True, True),
2221
"ic3": OptimizationOptions(True, False),
23-
"ic4": OptimizationOptions(
24-
True, False
25-
), # TODO[T163161310]: takes a long time to export to exec prog and save inception_v4 quantized model
22+
"ic4": OptimizationOptions(True, False),
23+
"mv2": OptimizationOptions(True, True),
24+
"mv3": OptimizationOptions(True, True),
25+
"resnet18": OptimizationOptions(True, True),
26+
"resnet50": OptimizationOptions(True, True),
27+
"vit": OptimizationOptions(False, True),
2628
"w2l": OptimizationOptions(False, True),
27-
"dl3": OptimizationOptions(True, False),
28-
"vit": OptimizationOptions(True, False),
2929
}

0 commit comments

Comments
 (0)