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 13 struct rocket_task { 14 u64 regcmd; 15 u32 regcmd_count; 16 }; 17 18 struct 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 44 int rocket_ioctl_submit(struct drm_device *dev, void *data, struct drm_file *file); 45 46 int rocket_job_init(struct rocket_core *core); 47 void rocket_job_fini(struct rocket_core *core); 48 int rocket_job_open(struct rocket_file_priv *rocket_priv); 49 void rocket_job_close(struct rocket_file_priv *rocket_priv); 50 int rocket_job_is_idle(struct rocket_core *core); 51 52 #endif 53