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