ethosu_job.h

 1/* SPDX-License-Identifier: GPL-2.0-only OR MIT */
 2/* Copyright 2024-2025 Tomeu Vizoso <tomeu@tomeuvizoso.net> */
 3/* Copyright 2025 Arm, Ltd. */
 4
 5#ifndef __ETHOSU_JOB_H__
 6#define __ETHOSU_JOB_H__
 7
 8#include <linux/kref.h>
 9#include <drm/gpu_scheduler.h>
10
11struct ethosu_device;
12struct ethosu_file_priv;
13
14struct ethosu_job {
15	struct drm_sched_job base;
16	struct ethosu_device *dev;
17
18	struct drm_gem_object *cmd_bo;
19	struct drm_gem_object *region_bo[NPU_BASEP_REGION_MAX];
20	u8 region_bo_num[NPU_BASEP_REGION_MAX];
21	u8 region_cnt;
22	u32 sram_size;
23
24	/* Fence to be signaled by drm-sched once its done with the job */
25	struct dma_fence *inference_done_fence;
26
27	/* Fence to be signaled by IRQ handler when the job is complete. */
28	struct dma_fence *done_fence;
29
30	struct kref refcount;
31};
32
33int ethosu_ioctl_submit(struct drm_device *dev, void *data, struct drm_file *file);
34
35int ethosu_job_init(struct ethosu_device *dev);
36void ethosu_job_fini(struct ethosu_device *dev);
37int ethosu_job_open(struct ethosu_file_priv *ethosu_priv);
38void ethosu_job_close(struct ethosu_file_priv *ethosu_priv);
39
40#endif