Book a Demo!
CoCalc Logo Icon
StoreFeaturesDocsShareSupportNewsAboutPoliciesSign UpSign In
torvalds
GitHub Repository: torvalds/linux
 Path: blob/master/drivers/accel/ethosu/ethosu_job.h
38310 views
1
/* SPDX-License-Identifier: GPL-2.0-only OR MIT */

2
/* Copyright 2024-2025 Tomeu Vizoso <[email protected]> */

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


11
struct ethosu_device;

12
struct ethosu_file_priv;

13


14
struct 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


33
int ethosu_ioctl_submit(struct drm_device *dev, void *data, struct drm_file *file);

34


35
int ethosu_job_init(struct ethosu_device *dev);

36
void ethosu_job_fini(struct ethosu_device *dev);

37
int ethosu_job_open(struct ethosu_file_priv *ethosu_priv);

38
void ethosu_job_close(struct ethosu_file_priv *ethosu_priv);

39


40
#endif

41


42