gecko-dev/taskcluster/gecko_taskgraph/optimize/backstop.py
Greg Mierzwinski 10025f7494 Bug 1909210 - Add new optimization strategy for android perf tasks. r=taskgraph-reviewers,jcristau
This patch adds a new backstop optimization strategy for android perf tasks. It will decrease the scheduling frequency of the tests on autoland to about half as often as the original backstop strategy.

Differential Revision: https://phabricator.services.mozilla.com/D217269
2024-07-24 12:25:28 +00:00

55 lines
1.8 KiB
Python

# This Source Code Form is subject to the terms of the Mozilla Public
# License, v. 2.0. If a copy of the MPL was not distributed with this
# file, You can obtain one at http://mozilla.org/MPL/2.0/.
from taskgraph.optimize.base import All, OptimizationStrategy, register_strategy
from gecko_taskgraph.util.backstop import BACKSTOP_PUSH_INTERVAL
@register_strategy("skip-unless-android-perftest-backstop")
class SkipUnlessAndroidPerftestBackstop(OptimizationStrategy):
"""Always removes tasks except on backstop pushes."""
def should_remove_task(self, task, params, _):
return not params["android_perftest_backstop"]
@register_strategy("skip-unless-backstop")
class SkipUnlessBackstop(OptimizationStrategy):
"""Always removes tasks except on backstop pushes."""
def should_remove_task(self, task, params, _):
return not params["backstop"]
class SkipUnlessPushInterval(OptimizationStrategy):
"""Always removes tasks except every N pushes.
Args:
push_interval (int): Number of pushes
"""
def __init__(self, push_interval, remove_on_projects=None):
self.push_interval = push_interval
@property
def description(self):
return f"skip-unless-push-interval-{self.push_interval}"
def should_remove_task(self, task, params, _):
# On every Nth push, want to run all tasks.
return int(params["pushlog_id"]) % self.push_interval != 0
# Strategy to run tasks on "expanded" pushes, currently defined as pushes that
# are half the backstop interval. The 'All' composite strategy means that the
# "backstop" strategy will prevent "expanded" from applying on backstop pushes.
register_strategy(
"skip-unless-expanded",
args=(
"skip-unless-backstop",
SkipUnlessPushInterval(BACKSTOP_PUSH_INTERVAL / 2),
),
)(All)