| .. | .. |
|---|
| 29 | 29 | #define AMDGPU_PREAMBLE_IB_PRESENT_FIRST (1 << 1) |
|---|
| 30 | 30 | /* bit set means context switch occured */ |
|---|
| 31 | 31 | #define AMDGPU_HAVE_CTX_SWITCH (1 << 2) |
|---|
| 32 | +/* bit set means IB is preempted */ |
|---|
| 33 | +#define AMDGPU_IB_PREEMPTED (1 << 3) |
|---|
| 32 | 34 | |
|---|
| 33 | 35 | #define to_amdgpu_job(sched_job) \ |
|---|
| 34 | 36 | container_of((sched_job), struct amdgpu_job, base) |
|---|
| 35 | 37 | |
|---|
| 38 | +#define AMDGPU_JOB_GET_VMID(job) ((job) ? (job)->vmid : 0) |
|---|
| 39 | + |
|---|
| 36 | 40 | struct amdgpu_fence; |
|---|
| 41 | +enum amdgpu_ib_pool_type; |
|---|
| 37 | 42 | |
|---|
| 38 | 43 | struct amdgpu_job { |
|---|
| 39 | 44 | struct drm_sched_job base; |
|---|
| .. | .. |
|---|
| 43 | 48 | struct amdgpu_ib *ibs; |
|---|
| 44 | 49 | struct dma_fence *fence; /* the hw fence */ |
|---|
| 45 | 50 | uint32_t preamble_status; |
|---|
| 51 | + uint32_t preemption_status; |
|---|
| 46 | 52 | uint32_t num_ibs; |
|---|
| 47 | | - void *owner; |
|---|
| 48 | 53 | bool vm_needs_flush; |
|---|
| 49 | 54 | uint64_t vm_pd_addr; |
|---|
| 50 | 55 | unsigned vmid; |
|---|
| .. | .. |
|---|
| 57 | 62 | /* user fence handling */ |
|---|
| 58 | 63 | uint64_t uf_addr; |
|---|
| 59 | 64 | uint64_t uf_sequence; |
|---|
| 60 | | - |
|---|
| 61 | 65 | }; |
|---|
| 62 | 66 | |
|---|
| 63 | 67 | int amdgpu_job_alloc(struct amdgpu_device *adev, unsigned num_ibs, |
|---|
| 64 | 68 | struct amdgpu_job **job, struct amdgpu_vm *vm); |
|---|
| 65 | 69 | int amdgpu_job_alloc_with_ib(struct amdgpu_device *adev, unsigned size, |
|---|
| 66 | | - struct amdgpu_job **job); |
|---|
| 67 | | - |
|---|
| 70 | + enum amdgpu_ib_pool_type pool, struct amdgpu_job **job); |
|---|
| 68 | 71 | void amdgpu_job_free_resources(struct amdgpu_job *job); |
|---|
| 69 | 72 | void amdgpu_job_free(struct amdgpu_job *job); |
|---|
| 70 | 73 | int amdgpu_job_submit(struct amdgpu_job *job, struct drm_sched_entity *entity, |
|---|
| 71 | 74 | void *owner, struct dma_fence **f); |
|---|
| 72 | 75 | int amdgpu_job_submit_direct(struct amdgpu_job *job, struct amdgpu_ring *ring, |
|---|
| 73 | 76 | struct dma_fence **fence); |
|---|
| 77 | + |
|---|
| 78 | +void amdgpu_job_stop_all_jobs_on_sched(struct drm_gpu_scheduler *sched); |
|---|
| 79 | + |
|---|
| 74 | 80 | #endif |
|---|