Linux kernel mirror (for testing)
git.kernel.org/pub/scm/linux/kernel/git/torvalds/linux.git
tjh.dev
kernel
os
linux
1/* SPDX-License-Identifier: GPL-2.0-only */
2/* Copyright 2024-2025 Tomeu Vizoso <tomeu@tomeuvizoso.net> */
3
4#ifndef __ROCKET_JOB_H__
5#define __ROCKET_JOB_H__
6
7#include <drm/drm_drv.h>
8#include <drm/gpu_scheduler.h>
9
10#include "rocket_core.h"
11#include "rocket_drv.h"
12
13struct rocket_task {
14 u64 regcmd;
15 u32 regcmd_count;
16};
17
18struct rocket_job {
19 struct drm_sched_job base;
20
21 struct rocket_device *rdev;
22
23 struct drm_gem_object **in_bos;
24 struct drm_gem_object **out_bos;
25
26 u32 in_bo_count;
27 u32 out_bo_count;
28
29 struct rocket_task *tasks;
30 u32 task_count;
31 u32 next_task_idx;
32
33 /* Fence to be signaled by drm-sched once its done with the job */
34 struct dma_fence *inference_done_fence;
35
36 /* Fence to be signaled by IRQ handler when the job is complete. */
37 struct dma_fence *done_fence;
38
39 struct rocket_iommu_domain *domain;
40
41 struct kref refcount;
42};
43
44int rocket_ioctl_submit(struct drm_device *dev, void *data, struct drm_file *file);
45
46int rocket_job_init(struct rocket_core *core);
47void rocket_job_fini(struct rocket_core *core);
48int rocket_job_open(struct rocket_file_priv *rocket_priv);
49void rocket_job_close(struct rocket_file_priv *rocket_priv);
50int rocket_job_is_idle(struct rocket_core *core);
51
52#endif