Merge tag 'pinctrl-v6.9-1' of git://git.kernel.org/pub/scm/linux/kernel/git/linusw...
[sfrench/cifs-2.6.git] / drivers / gpu / drm / amd / amdgpu / jpeg_v4_0.c
1 /*
2  * Copyright 2021 Advanced Micro Devices, Inc.
3  *
4  * Permission is hereby granted, free of charge, to any person obtaining a
5  * copy of this software and associated documentation files (the "Software"),
6  * to deal in the Software without restriction, including without limitation
7  * the rights to use, copy, modify, merge, publish, distribute, sublicense,
8  * and/or sell copies of the Software, and to permit persons to whom the
9  * Software is furnished to do so, subject to the following conditions:
10  *
11  * The above copyright notice and this permission notice shall be included in
12  * all copies or substantial portions of the Software.
13  *
14  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
15  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
16  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL
17  * THE COPYRIGHT HOLDER(S) OR AUTHOR(S) BE LIABLE FOR ANY CLAIM, DAMAGES OR
18  * OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE,
19  * ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
20  * OTHER DEALINGS IN THE SOFTWARE.
21  *
22  */
23
24 #include "amdgpu.h"
25 #include "amdgpu_jpeg.h"
26 #include "amdgpu_pm.h"
27 #include "soc15.h"
28 #include "soc15d.h"
29 #include "jpeg_v2_0.h"
30 #include "jpeg_v4_0.h"
31 #include "mmsch_v4_0.h"
32
33 #include "vcn/vcn_4_0_0_offset.h"
34 #include "vcn/vcn_4_0_0_sh_mask.h"
35 #include "ivsrcid/vcn/irqsrcs_vcn_4_0.h"
36
37 #define regUVD_JPEG_PITCH_INTERNAL_OFFSET                  0x401f
38
39 static int jpeg_v4_0_start_sriov(struct amdgpu_device *adev);
40 static void jpeg_v4_0_set_dec_ring_funcs(struct amdgpu_device *adev);
41 static void jpeg_v4_0_set_irq_funcs(struct amdgpu_device *adev);
42 static int jpeg_v4_0_set_powergating_state(void *handle,
43                                 enum amd_powergating_state state);
44 static void jpeg_v4_0_set_ras_funcs(struct amdgpu_device *adev);
45
46 static void jpeg_v4_0_dec_ring_set_wptr(struct amdgpu_ring *ring);
47
48 /**
49  * jpeg_v4_0_early_init - set function pointers
50  *
51  * @handle: amdgpu_device pointer
52  *
53  * Set ring and irq function pointers
54  */
55 static int jpeg_v4_0_early_init(void *handle)
56 {
57         struct amdgpu_device *adev = (struct amdgpu_device *)handle;
58
59
60         adev->jpeg.num_jpeg_inst = 1;
61         adev->jpeg.num_jpeg_rings = 1;
62
63         jpeg_v4_0_set_dec_ring_funcs(adev);
64         jpeg_v4_0_set_irq_funcs(adev);
65         jpeg_v4_0_set_ras_funcs(adev);
66
67         return 0;
68 }
69
70 /**
71  * jpeg_v4_0_sw_init - sw init for JPEG block
72  *
73  * @handle: amdgpu_device pointer
74  *
75  * Load firmware and sw initialization
76  */
77 static int jpeg_v4_0_sw_init(void *handle)
78 {
79         struct amdgpu_device *adev = (struct amdgpu_device *)handle;
80         struct amdgpu_ring *ring;
81         int r;
82
83         /* JPEG TRAP */
84         r = amdgpu_irq_add_id(adev, SOC15_IH_CLIENTID_VCN,
85                 VCN_4_0__SRCID__JPEG_DECODE, &adev->jpeg.inst->irq);
86         if (r)
87                 return r;
88
89         /* JPEG DJPEG POISON EVENT */
90         r = amdgpu_irq_add_id(adev, SOC15_IH_CLIENTID_VCN,
91                         VCN_4_0__SRCID_DJPEG0_POISON, &adev->jpeg.inst->ras_poison_irq);
92         if (r)
93                 return r;
94
95         /* JPEG EJPEG POISON EVENT */
96         r = amdgpu_irq_add_id(adev, SOC15_IH_CLIENTID_VCN,
97                         VCN_4_0__SRCID_EJPEG0_POISON, &adev->jpeg.inst->ras_poison_irq);
98         if (r)
99                 return r;
100
101         r = amdgpu_jpeg_sw_init(adev);
102         if (r)
103                 return r;
104
105         r = amdgpu_jpeg_resume(adev);
106         if (r)
107                 return r;
108
109         ring = adev->jpeg.inst->ring_dec;
110         ring->use_doorbell = true;
111         ring->doorbell_index = amdgpu_sriov_vf(adev) ? (((adev->doorbell_index.vcn.vcn_ring0_1) << 1) + 4) : ((adev->doorbell_index.vcn.vcn_ring0_1 << 1) + 1);
112         ring->vm_hub = AMDGPU_MMHUB0(0);
113
114         sprintf(ring->name, "jpeg_dec");
115         r = amdgpu_ring_init(adev, ring, 512, &adev->jpeg.inst->irq, 0,
116                              AMDGPU_RING_PRIO_DEFAULT, NULL);
117         if (r)
118                 return r;
119
120         adev->jpeg.internal.jpeg_pitch[0] = regUVD_JPEG_PITCH_INTERNAL_OFFSET;
121         adev->jpeg.inst->external.jpeg_pitch[0] = SOC15_REG_OFFSET(JPEG, 0, regUVD_JPEG_PITCH);
122
123         r = amdgpu_jpeg_ras_sw_init(adev);
124         if (r)
125                 return r;
126
127         return 0;
128 }
129
130 /**
131  * jpeg_v4_0_sw_fini - sw fini for JPEG block
132  *
133  * @handle: amdgpu_device pointer
134  *
135  * JPEG suspend and free up sw allocation
136  */
137 static int jpeg_v4_0_sw_fini(void *handle)
138 {
139         struct amdgpu_device *adev = (struct amdgpu_device *)handle;
140         int r;
141
142         r = amdgpu_jpeg_suspend(adev);
143         if (r)
144                 return r;
145
146         r = amdgpu_jpeg_sw_fini(adev);
147
148         return r;
149 }
150
151 /**
152  * jpeg_v4_0_hw_init - start and test JPEG block
153  *
154  * @handle: amdgpu_device pointer
155  *
156  */
157 static int jpeg_v4_0_hw_init(void *handle)
158 {
159         struct amdgpu_device *adev = (struct amdgpu_device *)handle;
160         struct amdgpu_ring *ring = adev->jpeg.inst->ring_dec;
161         int r;
162
163         if (amdgpu_sriov_vf(adev)) {
164                 r = jpeg_v4_0_start_sriov(adev);
165                 if (r)
166                         return r;
167                 ring->wptr = 0;
168                 ring->wptr_old = 0;
169                 jpeg_v4_0_dec_ring_set_wptr(ring);
170                 ring->sched.ready = true;
171         } else {
172                 adev->nbio.funcs->vcn_doorbell_range(adev, ring->use_doorbell,
173                                                 (adev->doorbell_index.vcn.vcn_ring0_1 << 1), 0);
174
175                 WREG32_SOC15(VCN, 0, regVCN_JPEG_DB_CTRL,
176                         ring->doorbell_index << VCN_JPEG_DB_CTRL__OFFSET__SHIFT |
177                         VCN_JPEG_DB_CTRL__EN_MASK);
178
179                 r = amdgpu_ring_test_helper(ring);
180                 if (r)
181                         return r;
182         }
183
184         DRM_DEV_INFO(adev->dev, "JPEG decode initialized successfully.\n");
185
186         return 0;
187 }
188
189 /**
190  * jpeg_v4_0_hw_fini - stop the hardware block
191  *
192  * @handle: amdgpu_device pointer
193  *
194  * Stop the JPEG block, mark ring as not ready any more
195  */
196 static int jpeg_v4_0_hw_fini(void *handle)
197 {
198         struct amdgpu_device *adev = (struct amdgpu_device *)handle;
199
200         cancel_delayed_work_sync(&adev->vcn.idle_work);
201         if (!amdgpu_sriov_vf(adev)) {
202                 if (adev->jpeg.cur_state != AMD_PG_STATE_GATE &&
203                         RREG32_SOC15(JPEG, 0, regUVD_JRBC_STATUS))
204                         jpeg_v4_0_set_powergating_state(adev, AMD_PG_STATE_GATE);
205         }
206         if (amdgpu_ras_is_supported(adev, AMDGPU_RAS_BLOCK__JPEG))
207                 amdgpu_irq_put(adev, &adev->jpeg.inst->ras_poison_irq, 0);
208
209         return 0;
210 }
211
212 /**
213  * jpeg_v4_0_suspend - suspend JPEG block
214  *
215  * @handle: amdgpu_device pointer
216  *
217  * HW fini and suspend JPEG block
218  */
219 static int jpeg_v4_0_suspend(void *handle)
220 {
221         struct amdgpu_device *adev = (struct amdgpu_device *)handle;
222         int r;
223
224         r = jpeg_v4_0_hw_fini(adev);
225         if (r)
226                 return r;
227
228         r = amdgpu_jpeg_suspend(adev);
229
230         return r;
231 }
232
233 /**
234  * jpeg_v4_0_resume - resume JPEG block
235  *
236  * @handle: amdgpu_device pointer
237  *
238  * Resume firmware and hw init JPEG block
239  */
240 static int jpeg_v4_0_resume(void *handle)
241 {
242         struct amdgpu_device *adev = (struct amdgpu_device *)handle;
243         int r;
244
245         r = amdgpu_jpeg_resume(adev);
246         if (r)
247                 return r;
248
249         r = jpeg_v4_0_hw_init(adev);
250
251         return r;
252 }
253
254 static void jpeg_v4_0_disable_clock_gating(struct amdgpu_device *adev)
255 {
256         uint32_t data = 0;
257
258         data = RREG32_SOC15(JPEG, 0, regJPEG_CGC_CTRL);
259         if (adev->cg_flags & AMD_CG_SUPPORT_JPEG_MGCG) {
260                 data |= 1 << JPEG_CGC_CTRL__DYN_CLOCK_MODE__SHIFT;
261                 data &= (~JPEG_CGC_CTRL__JPEG_DEC_MODE_MASK);
262         } else {
263                 data &= ~JPEG_CGC_CTRL__DYN_CLOCK_MODE__SHIFT;
264         }
265
266         data |= 1 << JPEG_CGC_CTRL__CLK_GATE_DLY_TIMER__SHIFT;
267         data |= 4 << JPEG_CGC_CTRL__CLK_OFF_DELAY__SHIFT;
268         WREG32_SOC15(JPEG, 0, regJPEG_CGC_CTRL, data);
269
270         data = RREG32_SOC15(JPEG, 0, regJPEG_CGC_GATE);
271         data &= ~(JPEG_CGC_GATE__JPEG_DEC_MASK
272                 | JPEG_CGC_GATE__JPEG2_DEC_MASK
273                 | JPEG_CGC_GATE__JMCIF_MASK
274                 | JPEG_CGC_GATE__JRBBM_MASK);
275         WREG32_SOC15(JPEG, 0, regJPEG_CGC_GATE, data);
276 }
277
278 static void jpeg_v4_0_enable_clock_gating(struct amdgpu_device *adev)
279 {
280         uint32_t data = 0;
281
282         data = RREG32_SOC15(JPEG, 0, regJPEG_CGC_CTRL);
283         if (adev->cg_flags & AMD_CG_SUPPORT_JPEG_MGCG) {
284                 data |= 1 << JPEG_CGC_CTRL__DYN_CLOCK_MODE__SHIFT;
285                 data |= JPEG_CGC_CTRL__JPEG_DEC_MODE_MASK;
286         } else {
287                 data &= ~JPEG_CGC_CTRL__DYN_CLOCK_MODE__SHIFT;
288         }
289
290         data |= 1 << JPEG_CGC_CTRL__CLK_GATE_DLY_TIMER__SHIFT;
291         data |= 4 << JPEG_CGC_CTRL__CLK_OFF_DELAY__SHIFT;
292         WREG32_SOC15(JPEG, 0, regJPEG_CGC_CTRL, data);
293
294         data = RREG32_SOC15(JPEG, 0, regJPEG_CGC_GATE);
295         data |= (JPEG_CGC_GATE__JPEG_DEC_MASK
296                 |JPEG_CGC_GATE__JPEG2_DEC_MASK
297                 |JPEG_CGC_GATE__JMCIF_MASK
298                 |JPEG_CGC_GATE__JRBBM_MASK);
299         WREG32_SOC15(JPEG, 0, regJPEG_CGC_GATE, data);
300 }
301
302 static int jpeg_v4_0_disable_static_power_gating(struct amdgpu_device *adev)
303 {
304         if (adev->pg_flags & AMD_PG_SUPPORT_JPEG) {
305                 uint32_t data = 0;
306                 int r = 0;
307
308                 data = 1 << UVD_PGFSM_CONFIG__UVDJ_PWR_CONFIG__SHIFT;
309                 WREG32(SOC15_REG_OFFSET(JPEG, 0, regUVD_PGFSM_CONFIG), data);
310
311                 r = SOC15_WAIT_ON_RREG(JPEG, 0,
312                         regUVD_PGFSM_STATUS, UVD_PGFSM_STATUS_UVDJ_PWR_ON,
313                         UVD_PGFSM_STATUS__UVDJ_PWR_STATUS_MASK);
314
315                 if (r) {
316                         DRM_DEV_ERROR(adev->dev, "amdgpu: JPEG disable power gating failed\n");
317                         return r;
318                 }
319         }
320
321         /* disable anti hang mechanism */
322         WREG32_P(SOC15_REG_OFFSET(JPEG, 0, regUVD_JPEG_POWER_STATUS), 0,
323                 ~UVD_JPEG_POWER_STATUS__JPEG_POWER_STATUS_MASK);
324
325         /* keep the JPEG in static PG mode */
326         WREG32_P(SOC15_REG_OFFSET(JPEG, 0, regUVD_JPEG_POWER_STATUS), 0,
327                 ~UVD_JPEG_POWER_STATUS__JPEG_PG_MODE_MASK);
328
329         return 0;
330 }
331
332 static int jpeg_v4_0_enable_static_power_gating(struct amdgpu_device *adev)
333 {
334         /* enable anti hang mechanism */
335         WREG32_P(SOC15_REG_OFFSET(JPEG, 0, regUVD_JPEG_POWER_STATUS),
336                 UVD_JPEG_POWER_STATUS__JPEG_POWER_STATUS_MASK,
337                 ~UVD_JPEG_POWER_STATUS__JPEG_POWER_STATUS_MASK);
338
339         if (adev->pg_flags & AMD_PG_SUPPORT_JPEG) {
340                 uint32_t data = 0;
341                 int r = 0;
342
343                 data = 2 << UVD_PGFSM_CONFIG__UVDJ_PWR_CONFIG__SHIFT;
344                 WREG32(SOC15_REG_OFFSET(JPEG, 0, regUVD_PGFSM_CONFIG), data);
345
346                 r = SOC15_WAIT_ON_RREG(JPEG, 0, regUVD_PGFSM_STATUS,
347                         (2 << UVD_PGFSM_STATUS__UVDJ_PWR_STATUS__SHIFT),
348                         UVD_PGFSM_STATUS__UVDJ_PWR_STATUS_MASK);
349
350                 if (r) {
351                         DRM_DEV_ERROR(adev->dev, "amdgpu: JPEG enable power gating failed\n");
352                         return r;
353                 }
354         }
355
356         return 0;
357 }
358
359 /**
360  * jpeg_v4_0_start - start JPEG block
361  *
362  * @adev: amdgpu_device pointer
363  *
364  * Setup and start the JPEG block
365  */
366 static int jpeg_v4_0_start(struct amdgpu_device *adev)
367 {
368         struct amdgpu_ring *ring = adev->jpeg.inst->ring_dec;
369         int r;
370
371         if (adev->pm.dpm_enabled)
372                 amdgpu_dpm_enable_jpeg(adev, true);
373
374         /* disable power gating */
375         r = jpeg_v4_0_disable_static_power_gating(adev);
376         if (r)
377                 return r;
378
379         /* JPEG disable CGC */
380         jpeg_v4_0_disable_clock_gating(adev);
381
382         /* MJPEG global tiling registers */
383         WREG32_SOC15(JPEG, 0, regJPEG_DEC_GFX10_ADDR_CONFIG,
384                 adev->gfx.config.gb_addr_config);
385
386
387         /* enable JMI channel */
388         WREG32_P(SOC15_REG_OFFSET(JPEG, 0, regUVD_JMI_CNTL), 0,
389                 ~UVD_JMI_CNTL__SOFT_RESET_MASK);
390
391         /* enable System Interrupt for JRBC */
392         WREG32_P(SOC15_REG_OFFSET(JPEG, 0, regJPEG_SYS_INT_EN),
393                 JPEG_SYS_INT_EN__DJRBC_MASK,
394                 ~JPEG_SYS_INT_EN__DJRBC_MASK);
395
396         WREG32_SOC15(JPEG, 0, regUVD_LMI_JRBC_RB_VMID, 0);
397         WREG32_SOC15(JPEG, 0, regUVD_JRBC_RB_CNTL, (0x00000001L | 0x00000002L));
398         WREG32_SOC15(JPEG, 0, regUVD_LMI_JRBC_RB_64BIT_BAR_LOW,
399                 lower_32_bits(ring->gpu_addr));
400         WREG32_SOC15(JPEG, 0, regUVD_LMI_JRBC_RB_64BIT_BAR_HIGH,
401                 upper_32_bits(ring->gpu_addr));
402         WREG32_SOC15(JPEG, 0, regUVD_JRBC_RB_RPTR, 0);
403         WREG32_SOC15(JPEG, 0, regUVD_JRBC_RB_WPTR, 0);
404         WREG32_SOC15(JPEG, 0, regUVD_JRBC_RB_CNTL, 0x00000002L);
405         WREG32_SOC15(JPEG, 0, regUVD_JRBC_RB_SIZE, ring->ring_size / 4);
406         ring->wptr = RREG32_SOC15(JPEG, 0, regUVD_JRBC_RB_WPTR);
407
408         return 0;
409 }
410
411 static int jpeg_v4_0_start_sriov(struct amdgpu_device *adev)
412 {
413         struct amdgpu_ring *ring;
414         uint64_t ctx_addr;
415         uint32_t param, resp, expected;
416         uint32_t tmp, timeout;
417
418         struct amdgpu_mm_table *table = &adev->virt.mm_table;
419         uint32_t *table_loc;
420         uint32_t table_size;
421         uint32_t size, size_dw;
422         uint32_t init_status;
423
424         struct mmsch_v4_0_cmd_direct_write
425                 direct_wt = { {0} };
426         struct mmsch_v4_0_cmd_end end = { {0} };
427         struct mmsch_v4_0_init_header header;
428
429         direct_wt.cmd_header.command_type =
430                 MMSCH_COMMAND__DIRECT_REG_WRITE;
431         end.cmd_header.command_type =
432                 MMSCH_COMMAND__END;
433
434         size = sizeof(struct mmsch_v4_0_init_header);
435         table_loc = (uint32_t *)table->cpu_addr;
436         memcpy(&header, (void *)table_loc, size);
437
438         header.version = MMSCH_VERSION;
439         header.total_size = RREG32_SOC15(VCN, 0, regMMSCH_VF_CTX_SIZE);
440
441         header.jpegdec.init_status = 0;
442         header.jpegdec.table_offset = 0;
443         header.jpegdec.table_size = 0;
444
445         table_loc = (uint32_t *)table->cpu_addr;
446         table_loc += header.total_size;
447
448         table_size = 0;
449
450         ring = adev->jpeg.inst->ring_dec;
451
452         MMSCH_V4_0_INSERT_DIRECT_WT(SOC15_REG_OFFSET(JPEG, 0,
453                 regUVD_LMI_JRBC_RB_64BIT_BAR_LOW),
454                 lower_32_bits(ring->gpu_addr));
455         MMSCH_V4_0_INSERT_DIRECT_WT(SOC15_REG_OFFSET(JPEG, 0,
456                 regUVD_LMI_JRBC_RB_64BIT_BAR_HIGH),
457                 upper_32_bits(ring->gpu_addr));
458         MMSCH_V4_0_INSERT_DIRECT_WT(SOC15_REG_OFFSET(JPEG, 0,
459                 regUVD_JRBC_RB_SIZE), ring->ring_size / 4);
460
461         /* add end packet */
462         MMSCH_V4_0_INSERT_END();
463
464         /* refine header */
465         header.jpegdec.init_status = 0;
466         header.jpegdec.table_offset = header.total_size;
467         header.jpegdec.table_size = table_size;
468         header.total_size += table_size;
469
470         /* Update init table header in memory */
471         size = sizeof(struct mmsch_v4_0_init_header);
472         table_loc = (uint32_t *)table->cpu_addr;
473         memcpy((void *)table_loc, &header, size);
474
475         /* Perform HDP flush before writing to MMSCH registers */
476         amdgpu_device_flush_hdp(adev, NULL);
477
478         /* message MMSCH (in VCN[0]) to initialize this client
479          * 1, write to mmsch_vf_ctx_addr_lo/hi register with GPU mc addr
480          * of memory descriptor location
481          */
482         ctx_addr = table->gpu_addr;
483         WREG32_SOC15(VCN, 0, regMMSCH_VF_CTX_ADDR_LO, lower_32_bits(ctx_addr));
484         WREG32_SOC15(VCN, 0, regMMSCH_VF_CTX_ADDR_HI, upper_32_bits(ctx_addr));
485
486         /* 2, update vmid of descriptor */
487         tmp = RREG32_SOC15(VCN, 0, regMMSCH_VF_VMID);
488         tmp &= ~MMSCH_VF_VMID__VF_CTX_VMID_MASK;
489         /* use domain0 for MM scheduler */
490         tmp |= (0 << MMSCH_VF_VMID__VF_CTX_VMID__SHIFT);
491         WREG32_SOC15(VCN, 0, regMMSCH_VF_VMID, tmp);
492
493         /* 3, notify mmsch about the size of this descriptor */
494         size = header.total_size;
495         WREG32_SOC15(VCN, 0, regMMSCH_VF_CTX_SIZE, size);
496
497         /* 4, set resp to zero */
498         WREG32_SOC15(VCN, 0, regMMSCH_VF_MAILBOX_RESP, 0);
499
500         /* 5, kick off the initialization and wait until
501          * MMSCH_VF_MAILBOX_RESP becomes non-zero
502          */
503         param = 0x00000001;
504         WREG32_SOC15(VCN, 0, regMMSCH_VF_MAILBOX_HOST, param);
505         tmp = 0;
506         timeout = 1000;
507         resp = 0;
508         expected = MMSCH_VF_MAILBOX_RESP__OK;
509         init_status = ((struct mmsch_v4_0_init_header *)(table_loc))->jpegdec.init_status;
510         while (resp != expected) {
511                 resp = RREG32_SOC15(VCN, 0, regMMSCH_VF_MAILBOX_RESP);
512
513                 if (resp != 0)
514                         break;
515                 udelay(10);
516                 tmp = tmp + 10;
517                 if (tmp >= timeout) {
518                         DRM_ERROR("failed to init MMSCH. TIME-OUT after %d usec"\
519                                 " waiting for regMMSCH_VF_MAILBOX_RESP "\
520                                 "(expected=0x%08x, readback=0x%08x)\n",
521                                 tmp, expected, resp);
522                         return -EBUSY;
523                 }
524         }
525         if (resp != expected && resp != MMSCH_VF_MAILBOX_RESP__INCOMPLETE
526                         && init_status != MMSCH_VF_ENGINE_STATUS__PASS) {
527                 DRM_ERROR("MMSCH init status is incorrect! readback=0x%08x, header init status for jpeg: %x\n", resp, init_status);
528                 return -EINVAL;
529         }
530
531         return 0;
532
533 }
534
535 /**
536  * jpeg_v4_0_stop - stop JPEG block
537  *
538  * @adev: amdgpu_device pointer
539  *
540  * stop the JPEG block
541  */
542 static int jpeg_v4_0_stop(struct amdgpu_device *adev)
543 {
544         int r;
545
546         /* reset JMI */
547         WREG32_P(SOC15_REG_OFFSET(JPEG, 0, regUVD_JMI_CNTL),
548                 UVD_JMI_CNTL__SOFT_RESET_MASK,
549                 ~UVD_JMI_CNTL__SOFT_RESET_MASK);
550
551         jpeg_v4_0_enable_clock_gating(adev);
552
553         /* enable power gating */
554         r = jpeg_v4_0_enable_static_power_gating(adev);
555         if (r)
556                 return r;
557
558         if (adev->pm.dpm_enabled)
559                 amdgpu_dpm_enable_jpeg(adev, false);
560
561         return 0;
562 }
563
564 /**
565  * jpeg_v4_0_dec_ring_get_rptr - get read pointer
566  *
567  * @ring: amdgpu_ring pointer
568  *
569  * Returns the current hardware read pointer
570  */
571 static uint64_t jpeg_v4_0_dec_ring_get_rptr(struct amdgpu_ring *ring)
572 {
573         struct amdgpu_device *adev = ring->adev;
574
575         return RREG32_SOC15(JPEG, 0, regUVD_JRBC_RB_RPTR);
576 }
577
578 /**
579  * jpeg_v4_0_dec_ring_get_wptr - get write pointer
580  *
581  * @ring: amdgpu_ring pointer
582  *
583  * Returns the current hardware write pointer
584  */
585 static uint64_t jpeg_v4_0_dec_ring_get_wptr(struct amdgpu_ring *ring)
586 {
587         struct amdgpu_device *adev = ring->adev;
588
589         if (ring->use_doorbell)
590                 return *ring->wptr_cpu_addr;
591         else
592                 return RREG32_SOC15(JPEG, 0, regUVD_JRBC_RB_WPTR);
593 }
594
595 /**
596  * jpeg_v4_0_dec_ring_set_wptr - set write pointer
597  *
598  * @ring: amdgpu_ring pointer
599  *
600  * Commits the write pointer to the hardware
601  */
602 static void jpeg_v4_0_dec_ring_set_wptr(struct amdgpu_ring *ring)
603 {
604         struct amdgpu_device *adev = ring->adev;
605
606         if (ring->use_doorbell) {
607                 *ring->wptr_cpu_addr = lower_32_bits(ring->wptr);
608                 WDOORBELL32(ring->doorbell_index, lower_32_bits(ring->wptr));
609         } else {
610                 WREG32_SOC15(JPEG, 0, regUVD_JRBC_RB_WPTR, lower_32_bits(ring->wptr));
611         }
612 }
613
614 static bool jpeg_v4_0_is_idle(void *handle)
615 {
616         struct amdgpu_device *adev = (struct amdgpu_device *)handle;
617         int ret = 1;
618
619         ret &= (((RREG32_SOC15(JPEG, 0, regUVD_JRBC_STATUS) &
620                 UVD_JRBC_STATUS__RB_JOB_DONE_MASK) ==
621                 UVD_JRBC_STATUS__RB_JOB_DONE_MASK));
622
623         return ret;
624 }
625
626 static int jpeg_v4_0_wait_for_idle(void *handle)
627 {
628         struct amdgpu_device *adev = (struct amdgpu_device *)handle;
629
630         return SOC15_WAIT_ON_RREG(JPEG, 0, regUVD_JRBC_STATUS,
631                 UVD_JRBC_STATUS__RB_JOB_DONE_MASK,
632                 UVD_JRBC_STATUS__RB_JOB_DONE_MASK);
633 }
634
635 static int jpeg_v4_0_set_clockgating_state(void *handle,
636                                           enum amd_clockgating_state state)
637 {
638         struct amdgpu_device *adev = (struct amdgpu_device *)handle;
639         bool enable = state == AMD_CG_STATE_GATE;
640
641         if (enable) {
642                 if (!jpeg_v4_0_is_idle(handle))
643                         return -EBUSY;
644                 jpeg_v4_0_enable_clock_gating(adev);
645         } else {
646                 jpeg_v4_0_disable_clock_gating(adev);
647         }
648
649         return 0;
650 }
651
652 static int jpeg_v4_0_set_powergating_state(void *handle,
653                                           enum amd_powergating_state state)
654 {
655         struct amdgpu_device *adev = (struct amdgpu_device *)handle;
656         int ret;
657
658         if (amdgpu_sriov_vf(adev)) {
659                 adev->jpeg.cur_state = AMD_PG_STATE_UNGATE;
660                 return 0;
661         }
662
663         if (state == adev->jpeg.cur_state)
664                 return 0;
665
666         if (state == AMD_PG_STATE_GATE)
667                 ret = jpeg_v4_0_stop(adev);
668         else
669                 ret = jpeg_v4_0_start(adev);
670
671         if (!ret)
672                 adev->jpeg.cur_state = state;
673
674         return ret;
675 }
676
677 static int jpeg_v4_0_set_ras_interrupt_state(struct amdgpu_device *adev,
678                                         struct amdgpu_irq_src *source,
679                                         unsigned int type,
680                                         enum amdgpu_interrupt_state state)
681 {
682         return 0;
683 }
684
685 static int jpeg_v4_0_process_interrupt(struct amdgpu_device *adev,
686                                       struct amdgpu_irq_src *source,
687                                       struct amdgpu_iv_entry *entry)
688 {
689         DRM_DEBUG("IH: JPEG TRAP\n");
690
691         switch (entry->src_id) {
692         case VCN_4_0__SRCID__JPEG_DECODE:
693                 amdgpu_fence_process(adev->jpeg.inst->ring_dec);
694                 break;
695         default:
696                 DRM_DEV_ERROR(adev->dev, "Unhandled interrupt: %d %d\n",
697                           entry->src_id, entry->src_data[0]);
698                 break;
699         }
700
701         return 0;
702 }
703
704 static const struct amd_ip_funcs jpeg_v4_0_ip_funcs = {
705         .name = "jpeg_v4_0",
706         .early_init = jpeg_v4_0_early_init,
707         .late_init = NULL,
708         .sw_init = jpeg_v4_0_sw_init,
709         .sw_fini = jpeg_v4_0_sw_fini,
710         .hw_init = jpeg_v4_0_hw_init,
711         .hw_fini = jpeg_v4_0_hw_fini,
712         .suspend = jpeg_v4_0_suspend,
713         .resume = jpeg_v4_0_resume,
714         .is_idle = jpeg_v4_0_is_idle,
715         .wait_for_idle = jpeg_v4_0_wait_for_idle,
716         .check_soft_reset = NULL,
717         .pre_soft_reset = NULL,
718         .soft_reset = NULL,
719         .post_soft_reset = NULL,
720         .set_clockgating_state = jpeg_v4_0_set_clockgating_state,
721         .set_powergating_state = jpeg_v4_0_set_powergating_state,
722 };
723
724 static const struct amdgpu_ring_funcs jpeg_v4_0_dec_ring_vm_funcs = {
725         .type = AMDGPU_RING_TYPE_VCN_JPEG,
726         .align_mask = 0xf,
727         .get_rptr = jpeg_v4_0_dec_ring_get_rptr,
728         .get_wptr = jpeg_v4_0_dec_ring_get_wptr,
729         .set_wptr = jpeg_v4_0_dec_ring_set_wptr,
730         .emit_frame_size =
731                 SOC15_FLUSH_GPU_TLB_NUM_WREG * 6 +
732                 SOC15_FLUSH_GPU_TLB_NUM_REG_WAIT * 8 +
733                 8 + /* jpeg_v4_0_dec_ring_emit_vm_flush */
734                 18 + 18 + /* jpeg_v4_0_dec_ring_emit_fence x2 vm fence */
735                 8 + 16,
736         .emit_ib_size = 22, /* jpeg_v4_0_dec_ring_emit_ib */
737         .emit_ib = jpeg_v2_0_dec_ring_emit_ib,
738         .emit_fence = jpeg_v2_0_dec_ring_emit_fence,
739         .emit_vm_flush = jpeg_v2_0_dec_ring_emit_vm_flush,
740         .test_ring = amdgpu_jpeg_dec_ring_test_ring,
741         .test_ib = amdgpu_jpeg_dec_ring_test_ib,
742         .insert_nop = jpeg_v2_0_dec_ring_nop,
743         .insert_start = jpeg_v2_0_dec_ring_insert_start,
744         .insert_end = jpeg_v2_0_dec_ring_insert_end,
745         .pad_ib = amdgpu_ring_generic_pad_ib,
746         .begin_use = amdgpu_jpeg_ring_begin_use,
747         .end_use = amdgpu_jpeg_ring_end_use,
748         .emit_wreg = jpeg_v2_0_dec_ring_emit_wreg,
749         .emit_reg_wait = jpeg_v2_0_dec_ring_emit_reg_wait,
750         .emit_reg_write_reg_wait = amdgpu_ring_emit_reg_write_reg_wait_helper,
751 };
752
753 static void jpeg_v4_0_set_dec_ring_funcs(struct amdgpu_device *adev)
754 {
755         adev->jpeg.inst->ring_dec->funcs = &jpeg_v4_0_dec_ring_vm_funcs;
756         DRM_DEV_INFO(adev->dev, "JPEG decode is enabled in VM mode\n");
757 }
758
759 static const struct amdgpu_irq_src_funcs jpeg_v4_0_irq_funcs = {
760         .process = jpeg_v4_0_process_interrupt,
761 };
762
763 static const struct amdgpu_irq_src_funcs jpeg_v4_0_ras_irq_funcs = {
764         .set = jpeg_v4_0_set_ras_interrupt_state,
765         .process = amdgpu_jpeg_process_poison_irq,
766 };
767
768 static void jpeg_v4_0_set_irq_funcs(struct amdgpu_device *adev)
769 {
770         adev->jpeg.inst->irq.num_types = 1;
771         adev->jpeg.inst->irq.funcs = &jpeg_v4_0_irq_funcs;
772
773         adev->jpeg.inst->ras_poison_irq.num_types = 1;
774         adev->jpeg.inst->ras_poison_irq.funcs = &jpeg_v4_0_ras_irq_funcs;
775 }
776
777 const struct amdgpu_ip_block_version jpeg_v4_0_ip_block = {
778         .type = AMD_IP_BLOCK_TYPE_JPEG,
779         .major = 4,
780         .minor = 0,
781         .rev = 0,
782         .funcs = &jpeg_v4_0_ip_funcs,
783 };
784
785 static uint32_t jpeg_v4_0_query_poison_by_instance(struct amdgpu_device *adev,
786                 uint32_t instance, uint32_t sub_block)
787 {
788         uint32_t poison_stat = 0, reg_value = 0;
789
790         switch (sub_block) {
791         case AMDGPU_JPEG_V4_0_JPEG0:
792                 reg_value = RREG32_SOC15(JPEG, instance, regUVD_RAS_JPEG0_STATUS);
793                 poison_stat = REG_GET_FIELD(reg_value, UVD_RAS_JPEG0_STATUS, POISONED_PF);
794                 break;
795         case AMDGPU_JPEG_V4_0_JPEG1:
796                 reg_value = RREG32_SOC15(JPEG, instance, regUVD_RAS_JPEG1_STATUS);
797                 poison_stat = REG_GET_FIELD(reg_value, UVD_RAS_JPEG1_STATUS, POISONED_PF);
798                 break;
799         default:
800                 break;
801         }
802
803         if (poison_stat)
804                 dev_info(adev->dev, "Poison detected in JPEG%d sub_block%d\n",
805                         instance, sub_block);
806
807         return poison_stat;
808 }
809
810 static bool jpeg_v4_0_query_ras_poison_status(struct amdgpu_device *adev)
811 {
812         uint32_t inst = 0, sub = 0, poison_stat = 0;
813
814         for (inst = 0; inst < adev->jpeg.num_jpeg_inst; inst++)
815                 for (sub = 0; sub < AMDGPU_JPEG_V4_0_MAX_SUB_BLOCK; sub++)
816                         poison_stat +=
817                                 jpeg_v4_0_query_poison_by_instance(adev, inst, sub);
818
819         return !!poison_stat;
820 }
821
822 const struct amdgpu_ras_block_hw_ops jpeg_v4_0_ras_hw_ops = {
823         .query_poison_status = jpeg_v4_0_query_ras_poison_status,
824 };
825
826 static struct amdgpu_jpeg_ras jpeg_v4_0_ras = {
827         .ras_block = {
828                 .hw_ops = &jpeg_v4_0_ras_hw_ops,
829                 .ras_late_init = amdgpu_jpeg_ras_late_init,
830         },
831 };
832
833 static void jpeg_v4_0_set_ras_funcs(struct amdgpu_device *adev)
834 {
835         switch (amdgpu_ip_version(adev, JPEG_HWIP, 0)) {
836         case IP_VERSION(4, 0, 0):
837                 adev->jpeg.ras = &jpeg_v4_0_ras;
838                 break;
839         default:
840                 break;
841         }
842 }