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