tjh.dev
Linux kernel mirror (for testing)
git.kernel.org/pub/scm/linux/kernel/git/torvalds/linux.git
kernel
os
linux
1/* SPDX-License-Identifier: MIT */
2/*
3 * Copyright © 2023 Intel Corporation
4 */
5
6#ifndef _XE_GPU_SCHEDULER_H_
7#define _XE_GPU_SCHEDULER_H_
8
9#include "xe_gpu_scheduler_types.h"
10#include "xe_sched_job_types.h"
11
12int xe_sched_init(struct xe_gpu_scheduler *sched,
13 const struct drm_sched_backend_ops *ops,
14 const struct xe_sched_backend_ops *xe_ops,
15 struct workqueue_struct *submit_wq,
16 uint32_t hw_submission, unsigned hang_limit,
17 long timeout, struct workqueue_struct *timeout_wq,
18 atomic_t *score, const char *name,
19 struct device *dev);
20void xe_sched_fini(struct xe_gpu_scheduler *sched);
21
22void xe_sched_submission_start(struct xe_gpu_scheduler *sched);
23void xe_sched_submission_stop(struct xe_gpu_scheduler *sched);
24void xe_sched_submission_stop_async(struct xe_gpu_scheduler *sched);
25
26void xe_sched_submission_resume_tdr(struct xe_gpu_scheduler *sched);
27
28void xe_sched_add_msg(struct xe_gpu_scheduler *sched,
29 struct xe_sched_msg *msg);
30void xe_sched_add_msg_locked(struct xe_gpu_scheduler *sched,
31 struct xe_sched_msg *msg);
32
33static inline void xe_sched_msg_lock(struct xe_gpu_scheduler *sched)
34{
35 spin_lock(&sched->base.job_list_lock);
36}
37
38static inline void xe_sched_msg_unlock(struct xe_gpu_scheduler *sched)
39{
40 spin_unlock(&sched->base.job_list_lock);
41}
42
43static inline void xe_sched_stop(struct xe_gpu_scheduler *sched)
44{
45 drm_sched_stop(&sched->base, NULL);
46}
47
48static inline void xe_sched_tdr_queue_imm(struct xe_gpu_scheduler *sched)
49{
50 drm_sched_tdr_queue_imm(&sched->base);
51}
52
53static inline void xe_sched_resubmit_jobs(struct xe_gpu_scheduler *sched)
54{
55 struct drm_sched_job *s_job;
56
57 list_for_each_entry(s_job, &sched->base.pending_list, list) {
58 struct drm_sched_fence *s_fence = s_job->s_fence;
59 struct dma_fence *hw_fence = s_fence->parent;
60
61 if (hw_fence && !dma_fence_is_signaled(hw_fence))
62 sched->base.ops->run_job(s_job);
63 }
64}
65
66static inline bool
67xe_sched_invalidate_job(struct xe_sched_job *job, int threshold)
68{
69 return drm_sched_invalidate_job(&job->drm, threshold);
70}
71
72static inline void xe_sched_add_pending_job(struct xe_gpu_scheduler *sched,
73 struct xe_sched_job *job)
74{
75 spin_lock(&sched->base.job_list_lock);
76 list_add(&job->drm.list, &sched->base.pending_list);
77 spin_unlock(&sched->base.job_list_lock);
78}
79
80static inline
81struct xe_sched_job *xe_sched_first_pending_job(struct xe_gpu_scheduler *sched)
82{
83 struct xe_sched_job *job;
84
85 spin_lock(&sched->base.job_list_lock);
86 job = list_first_entry_or_null(&sched->base.pending_list,
87 struct xe_sched_job, drm.list);
88 spin_unlock(&sched->base.job_list_lock);
89
90 return job;
91}
92
93static inline int
94xe_sched_entity_init(struct xe_sched_entity *entity,
95 struct xe_gpu_scheduler *sched)
96{
97 return drm_sched_entity_init(entity, 0,
98 (struct drm_gpu_scheduler **)&sched,
99 1, NULL);
100}
101
102#define xe_sched_entity_fini drm_sched_entity_fini
103
104#endif