/* SPDX-License-Identifier: MIT */
/*
 * Copyright © 2021 Intel Corporation
 */

#ifndef _INTEL_SELFTEST_SCHEDULER_HELPERS_H_
#define _INTEL_SELFTEST_SCHEDULER_HELPERS_H_

#include <linux/types.h>

struct i915_request;
struct intel_engine_cs;
struct intel_gt;

struct intel_selftest_saved_policy {
        u32 flags;
        u32 reset;
        u64 timeslice;
        u64 preempt_timeout;
};

enum selftest_scheduler_modify {
        SELFTEST_SCHEDULER_MODIFY_NO_HANGCHECK = 0,
        SELFTEST_SCHEDULER_MODIFY_FAST_RESET,
};

struct intel_engine_cs *intel_selftest_find_any_engine(struct intel_gt *gt);
int intel_selftest_modify_policy(struct intel_engine_cs *engine,
                                 struct intel_selftest_saved_policy *saved,
                                 enum selftest_scheduler_modify modify_type);
int intel_selftest_restore_policy(struct intel_engine_cs *engine,
                                  struct intel_selftest_saved_policy *saved);
int intel_selftest_wait_for_rq(struct i915_request *rq);

#endif