Skip to content

Commit

Permalink
OSS CI skip if xnnpack is not used (pytorch#1125)
Browse files Browse the repository at this point in the history
Summary:
When we check for a model, we don't want jobs like test-models-macos (cmake, emformer_join, xnnpack, macos-m1-12) / macos-job which does nothing (non delegated nor quantized).

Pull Request resolved: pytorch#1125

Reviewed By: huydhn

Differential Revision: D50896620

Pulled By: kirklandsign

fbshipit-source-id: ebd905dc6f37577fd17994abadcb21d51c81b0da
  • Loading branch information
kirklandsign authored and facebook-github-bot committed Nov 1, 2023
1 parent fdce5db commit a8e05cc
Showing 1 changed file with 4 additions and 5 deletions.
9 changes: 4 additions & 5 deletions .ci/scripts/gather_test_models.py
Original file line number Diff line number Diff line change
Expand Up @@ -108,13 +108,12 @@ def export_models_for_ci() -> dict[str, dict]:
continue

if backend == "xnnpack":
if (
name in MODEL_NAME_TO_OPTIONS
and MODEL_NAME_TO_OPTIONS[name].quantization
):
if name not in MODEL_NAME_TO_OPTIONS:
continue
if MODEL_NAME_TO_OPTIONS[name].quantization:
backend += "-quantization"

if name in MODEL_NAME_TO_OPTIONS and MODEL_NAME_TO_OPTIONS[name].delegation:
if MODEL_NAME_TO_OPTIONS[name].delegation:
backend += "-delegation"

record = {
Expand Down

0 comments on commit a8e05cc

Please sign in to comment.