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