Linux 6.9-rc6
[sfrench/cifs-2.6.git] / drivers / gpu / drm / amd / amdgpu / umsch_mm_v4_0.c
1 // SPDX-License-Identifier: MIT
2 /*
3  * Copyright 2023 Advanced Micro Devices, Inc.
4  *
5  * Permission is hereby granted, free of charge, to any person obtaining a
6  * copy of this software and associated documentation files (the "Software"),
7  * to deal in the Software without restriction, including without limitation
8  * the rights to use, copy, modify, merge, publish, distribute, sublicense,
9  * and/or sell copies of the Software, and to permit persons to whom the
10  * Software is furnished to do so, subject to the following conditions:
11  *
12  * The above copyright notice and this permission notice shall be included in
13  * all copies or substantial portions of the Software.
14  *
15  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
16  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
17  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL
18  * THE COPYRIGHT HOLDER(S) OR AUTHOR(S) BE LIABLE FOR ANY CLAIM, DAMAGES OR
19  * OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE,
20  * ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
21  * OTHER DEALINGS IN THE SOFTWARE.
22  *
23  */
24
25 #include <linux/firmware.h>
26 #include <linux/module.h>
27 #include "amdgpu.h"
28 #include "soc15_common.h"
29 #include "soc21.h"
30 #include "vcn/vcn_4_0_0_offset.h"
31 #include "vcn/vcn_4_0_0_sh_mask.h"
32
33 #include "amdgpu_umsch_mm.h"
34 #include "umsch_mm_4_0_api_def.h"
35 #include "umsch_mm_v4_0.h"
36
37 #define regUVD_IPX_DLDO_CONFIG                             0x0064
38 #define regUVD_IPX_DLDO_CONFIG_BASE_IDX                    1
39 #define regUVD_IPX_DLDO_STATUS                             0x0065
40 #define regUVD_IPX_DLDO_STATUS_BASE_IDX                    1
41
42 #define UVD_IPX_DLDO_CONFIG__ONO0_PWR_CONFIG__SHIFT        0x00000002
43 #define UVD_IPX_DLDO_CONFIG__ONO0_PWR_CONFIG_MASK          0x0000000cUL
44 #define UVD_IPX_DLDO_STATUS__ONO0_PWR_STATUS__SHIFT        0x00000001
45 #define UVD_IPX_DLDO_STATUS__ONO0_PWR_STATUS_MASK          0x00000002UL
46
47 static int umsch_mm_v4_0_load_microcode(struct amdgpu_umsch_mm *umsch)
48 {
49         struct amdgpu_device *adev = umsch->ring.adev;
50         uint64_t data;
51         int r;
52
53         r = amdgpu_umsch_mm_allocate_ucode_buffer(umsch);
54         if (r)
55                 return r;
56
57         r = amdgpu_umsch_mm_allocate_ucode_data_buffer(umsch);
58         if (r)
59                 goto err_free_ucode_bo;
60
61         umsch->cmd_buf_curr_ptr = umsch->cmd_buf_ptr;
62
63         if (amdgpu_ip_version(adev, VCN_HWIP, 0) >= IP_VERSION(4, 0, 5)) {
64                 WREG32_SOC15(VCN, 0, regUVD_IPX_DLDO_CONFIG,
65                         1 << UVD_IPX_DLDO_CONFIG__ONO0_PWR_CONFIG__SHIFT);
66                 SOC15_WAIT_ON_RREG(VCN, 0, regUVD_IPX_DLDO_STATUS,
67                         0 << UVD_IPX_DLDO_STATUS__ONO0_PWR_STATUS__SHIFT,
68                         UVD_IPX_DLDO_STATUS__ONO0_PWR_STATUS_MASK);
69         }
70
71         data = RREG32_SOC15(VCN, 0, regUMSCH_MES_RESET_CTRL);
72         data = REG_SET_FIELD(data, UMSCH_MES_RESET_CTRL, MES_CORE_SOFT_RESET, 0);
73         WREG32_SOC15_UMSCH(regUMSCH_MES_RESET_CTRL, data);
74
75         data = RREG32_SOC15(VCN, 0, regVCN_MES_CNTL);
76         data = REG_SET_FIELD(data, VCN_MES_CNTL, MES_INVALIDATE_ICACHE, 1);
77         data = REG_SET_FIELD(data, VCN_MES_CNTL, MES_PIPE0_RESET, 1);
78         data = REG_SET_FIELD(data, VCN_MES_CNTL, MES_PIPE0_ACTIVE, 0);
79         data = REG_SET_FIELD(data, VCN_MES_CNTL, MES_HALT, 1);
80         WREG32_SOC15_UMSCH(regVCN_MES_CNTL, data);
81
82         data = RREG32_SOC15(VCN, 0, regVCN_MES_IC_BASE_CNTL);
83         data = REG_SET_FIELD(data, VCN_MES_IC_BASE_CNTL, VMID, 0);
84         data = REG_SET_FIELD(data, VCN_MES_IC_BASE_CNTL, EXE_DISABLE, 0);
85         data = REG_SET_FIELD(data, VCN_MES_IC_BASE_CNTL, CACHE_POLICY, 0);
86         WREG32_SOC15_UMSCH(regVCN_MES_IC_BASE_CNTL, data);
87
88         WREG32_SOC15_UMSCH(regVCN_MES_INTR_ROUTINE_START,
89                 lower_32_bits(adev->umsch_mm.irq_start_addr >> 2));
90         WREG32_SOC15_UMSCH(regVCN_MES_INTR_ROUTINE_START_HI,
91                 upper_32_bits(adev->umsch_mm.irq_start_addr >> 2));
92
93         WREG32_SOC15_UMSCH(regVCN_MES_PRGRM_CNTR_START,
94                 lower_32_bits(adev->umsch_mm.uc_start_addr >> 2));
95         WREG32_SOC15_UMSCH(regVCN_MES_PRGRM_CNTR_START_HI,
96                 upper_32_bits(adev->umsch_mm.uc_start_addr >> 2));
97
98         WREG32_SOC15_UMSCH(regVCN_MES_LOCAL_INSTR_BASE_LO, 0);
99         WREG32_SOC15_UMSCH(regVCN_MES_LOCAL_INSTR_BASE_HI, 0);
100
101         data = adev->umsch_mm.uc_start_addr + adev->umsch_mm.ucode_size - 1;
102         WREG32_SOC15_UMSCH(regVCN_MES_LOCAL_INSTR_MASK_LO, lower_32_bits(data));
103         WREG32_SOC15_UMSCH(regVCN_MES_LOCAL_INSTR_MASK_HI, upper_32_bits(data));
104
105         data = adev->firmware.load_type == AMDGPU_FW_LOAD_PSP ?
106                0 : adev->umsch_mm.ucode_fw_gpu_addr;
107         WREG32_SOC15_UMSCH(regVCN_MES_IC_BASE_LO, lower_32_bits(data));
108         WREG32_SOC15_UMSCH(regVCN_MES_IC_BASE_HI, upper_32_bits(data));
109
110         WREG32_SOC15_UMSCH(regVCN_MES_MIBOUND_LO, 0x1FFFFF);
111
112         WREG32_SOC15_UMSCH(regVCN_MES_LOCAL_BASE0_LO,
113                 lower_32_bits(adev->umsch_mm.data_start_addr));
114         WREG32_SOC15_UMSCH(regVCN_MES_LOCAL_BASE0_HI,
115                 upper_32_bits(adev->umsch_mm.data_start_addr));
116
117         WREG32_SOC15_UMSCH(regVCN_MES_LOCAL_MASK0_LO,
118                 lower_32_bits(adev->umsch_mm.data_size - 1));
119         WREG32_SOC15_UMSCH(regVCN_MES_LOCAL_MASK0_HI,
120                 upper_32_bits(adev->umsch_mm.data_size - 1));
121
122         data = adev->firmware.load_type == AMDGPU_FW_LOAD_PSP ?
123                0 : adev->umsch_mm.data_fw_gpu_addr;
124         WREG32_SOC15_UMSCH(regVCN_MES_DC_BASE_LO, lower_32_bits(data));
125         WREG32_SOC15_UMSCH(regVCN_MES_DC_BASE_HI, upper_32_bits(data));
126
127         WREG32_SOC15_UMSCH(regVCN_MES_MDBOUND_LO, 0x3FFFF);
128
129         data = RREG32_SOC15(VCN, 0, regUVD_UMSCH_FORCE);
130         data = REG_SET_FIELD(data, UVD_UMSCH_FORCE, IC_FORCE_GPUVM, 1);
131         data = REG_SET_FIELD(data, UVD_UMSCH_FORCE, DC_FORCE_GPUVM, 1);
132         WREG32_SOC15_UMSCH(regUVD_UMSCH_FORCE, data);
133
134         data = RREG32_SOC15(VCN, 0, regVCN_MES_IC_OP_CNTL);
135         data = REG_SET_FIELD(data, VCN_MES_IC_OP_CNTL, PRIME_ICACHE, 0);
136         data = REG_SET_FIELD(data, VCN_MES_IC_OP_CNTL, INVALIDATE_CACHE, 1);
137         WREG32_SOC15_UMSCH(regVCN_MES_IC_OP_CNTL, data);
138
139         data = RREG32_SOC15(VCN, 0, regVCN_MES_IC_OP_CNTL);
140         data = REG_SET_FIELD(data, VCN_MES_IC_OP_CNTL, PRIME_ICACHE, 1);
141         WREG32_SOC15_UMSCH(regVCN_MES_IC_OP_CNTL, data);
142
143         WREG32_SOC15_UMSCH(regVCN_MES_GP0_LO, 0);
144         WREG32_SOC15_UMSCH(regVCN_MES_GP0_HI, 0);
145
146         WREG32_SOC15_UMSCH(regVCN_MES_GP1_LO, 0);
147         WREG32_SOC15_UMSCH(regVCN_MES_GP1_HI, 0);
148
149         data = RREG32_SOC15(VCN, 0, regVCN_MES_CNTL);
150         data = REG_SET_FIELD(data, VCN_MES_CNTL, MES_INVALIDATE_ICACHE, 0);
151         data = REG_SET_FIELD(data, VCN_MES_CNTL, MES_PIPE0_RESET, 0);
152         data = REG_SET_FIELD(data, VCN_MES_CNTL, MES_HALT, 0);
153         data = REG_SET_FIELD(data, VCN_MES_CNTL, MES_PIPE0_ACTIVE, 1);
154         WREG32_SOC15_UMSCH(regVCN_MES_CNTL, data);
155
156         if (adev->firmware.load_type == AMDGPU_FW_LOAD_PSP)
157                 amdgpu_umsch_mm_psp_execute_cmd_buf(umsch);
158
159         r = SOC15_WAIT_ON_RREG(VCN, 0, regVCN_MES_MSTATUS_LO, 0xAAAAAAAA, 0xFFFFFFFF);
160         if (r) {
161                 dev_err(adev->dev, "UMSCH FW Load: Failed, regVCN_MES_MSTATUS_LO: 0x%08x\n",
162                         RREG32_SOC15(VCN, 0, regVCN_MES_MSTATUS_LO));
163                 goto err_free_data_bo;
164         }
165
166         return 0;
167
168 err_free_data_bo:
169         amdgpu_bo_free_kernel(&adev->umsch_mm.data_fw_obj,
170                               &adev->umsch_mm.data_fw_gpu_addr,
171                               (void **)&adev->umsch_mm.data_fw_ptr);
172 err_free_ucode_bo:
173         amdgpu_bo_free_kernel(&adev->umsch_mm.ucode_fw_obj,
174                               &adev->umsch_mm.ucode_fw_gpu_addr,
175                               (void **)&adev->umsch_mm.ucode_fw_ptr);
176         return r;
177 }
178
179 static void umsch_mm_v4_0_aggregated_doorbell_init(struct amdgpu_umsch_mm *umsch)
180 {
181         struct amdgpu_device *adev = umsch->ring.adev;
182         uint32_t data;
183
184         data = RREG32_SOC15(VCN, 0, regVCN_AGDB_CTRL0);
185         data = REG_SET_FIELD(data, VCN_AGDB_CTRL0, OFFSET,
186                umsch->agdb_index[CONTEXT_PRIORITY_LEVEL_REALTIME]);
187         data = REG_SET_FIELD(data, VCN_AGDB_CTRL0, EN, 1);
188         WREG32_SOC15(VCN, 0, regVCN_AGDB_CTRL0, data);
189
190         data = RREG32_SOC15(VCN, 0, regVCN_AGDB_CTRL1);
191         data = REG_SET_FIELD(data, VCN_AGDB_CTRL1, OFFSET,
192                umsch->agdb_index[CONTEXT_PRIORITY_LEVEL_FOCUS]);
193         data = REG_SET_FIELD(data, VCN_AGDB_CTRL1, EN, 1);
194         WREG32_SOC15(VCN, 0, regVCN_AGDB_CTRL1, data);
195
196         data = RREG32_SOC15(VCN, 0, regVCN_AGDB_CTRL2);
197         data = REG_SET_FIELD(data, VCN_AGDB_CTRL2, OFFSET,
198                umsch->agdb_index[CONTEXT_PRIORITY_LEVEL_NORMAL]);
199         data = REG_SET_FIELD(data, VCN_AGDB_CTRL2, EN, 1);
200         WREG32_SOC15(VCN, 0, regVCN_AGDB_CTRL2, data);
201
202         data = RREG32_SOC15(VCN, 0, regVCN_AGDB_CTRL3);
203         data = REG_SET_FIELD(data, VCN_AGDB_CTRL3, OFFSET,
204                umsch->agdb_index[CONTEXT_PRIORITY_LEVEL_IDLE]);
205         data = REG_SET_FIELD(data, VCN_AGDB_CTRL3, EN, 1);
206         WREG32_SOC15(VCN, 0, regVCN_AGDB_CTRL3, data);
207 }
208
209 static int umsch_mm_v4_0_ring_start(struct amdgpu_umsch_mm *umsch)
210 {
211         struct amdgpu_ring *ring = &umsch->ring;
212         struct amdgpu_device *adev = ring->adev;
213         uint32_t data;
214
215         data = RREG32_SOC15(VCN, 0, regVCN_UMSCH_RB_DB_CTRL);
216         data = REG_SET_FIELD(data, VCN_UMSCH_RB_DB_CTRL, OFFSET, ring->doorbell_index);
217         data = REG_SET_FIELD(data, VCN_UMSCH_RB_DB_CTRL, EN, 1);
218         WREG32_SOC15(VCN, 0, regVCN_UMSCH_RB_DB_CTRL, data);
219
220         adev->nbio.funcs->vcn_doorbell_range(adev, ring->use_doorbell,
221                 (adev->doorbell_index.vcn.vcn_ring0_1 << 1), 0);
222
223         WREG32_SOC15(VCN, 0, regVCN_UMSCH_RB_BASE_LO, lower_32_bits(ring->gpu_addr));
224         WREG32_SOC15(VCN, 0, regVCN_UMSCH_RB_BASE_HI, upper_32_bits(ring->gpu_addr));
225
226         WREG32_SOC15(VCN, 0, regVCN_UMSCH_RB_SIZE, ring->ring_size);
227
228         ring->wptr = 0;
229
230         data = RREG32_SOC15(VCN, 0, regVCN_RB_ENABLE);
231         data &= ~(VCN_RB_ENABLE__AUDIO_RB_EN_MASK);
232         WREG32_SOC15(VCN, 0, regVCN_RB_ENABLE, data);
233
234         umsch_mm_v4_0_aggregated_doorbell_init(umsch);
235
236         return 0;
237 }
238
239 static int umsch_mm_v4_0_ring_stop(struct amdgpu_umsch_mm *umsch)
240 {
241         struct amdgpu_ring *ring = &umsch->ring;
242         struct amdgpu_device *adev = ring->adev;
243         uint32_t data;
244
245         data = RREG32_SOC15(VCN, 0, regVCN_RB_ENABLE);
246         data = REG_SET_FIELD(data, VCN_RB_ENABLE, UMSCH_RB_EN, 0);
247         WREG32_SOC15(VCN, 0, regVCN_RB_ENABLE, data);
248
249         data = RREG32_SOC15(VCN, 0, regVCN_UMSCH_RB_DB_CTRL);
250         data = REG_SET_FIELD(data, VCN_UMSCH_RB_DB_CTRL, EN, 0);
251         WREG32_SOC15(VCN, 0, regVCN_UMSCH_RB_DB_CTRL, data);
252
253         if (amdgpu_ip_version(adev, VCN_HWIP, 0) >= IP_VERSION(4, 0, 5)) {
254                 WREG32_SOC15(VCN, 0, regUVD_IPX_DLDO_CONFIG,
255                         2 << UVD_IPX_DLDO_CONFIG__ONO0_PWR_CONFIG__SHIFT);
256                 SOC15_WAIT_ON_RREG(VCN, 0, regUVD_IPX_DLDO_STATUS,
257                         1 << UVD_IPX_DLDO_STATUS__ONO0_PWR_STATUS__SHIFT,
258                         UVD_IPX_DLDO_STATUS__ONO0_PWR_STATUS_MASK);
259         }
260
261         return 0;
262 }
263
264 static int umsch_mm_v4_0_set_hw_resources(struct amdgpu_umsch_mm *umsch)
265 {
266         union UMSCHAPI__SET_HW_RESOURCES set_hw_resources = {};
267         struct amdgpu_device *adev = umsch->ring.adev;
268         int r;
269
270         set_hw_resources.header.type = UMSCH_API_TYPE_SCHEDULER;
271         set_hw_resources.header.opcode = UMSCH_API_SET_HW_RSRC;
272         set_hw_resources.header.dwsize = API_FRAME_SIZE_IN_DWORDS;
273
274         set_hw_resources.vmid_mask_mm_vcn = umsch->vmid_mask_mm_vcn;
275         set_hw_resources.vmid_mask_mm_vpe = umsch->vmid_mask_mm_vpe;
276         set_hw_resources.collaboration_mask_vpe =
277                 adev->vpe.collaborate_mode ? 0x3 : 0x0;
278         set_hw_resources.engine_mask = umsch->engine_mask;
279
280         set_hw_resources.vcn0_hqd_mask[0] = umsch->vcn0_hqd_mask;
281         set_hw_resources.vcn1_hqd_mask[0] = umsch->vcn1_hqd_mask;
282         set_hw_resources.vcn_hqd_mask[0] = umsch->vcn_hqd_mask[0];
283         set_hw_resources.vcn_hqd_mask[1] = umsch->vcn_hqd_mask[1];
284         set_hw_resources.vpe_hqd_mask[0] = umsch->vpe_hqd_mask;
285
286         set_hw_resources.g_sch_ctx_gpu_mc_ptr = umsch->sch_ctx_gpu_addr;
287
288         set_hw_resources.enable_level_process_quantum_check = 1;
289
290         memcpy(set_hw_resources.mmhub_base, adev->reg_offset[MMHUB_HWIP][0],
291                sizeof(uint32_t) * 5);
292         set_hw_resources.mmhub_version =
293                 IP_VERSION_MAJ_MIN_REV(amdgpu_ip_version(adev, MMHUB_HWIP, 0));
294
295         memcpy(set_hw_resources.osssys_base, adev->reg_offset[OSSSYS_HWIP][0],
296                sizeof(uint32_t) * 5);
297         set_hw_resources.osssys_version =
298                 IP_VERSION_MAJ_MIN_REV(amdgpu_ip_version(adev, OSSSYS_HWIP, 0));
299
300         set_hw_resources.vcn_version =
301                 IP_VERSION_MAJ_MIN_REV(amdgpu_ip_version(adev, VCN_HWIP, 0));
302         set_hw_resources.vpe_version =
303                 IP_VERSION_MAJ_MIN_REV(amdgpu_ip_version(adev, VPE_HWIP, 0));
304
305         set_hw_resources.api_status.api_completion_fence_addr = umsch->ring.fence_drv.gpu_addr;
306         set_hw_resources.api_status.api_completion_fence_value = ++umsch->ring.fence_drv.sync_seq;
307
308         r = amdgpu_umsch_mm_submit_pkt(umsch, &set_hw_resources.max_dwords_in_api,
309                                        API_FRAME_SIZE_IN_DWORDS);
310         if (r)
311                 return r;
312
313         r = amdgpu_umsch_mm_query_fence(umsch);
314         if (r) {
315                 dev_err(adev->dev, "UMSCH SET_HW_RESOURCES: Failed\n");
316                 return r;
317         }
318
319         return 0;
320 }
321
322 static int umsch_mm_v4_0_add_queue(struct amdgpu_umsch_mm *umsch,
323                                    struct umsch_mm_add_queue_input *input_ptr)
324 {
325         struct amdgpu_device *adev = umsch->ring.adev;
326         union UMSCHAPI__ADD_QUEUE add_queue = {};
327         int r;
328
329         add_queue.header.type = UMSCH_API_TYPE_SCHEDULER;
330         add_queue.header.opcode = UMSCH_API_ADD_QUEUE;
331         add_queue.header.dwsize = API_FRAME_SIZE_IN_DWORDS;
332
333         add_queue.process_id = input_ptr->process_id;
334         add_queue.page_table_base_addr = input_ptr->page_table_base_addr;
335         add_queue.process_va_start = input_ptr->process_va_start;
336         add_queue.process_va_end = input_ptr->process_va_end;
337         add_queue.process_quantum = input_ptr->process_quantum;
338         add_queue.process_csa_addr = input_ptr->process_csa_addr;
339         add_queue.context_quantum = input_ptr->context_quantum;
340         add_queue.context_csa_addr = input_ptr->context_csa_addr;
341         add_queue.inprocess_context_priority = input_ptr->inprocess_context_priority;
342         add_queue.context_global_priority_level =
343                 (enum UMSCH_AMD_PRIORITY_LEVEL)input_ptr->context_global_priority_level;
344         add_queue.doorbell_offset_0 = input_ptr->doorbell_offset_0;
345         add_queue.doorbell_offset_1 = input_ptr->doorbell_offset_1;
346         add_queue.affinity.u32All = input_ptr->affinity;
347         add_queue.mqd_addr = input_ptr->mqd_addr;
348         add_queue.engine_type = (enum UMSCH_ENGINE_TYPE)input_ptr->engine_type;
349         add_queue.h_context = input_ptr->h_context;
350         add_queue.h_queue = input_ptr->h_queue;
351         add_queue.vm_context_cntl = input_ptr->vm_context_cntl;
352         add_queue.is_context_suspended = input_ptr->is_context_suspended;
353         add_queue.collaboration_mode = adev->vpe.collaborate_mode ? 1 : 0;
354
355         add_queue.api_status.api_completion_fence_addr = umsch->ring.fence_drv.gpu_addr;
356         add_queue.api_status.api_completion_fence_value = ++umsch->ring.fence_drv.sync_seq;
357
358         r = amdgpu_umsch_mm_submit_pkt(umsch, &add_queue.max_dwords_in_api,
359                                        API_FRAME_SIZE_IN_DWORDS);
360         if (r)
361                 return r;
362
363         r = amdgpu_umsch_mm_query_fence(umsch);
364         if (r) {
365                 dev_err(adev->dev, "UMSCH ADD_QUEUE: Failed\n");
366                 return r;
367         }
368
369         return 0;
370 }
371
372 static int umsch_mm_v4_0_remove_queue(struct amdgpu_umsch_mm *umsch,
373                                       struct umsch_mm_remove_queue_input *input_ptr)
374 {
375         union UMSCHAPI__REMOVE_QUEUE remove_queue = {};
376         struct amdgpu_device *adev = umsch->ring.adev;
377         int r;
378
379         remove_queue.header.type = UMSCH_API_TYPE_SCHEDULER;
380         remove_queue.header.opcode = UMSCH_API_REMOVE_QUEUE;
381         remove_queue.header.dwsize = API_FRAME_SIZE_IN_DWORDS;
382
383         remove_queue.doorbell_offset_0 = input_ptr->doorbell_offset_0;
384         remove_queue.doorbell_offset_1 = input_ptr->doorbell_offset_1;
385         remove_queue.context_csa_addr = input_ptr->context_csa_addr;
386
387         remove_queue.api_status.api_completion_fence_addr = umsch->ring.fence_drv.gpu_addr;
388         remove_queue.api_status.api_completion_fence_value = ++umsch->ring.fence_drv.sync_seq;
389
390         r = amdgpu_umsch_mm_submit_pkt(umsch, &remove_queue.max_dwords_in_api,
391                                        API_FRAME_SIZE_IN_DWORDS);
392         if (r)
393                 return r;
394
395         r = amdgpu_umsch_mm_query_fence(umsch);
396         if (r) {
397                 dev_err(adev->dev, "UMSCH REMOVE_QUEUE: Failed\n");
398                 return r;
399         }
400
401         return 0;
402 }
403
404 static int umsch_mm_v4_0_set_regs(struct amdgpu_umsch_mm *umsch)
405 {
406         struct amdgpu_device *adev = container_of(umsch, struct amdgpu_device, umsch_mm);
407
408         umsch->rb_wptr = SOC15_REG_OFFSET(VCN, 0, regVCN_UMSCH_RB_WPTR);
409         umsch->rb_rptr = SOC15_REG_OFFSET(VCN, 0, regVCN_UMSCH_RB_RPTR);
410
411         return 0;
412 }
413
414 static const struct umsch_mm_funcs umsch_mm_v4_0_funcs = {
415         .set_hw_resources = umsch_mm_v4_0_set_hw_resources,
416         .add_queue = umsch_mm_v4_0_add_queue,
417         .remove_queue = umsch_mm_v4_0_remove_queue,
418         .set_regs = umsch_mm_v4_0_set_regs,
419         .init_microcode = amdgpu_umsch_mm_init_microcode,
420         .load_microcode = umsch_mm_v4_0_load_microcode,
421         .ring_init = amdgpu_umsch_mm_ring_init,
422         .ring_start = umsch_mm_v4_0_ring_start,
423         .ring_stop = umsch_mm_v4_0_ring_stop,
424 };
425
426 void umsch_mm_v4_0_set_funcs(struct amdgpu_umsch_mm *umsch)
427 {
428         umsch->funcs = &umsch_mm_v4_0_funcs;
429 }