1*7688df22SAndroid Build Coastguard Worker /*
2*7688df22SAndroid Build Coastguard Worker * Copyright 2017 Advanced Micro Devices, Inc.
3*7688df22SAndroid Build Coastguard Worker *
4*7688df22SAndroid Build Coastguard Worker * Permission is hereby granted, free of charge, to any person obtaining a
5*7688df22SAndroid Build Coastguard Worker * copy of this software and associated documentation files (the "Software"),
6*7688df22SAndroid Build Coastguard Worker * to deal in the Software without restriction, including without limitation
7*7688df22SAndroid Build Coastguard Worker * the rights to use, copy, modify, merge, publish, distribute, sublicense,
8*7688df22SAndroid Build Coastguard Worker * and/or sell copies of the Software, and to permit persons to whom the
9*7688df22SAndroid Build Coastguard Worker * Software is furnished to do so, subject to the following conditions:
10*7688df22SAndroid Build Coastguard Worker *
11*7688df22SAndroid Build Coastguard Worker * The above copyright notice and this permission notice shall be included in
12*7688df22SAndroid Build Coastguard Worker * all copies or substantial portions of the Software.
13*7688df22SAndroid Build Coastguard Worker *
14*7688df22SAndroid Build Coastguard Worker * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
15*7688df22SAndroid Build Coastguard Worker * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
16*7688df22SAndroid Build Coastguard Worker * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
17*7688df22SAndroid Build Coastguard Worker * THE COPYRIGHT HOLDER(S) OR AUTHOR(S) BE LIABLE FOR ANY CLAIM, DAMAGES OR
18*7688df22SAndroid Build Coastguard Worker * OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE,
19*7688df22SAndroid Build Coastguard Worker * ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
20*7688df22SAndroid Build Coastguard Worker * OTHER DEALINGS IN THE SOFTWARE.
21*7688df22SAndroid Build Coastguard Worker *
22*7688df22SAndroid Build Coastguard Worker */
23*7688df22SAndroid Build Coastguard Worker
24*7688df22SAndroid Build Coastguard Worker #include <stdio.h>
25*7688df22SAndroid Build Coastguard Worker #include <inttypes.h>
26*7688df22SAndroid Build Coastguard Worker
27*7688df22SAndroid Build Coastguard Worker #include "CUnit/Basic.h"
28*7688df22SAndroid Build Coastguard Worker
29*7688df22SAndroid Build Coastguard Worker #include "util_math.h"
30*7688df22SAndroid Build Coastguard Worker
31*7688df22SAndroid Build Coastguard Worker #include "amdgpu_test.h"
32*7688df22SAndroid Build Coastguard Worker #include "amdgpu_drm.h"
33*7688df22SAndroid Build Coastguard Worker #include "amdgpu_internal.h"
34*7688df22SAndroid Build Coastguard Worker #include "frame.h"
35*7688df22SAndroid Build Coastguard Worker #include "uve_ib.h"
36*7688df22SAndroid Build Coastguard Worker
37*7688df22SAndroid Build Coastguard Worker #define IB_SIZE 4096
38*7688df22SAndroid Build Coastguard Worker #define MAX_RESOURCES 16
39*7688df22SAndroid Build Coastguard Worker
40*7688df22SAndroid Build Coastguard Worker struct amdgpu_uvd_enc_bo {
41*7688df22SAndroid Build Coastguard Worker amdgpu_bo_handle handle;
42*7688df22SAndroid Build Coastguard Worker amdgpu_va_handle va_handle;
43*7688df22SAndroid Build Coastguard Worker uint64_t addr;
44*7688df22SAndroid Build Coastguard Worker uint64_t size;
45*7688df22SAndroid Build Coastguard Worker uint8_t *ptr;
46*7688df22SAndroid Build Coastguard Worker };
47*7688df22SAndroid Build Coastguard Worker
48*7688df22SAndroid Build Coastguard Worker struct amdgpu_uvd_enc {
49*7688df22SAndroid Build Coastguard Worker unsigned width;
50*7688df22SAndroid Build Coastguard Worker unsigned height;
51*7688df22SAndroid Build Coastguard Worker struct amdgpu_uvd_enc_bo session;
52*7688df22SAndroid Build Coastguard Worker struct amdgpu_uvd_enc_bo vbuf;
53*7688df22SAndroid Build Coastguard Worker struct amdgpu_uvd_enc_bo bs;
54*7688df22SAndroid Build Coastguard Worker struct amdgpu_uvd_enc_bo fb;
55*7688df22SAndroid Build Coastguard Worker struct amdgpu_uvd_enc_bo cpb;
56*7688df22SAndroid Build Coastguard Worker };
57*7688df22SAndroid Build Coastguard Worker
58*7688df22SAndroid Build Coastguard Worker static amdgpu_device_handle device_handle;
59*7688df22SAndroid Build Coastguard Worker static uint32_t major_version;
60*7688df22SAndroid Build Coastguard Worker static uint32_t minor_version;
61*7688df22SAndroid Build Coastguard Worker static uint32_t family_id;
62*7688df22SAndroid Build Coastguard Worker
63*7688df22SAndroid Build Coastguard Worker static amdgpu_context_handle context_handle;
64*7688df22SAndroid Build Coastguard Worker static amdgpu_bo_handle ib_handle;
65*7688df22SAndroid Build Coastguard Worker static amdgpu_va_handle ib_va_handle;
66*7688df22SAndroid Build Coastguard Worker static uint64_t ib_mc_address;
67*7688df22SAndroid Build Coastguard Worker static uint32_t *ib_cpu;
68*7688df22SAndroid Build Coastguard Worker
69*7688df22SAndroid Build Coastguard Worker static struct amdgpu_uvd_enc enc;
70*7688df22SAndroid Build Coastguard Worker static amdgpu_bo_handle resources[MAX_RESOURCES];
71*7688df22SAndroid Build Coastguard Worker static unsigned num_resources;
72*7688df22SAndroid Build Coastguard Worker
73*7688df22SAndroid Build Coastguard Worker static void amdgpu_cs_uvd_enc_create(void);
74*7688df22SAndroid Build Coastguard Worker static void amdgpu_cs_uvd_enc_session_init(void);
75*7688df22SAndroid Build Coastguard Worker static void amdgpu_cs_uvd_enc_encode(void);
76*7688df22SAndroid Build Coastguard Worker static void amdgpu_cs_uvd_enc_destroy(void);
77*7688df22SAndroid Build Coastguard Worker
78*7688df22SAndroid Build Coastguard Worker
79*7688df22SAndroid Build Coastguard Worker CU_TestInfo uvd_enc_tests[] = {
80*7688df22SAndroid Build Coastguard Worker { "UVD ENC create", amdgpu_cs_uvd_enc_create },
81*7688df22SAndroid Build Coastguard Worker { "UVD ENC session init", amdgpu_cs_uvd_enc_session_init },
82*7688df22SAndroid Build Coastguard Worker { "UVD ENC encode", amdgpu_cs_uvd_enc_encode },
83*7688df22SAndroid Build Coastguard Worker { "UVD ENC destroy", amdgpu_cs_uvd_enc_destroy },
84*7688df22SAndroid Build Coastguard Worker CU_TEST_INFO_NULL,
85*7688df22SAndroid Build Coastguard Worker };
86*7688df22SAndroid Build Coastguard Worker
suite_uvd_enc_tests_enable(void)87*7688df22SAndroid Build Coastguard Worker CU_BOOL suite_uvd_enc_tests_enable(void)
88*7688df22SAndroid Build Coastguard Worker {
89*7688df22SAndroid Build Coastguard Worker int r;
90*7688df22SAndroid Build Coastguard Worker struct drm_amdgpu_info_hw_ip info;
91*7688df22SAndroid Build Coastguard Worker
92*7688df22SAndroid Build Coastguard Worker if (amdgpu_device_initialize(drm_amdgpu[0], &major_version,
93*7688df22SAndroid Build Coastguard Worker &minor_version, &device_handle))
94*7688df22SAndroid Build Coastguard Worker return CU_FALSE;
95*7688df22SAndroid Build Coastguard Worker
96*7688df22SAndroid Build Coastguard Worker r = amdgpu_query_hw_ip_info(device_handle, AMDGPU_HW_IP_UVD_ENC, 0, &info);
97*7688df22SAndroid Build Coastguard Worker
98*7688df22SAndroid Build Coastguard Worker if (amdgpu_device_deinitialize(device_handle))
99*7688df22SAndroid Build Coastguard Worker return CU_FALSE;
100*7688df22SAndroid Build Coastguard Worker
101*7688df22SAndroid Build Coastguard Worker if (!info.available_rings)
102*7688df22SAndroid Build Coastguard Worker printf("\n\nThe ASIC NOT support UVD ENC, suite disabled.\n");
103*7688df22SAndroid Build Coastguard Worker
104*7688df22SAndroid Build Coastguard Worker return (r == 0 && (info.available_rings ? CU_TRUE : CU_FALSE));
105*7688df22SAndroid Build Coastguard Worker }
106*7688df22SAndroid Build Coastguard Worker
107*7688df22SAndroid Build Coastguard Worker
suite_uvd_enc_tests_init(void)108*7688df22SAndroid Build Coastguard Worker int suite_uvd_enc_tests_init(void)
109*7688df22SAndroid Build Coastguard Worker {
110*7688df22SAndroid Build Coastguard Worker int r;
111*7688df22SAndroid Build Coastguard Worker
112*7688df22SAndroid Build Coastguard Worker r = amdgpu_device_initialize(drm_amdgpu[0], &major_version,
113*7688df22SAndroid Build Coastguard Worker &minor_version, &device_handle);
114*7688df22SAndroid Build Coastguard Worker if (r)
115*7688df22SAndroid Build Coastguard Worker return CUE_SINIT_FAILED;
116*7688df22SAndroid Build Coastguard Worker
117*7688df22SAndroid Build Coastguard Worker family_id = device_handle->info.family_id;
118*7688df22SAndroid Build Coastguard Worker
119*7688df22SAndroid Build Coastguard Worker r = amdgpu_cs_ctx_create(device_handle, &context_handle);
120*7688df22SAndroid Build Coastguard Worker if (r)
121*7688df22SAndroid Build Coastguard Worker return CUE_SINIT_FAILED;
122*7688df22SAndroid Build Coastguard Worker
123*7688df22SAndroid Build Coastguard Worker r = amdgpu_bo_alloc_and_map(device_handle, IB_SIZE, 4096,
124*7688df22SAndroid Build Coastguard Worker AMDGPU_GEM_DOMAIN_GTT, 0,
125*7688df22SAndroid Build Coastguard Worker &ib_handle, (void**)&ib_cpu,
126*7688df22SAndroid Build Coastguard Worker &ib_mc_address, &ib_va_handle);
127*7688df22SAndroid Build Coastguard Worker if (r)
128*7688df22SAndroid Build Coastguard Worker return CUE_SINIT_FAILED;
129*7688df22SAndroid Build Coastguard Worker
130*7688df22SAndroid Build Coastguard Worker return CUE_SUCCESS;
131*7688df22SAndroid Build Coastguard Worker }
132*7688df22SAndroid Build Coastguard Worker
suite_uvd_enc_tests_clean(void)133*7688df22SAndroid Build Coastguard Worker int suite_uvd_enc_tests_clean(void)
134*7688df22SAndroid Build Coastguard Worker {
135*7688df22SAndroid Build Coastguard Worker int r;
136*7688df22SAndroid Build Coastguard Worker
137*7688df22SAndroid Build Coastguard Worker r = amdgpu_bo_unmap_and_free(ib_handle, ib_va_handle,
138*7688df22SAndroid Build Coastguard Worker ib_mc_address, IB_SIZE);
139*7688df22SAndroid Build Coastguard Worker if (r)
140*7688df22SAndroid Build Coastguard Worker return CUE_SCLEAN_FAILED;
141*7688df22SAndroid Build Coastguard Worker
142*7688df22SAndroid Build Coastguard Worker r = amdgpu_cs_ctx_free(context_handle);
143*7688df22SAndroid Build Coastguard Worker if (r)
144*7688df22SAndroid Build Coastguard Worker return CUE_SCLEAN_FAILED;
145*7688df22SAndroid Build Coastguard Worker
146*7688df22SAndroid Build Coastguard Worker r = amdgpu_device_deinitialize(device_handle);
147*7688df22SAndroid Build Coastguard Worker if (r)
148*7688df22SAndroid Build Coastguard Worker return CUE_SCLEAN_FAILED;
149*7688df22SAndroid Build Coastguard Worker
150*7688df22SAndroid Build Coastguard Worker return CUE_SUCCESS;
151*7688df22SAndroid Build Coastguard Worker }
152*7688df22SAndroid Build Coastguard Worker
submit(unsigned ndw,unsigned ip)153*7688df22SAndroid Build Coastguard Worker static int submit(unsigned ndw, unsigned ip)
154*7688df22SAndroid Build Coastguard Worker {
155*7688df22SAndroid Build Coastguard Worker struct amdgpu_cs_request ibs_request = {0};
156*7688df22SAndroid Build Coastguard Worker struct amdgpu_cs_ib_info ib_info = {0};
157*7688df22SAndroid Build Coastguard Worker struct amdgpu_cs_fence fence_status = {0};
158*7688df22SAndroid Build Coastguard Worker uint32_t expired;
159*7688df22SAndroid Build Coastguard Worker int r;
160*7688df22SAndroid Build Coastguard Worker
161*7688df22SAndroid Build Coastguard Worker ib_info.ib_mc_address = ib_mc_address;
162*7688df22SAndroid Build Coastguard Worker ib_info.size = ndw;
163*7688df22SAndroid Build Coastguard Worker
164*7688df22SAndroid Build Coastguard Worker ibs_request.ip_type = ip;
165*7688df22SAndroid Build Coastguard Worker
166*7688df22SAndroid Build Coastguard Worker r = amdgpu_bo_list_create(device_handle, num_resources, resources,
167*7688df22SAndroid Build Coastguard Worker NULL, &ibs_request.resources);
168*7688df22SAndroid Build Coastguard Worker if (r)
169*7688df22SAndroid Build Coastguard Worker return r;
170*7688df22SAndroid Build Coastguard Worker
171*7688df22SAndroid Build Coastguard Worker ibs_request.number_of_ibs = 1;
172*7688df22SAndroid Build Coastguard Worker ibs_request.ibs = &ib_info;
173*7688df22SAndroid Build Coastguard Worker ibs_request.fence_info.handle = NULL;
174*7688df22SAndroid Build Coastguard Worker
175*7688df22SAndroid Build Coastguard Worker r = amdgpu_cs_submit(context_handle, 0, &ibs_request, 1);
176*7688df22SAndroid Build Coastguard Worker if (r)
177*7688df22SAndroid Build Coastguard Worker return r;
178*7688df22SAndroid Build Coastguard Worker
179*7688df22SAndroid Build Coastguard Worker r = amdgpu_bo_list_destroy(ibs_request.resources);
180*7688df22SAndroid Build Coastguard Worker if (r)
181*7688df22SAndroid Build Coastguard Worker return r;
182*7688df22SAndroid Build Coastguard Worker
183*7688df22SAndroid Build Coastguard Worker fence_status.context = context_handle;
184*7688df22SAndroid Build Coastguard Worker fence_status.ip_type = ip;
185*7688df22SAndroid Build Coastguard Worker fence_status.fence = ibs_request.seq_no;
186*7688df22SAndroid Build Coastguard Worker
187*7688df22SAndroid Build Coastguard Worker r = amdgpu_cs_query_fence_status(&fence_status,
188*7688df22SAndroid Build Coastguard Worker AMDGPU_TIMEOUT_INFINITE,
189*7688df22SAndroid Build Coastguard Worker 0, &expired);
190*7688df22SAndroid Build Coastguard Worker if (r)
191*7688df22SAndroid Build Coastguard Worker return r;
192*7688df22SAndroid Build Coastguard Worker
193*7688df22SAndroid Build Coastguard Worker return 0;
194*7688df22SAndroid Build Coastguard Worker }
195*7688df22SAndroid Build Coastguard Worker
alloc_resource(struct amdgpu_uvd_enc_bo * uvd_enc_bo,unsigned size,unsigned domain)196*7688df22SAndroid Build Coastguard Worker static void alloc_resource(struct amdgpu_uvd_enc_bo *uvd_enc_bo,
197*7688df22SAndroid Build Coastguard Worker unsigned size, unsigned domain)
198*7688df22SAndroid Build Coastguard Worker {
199*7688df22SAndroid Build Coastguard Worker struct amdgpu_bo_alloc_request req = {0};
200*7688df22SAndroid Build Coastguard Worker amdgpu_bo_handle buf_handle;
201*7688df22SAndroid Build Coastguard Worker amdgpu_va_handle va_handle;
202*7688df22SAndroid Build Coastguard Worker uint64_t va = 0;
203*7688df22SAndroid Build Coastguard Worker int r;
204*7688df22SAndroid Build Coastguard Worker
205*7688df22SAndroid Build Coastguard Worker req.alloc_size = ALIGN(size, 4096);
206*7688df22SAndroid Build Coastguard Worker req.preferred_heap = domain;
207*7688df22SAndroid Build Coastguard Worker r = amdgpu_bo_alloc(device_handle, &req, &buf_handle);
208*7688df22SAndroid Build Coastguard Worker CU_ASSERT_EQUAL(r, 0);
209*7688df22SAndroid Build Coastguard Worker r = amdgpu_va_range_alloc(device_handle,
210*7688df22SAndroid Build Coastguard Worker amdgpu_gpu_va_range_general,
211*7688df22SAndroid Build Coastguard Worker req.alloc_size, 1, 0, &va,
212*7688df22SAndroid Build Coastguard Worker &va_handle, 0);
213*7688df22SAndroid Build Coastguard Worker CU_ASSERT_EQUAL(r, 0);
214*7688df22SAndroid Build Coastguard Worker r = amdgpu_bo_va_op(buf_handle, 0, req.alloc_size, va, 0,
215*7688df22SAndroid Build Coastguard Worker AMDGPU_VA_OP_MAP);
216*7688df22SAndroid Build Coastguard Worker CU_ASSERT_EQUAL(r, 0);
217*7688df22SAndroid Build Coastguard Worker uvd_enc_bo->addr = va;
218*7688df22SAndroid Build Coastguard Worker uvd_enc_bo->handle = buf_handle;
219*7688df22SAndroid Build Coastguard Worker uvd_enc_bo->size = req.alloc_size;
220*7688df22SAndroid Build Coastguard Worker uvd_enc_bo->va_handle = va_handle;
221*7688df22SAndroid Build Coastguard Worker r = amdgpu_bo_cpu_map(uvd_enc_bo->handle, (void **)&uvd_enc_bo->ptr);
222*7688df22SAndroid Build Coastguard Worker CU_ASSERT_EQUAL(r, 0);
223*7688df22SAndroid Build Coastguard Worker memset(uvd_enc_bo->ptr, 0, size);
224*7688df22SAndroid Build Coastguard Worker r = amdgpu_bo_cpu_unmap(uvd_enc_bo->handle);
225*7688df22SAndroid Build Coastguard Worker CU_ASSERT_EQUAL(r, 0);
226*7688df22SAndroid Build Coastguard Worker }
227*7688df22SAndroid Build Coastguard Worker
free_resource(struct amdgpu_uvd_enc_bo * uvd_enc_bo)228*7688df22SAndroid Build Coastguard Worker static void free_resource(struct amdgpu_uvd_enc_bo *uvd_enc_bo)
229*7688df22SAndroid Build Coastguard Worker {
230*7688df22SAndroid Build Coastguard Worker int r;
231*7688df22SAndroid Build Coastguard Worker
232*7688df22SAndroid Build Coastguard Worker r = amdgpu_bo_va_op(uvd_enc_bo->handle, 0, uvd_enc_bo->size,
233*7688df22SAndroid Build Coastguard Worker uvd_enc_bo->addr, 0, AMDGPU_VA_OP_UNMAP);
234*7688df22SAndroid Build Coastguard Worker CU_ASSERT_EQUAL(r, 0);
235*7688df22SAndroid Build Coastguard Worker
236*7688df22SAndroid Build Coastguard Worker r = amdgpu_va_range_free(uvd_enc_bo->va_handle);
237*7688df22SAndroid Build Coastguard Worker CU_ASSERT_EQUAL(r, 0);
238*7688df22SAndroid Build Coastguard Worker
239*7688df22SAndroid Build Coastguard Worker r = amdgpu_bo_free(uvd_enc_bo->handle);
240*7688df22SAndroid Build Coastguard Worker CU_ASSERT_EQUAL(r, 0);
241*7688df22SAndroid Build Coastguard Worker memset(uvd_enc_bo, 0, sizeof(*uvd_enc_bo));
242*7688df22SAndroid Build Coastguard Worker }
243*7688df22SAndroid Build Coastguard Worker
amdgpu_cs_uvd_enc_create(void)244*7688df22SAndroid Build Coastguard Worker static void amdgpu_cs_uvd_enc_create(void)
245*7688df22SAndroid Build Coastguard Worker {
246*7688df22SAndroid Build Coastguard Worker enc.width = 160;
247*7688df22SAndroid Build Coastguard Worker enc.height = 128;
248*7688df22SAndroid Build Coastguard Worker
249*7688df22SAndroid Build Coastguard Worker num_resources = 0;
250*7688df22SAndroid Build Coastguard Worker alloc_resource(&enc.session, 128 * 1024, AMDGPU_GEM_DOMAIN_GTT);
251*7688df22SAndroid Build Coastguard Worker resources[num_resources++] = enc.session.handle;
252*7688df22SAndroid Build Coastguard Worker resources[num_resources++] = ib_handle;
253*7688df22SAndroid Build Coastguard Worker }
254*7688df22SAndroid Build Coastguard Worker
check_result(struct amdgpu_uvd_enc * enc)255*7688df22SAndroid Build Coastguard Worker static void check_result(struct amdgpu_uvd_enc *enc)
256*7688df22SAndroid Build Coastguard Worker {
257*7688df22SAndroid Build Coastguard Worker uint64_t sum;
258*7688df22SAndroid Build Coastguard Worker uint32_t s = 175602;
259*7688df22SAndroid Build Coastguard Worker uint32_t *ptr, size;
260*7688df22SAndroid Build Coastguard Worker int j, r;
261*7688df22SAndroid Build Coastguard Worker
262*7688df22SAndroid Build Coastguard Worker r = amdgpu_bo_cpu_map(enc->fb.handle, (void **)&enc->fb.ptr);
263*7688df22SAndroid Build Coastguard Worker CU_ASSERT_EQUAL(r, 0);
264*7688df22SAndroid Build Coastguard Worker ptr = (uint32_t *)enc->fb.ptr;
265*7688df22SAndroid Build Coastguard Worker size = ptr[6];
266*7688df22SAndroid Build Coastguard Worker r = amdgpu_bo_cpu_unmap(enc->fb.handle);
267*7688df22SAndroid Build Coastguard Worker CU_ASSERT_EQUAL(r, 0);
268*7688df22SAndroid Build Coastguard Worker r = amdgpu_bo_cpu_map(enc->bs.handle, (void **)&enc->bs.ptr);
269*7688df22SAndroid Build Coastguard Worker CU_ASSERT_EQUAL(r, 0);
270*7688df22SAndroid Build Coastguard Worker for (j = 0, sum = 0; j < size; ++j)
271*7688df22SAndroid Build Coastguard Worker sum += enc->bs.ptr[j];
272*7688df22SAndroid Build Coastguard Worker CU_ASSERT_EQUAL(sum, s);
273*7688df22SAndroid Build Coastguard Worker r = amdgpu_bo_cpu_unmap(enc->bs.handle);
274*7688df22SAndroid Build Coastguard Worker CU_ASSERT_EQUAL(r, 0);
275*7688df22SAndroid Build Coastguard Worker
276*7688df22SAndroid Build Coastguard Worker }
277*7688df22SAndroid Build Coastguard Worker
amdgpu_cs_uvd_enc_session_init(void)278*7688df22SAndroid Build Coastguard Worker static void amdgpu_cs_uvd_enc_session_init(void)
279*7688df22SAndroid Build Coastguard Worker {
280*7688df22SAndroid Build Coastguard Worker int len, r;
281*7688df22SAndroid Build Coastguard Worker
282*7688df22SAndroid Build Coastguard Worker len = 0;
283*7688df22SAndroid Build Coastguard Worker memcpy((ib_cpu + len), uve_session_info, sizeof(uve_session_info));
284*7688df22SAndroid Build Coastguard Worker len += sizeof(uve_session_info) / 4;
285*7688df22SAndroid Build Coastguard Worker ib_cpu[len++] = enc.session.addr >> 32;
286*7688df22SAndroid Build Coastguard Worker ib_cpu[len++] = enc.session.addr;
287*7688df22SAndroid Build Coastguard Worker
288*7688df22SAndroid Build Coastguard Worker memcpy((ib_cpu + len), uve_task_info, sizeof(uve_task_info));
289*7688df22SAndroid Build Coastguard Worker len += sizeof(uve_task_info) / 4;
290*7688df22SAndroid Build Coastguard Worker ib_cpu[len++] = 0x000000d8;
291*7688df22SAndroid Build Coastguard Worker ib_cpu[len++] = 0x00000000;
292*7688df22SAndroid Build Coastguard Worker ib_cpu[len++] = 0x00000000;
293*7688df22SAndroid Build Coastguard Worker
294*7688df22SAndroid Build Coastguard Worker memcpy((ib_cpu + len), uve_op_init, sizeof(uve_op_init));
295*7688df22SAndroid Build Coastguard Worker len += sizeof(uve_op_init) / 4;
296*7688df22SAndroid Build Coastguard Worker
297*7688df22SAndroid Build Coastguard Worker memcpy((ib_cpu + len), uve_session_init, sizeof(uve_session_init));
298*7688df22SAndroid Build Coastguard Worker len += sizeof(uve_session_init) / 4;
299*7688df22SAndroid Build Coastguard Worker
300*7688df22SAndroid Build Coastguard Worker memcpy((ib_cpu + len), uve_layer_ctrl, sizeof(uve_layer_ctrl));
301*7688df22SAndroid Build Coastguard Worker len += sizeof(uve_layer_ctrl) / 4;
302*7688df22SAndroid Build Coastguard Worker
303*7688df22SAndroid Build Coastguard Worker memcpy((ib_cpu + len), uve_slice_ctrl, sizeof(uve_slice_ctrl));
304*7688df22SAndroid Build Coastguard Worker len += sizeof(uve_slice_ctrl) / 4;
305*7688df22SAndroid Build Coastguard Worker
306*7688df22SAndroid Build Coastguard Worker memcpy((ib_cpu + len), uve_spec_misc, sizeof(uve_spec_misc));
307*7688df22SAndroid Build Coastguard Worker len += sizeof(uve_spec_misc) / 4;
308*7688df22SAndroid Build Coastguard Worker
309*7688df22SAndroid Build Coastguard Worker memcpy((ib_cpu + len), uve_rc_session_init, sizeof(uve_rc_session_init));
310*7688df22SAndroid Build Coastguard Worker len += sizeof(uve_rc_session_init) / 4;
311*7688df22SAndroid Build Coastguard Worker
312*7688df22SAndroid Build Coastguard Worker memcpy((ib_cpu + len), uve_deblocking_filter, sizeof(uve_deblocking_filter));
313*7688df22SAndroid Build Coastguard Worker len += sizeof(uve_deblocking_filter) / 4;
314*7688df22SAndroid Build Coastguard Worker
315*7688df22SAndroid Build Coastguard Worker memcpy((ib_cpu + len), uve_quality_params, sizeof(uve_quality_params));
316*7688df22SAndroid Build Coastguard Worker len += sizeof(uve_quality_params) / 4;
317*7688df22SAndroid Build Coastguard Worker
318*7688df22SAndroid Build Coastguard Worker memcpy((ib_cpu + len), uve_op_init_rc, sizeof(uve_op_init_rc));
319*7688df22SAndroid Build Coastguard Worker len += sizeof(uve_op_init_rc) / 4;
320*7688df22SAndroid Build Coastguard Worker
321*7688df22SAndroid Build Coastguard Worker memcpy((ib_cpu + len), uve_op_init_rc_vbv_level, sizeof(uve_op_init_rc_vbv_level));
322*7688df22SAndroid Build Coastguard Worker len += sizeof(uve_op_init_rc_vbv_level) / 4;
323*7688df22SAndroid Build Coastguard Worker
324*7688df22SAndroid Build Coastguard Worker r = submit(len, AMDGPU_HW_IP_UVD_ENC);
325*7688df22SAndroid Build Coastguard Worker CU_ASSERT_EQUAL(r, 0);
326*7688df22SAndroid Build Coastguard Worker }
327*7688df22SAndroid Build Coastguard Worker
amdgpu_cs_uvd_enc_encode(void)328*7688df22SAndroid Build Coastguard Worker static void amdgpu_cs_uvd_enc_encode(void)
329*7688df22SAndroid Build Coastguard Worker {
330*7688df22SAndroid Build Coastguard Worker int len, r, i;
331*7688df22SAndroid Build Coastguard Worker uint64_t luma_offset, chroma_offset;
332*7688df22SAndroid Build Coastguard Worker uint32_t vbuf_size, bs_size = 0x003f4800, cpb_size;
333*7688df22SAndroid Build Coastguard Worker unsigned align = (family_id >= AMDGPU_FAMILY_AI) ? 256 : 16;
334*7688df22SAndroid Build Coastguard Worker vbuf_size = ALIGN(enc.width, align) * ALIGN(enc.height, 16) * 1.5;
335*7688df22SAndroid Build Coastguard Worker cpb_size = vbuf_size * 10;
336*7688df22SAndroid Build Coastguard Worker
337*7688df22SAndroid Build Coastguard Worker
338*7688df22SAndroid Build Coastguard Worker num_resources = 0;
339*7688df22SAndroid Build Coastguard Worker alloc_resource(&enc.fb, 4096, AMDGPU_GEM_DOMAIN_VRAM);
340*7688df22SAndroid Build Coastguard Worker resources[num_resources++] = enc.fb.handle;
341*7688df22SAndroid Build Coastguard Worker alloc_resource(&enc.bs, bs_size, AMDGPU_GEM_DOMAIN_VRAM);
342*7688df22SAndroid Build Coastguard Worker resources[num_resources++] = enc.bs.handle;
343*7688df22SAndroid Build Coastguard Worker alloc_resource(&enc.vbuf, vbuf_size, AMDGPU_GEM_DOMAIN_VRAM);
344*7688df22SAndroid Build Coastguard Worker resources[num_resources++] = enc.vbuf.handle;
345*7688df22SAndroid Build Coastguard Worker alloc_resource(&enc.cpb, cpb_size, AMDGPU_GEM_DOMAIN_VRAM);
346*7688df22SAndroid Build Coastguard Worker resources[num_resources++] = enc.cpb.handle;
347*7688df22SAndroid Build Coastguard Worker resources[num_resources++] = ib_handle;
348*7688df22SAndroid Build Coastguard Worker
349*7688df22SAndroid Build Coastguard Worker r = amdgpu_bo_cpu_map(enc.vbuf.handle, (void **)&enc.vbuf.ptr);
350*7688df22SAndroid Build Coastguard Worker CU_ASSERT_EQUAL(r, 0);
351*7688df22SAndroid Build Coastguard Worker
352*7688df22SAndroid Build Coastguard Worker memset(enc.vbuf.ptr, 0, vbuf_size);
353*7688df22SAndroid Build Coastguard Worker for (i = 0; i < enc.height; ++i) {
354*7688df22SAndroid Build Coastguard Worker memcpy(enc.vbuf.ptr, (frame + i * enc.width), enc.width);
355*7688df22SAndroid Build Coastguard Worker enc.vbuf.ptr += ALIGN(enc.width, align);
356*7688df22SAndroid Build Coastguard Worker }
357*7688df22SAndroid Build Coastguard Worker for (i = 0; i < enc.height / 2; ++i) {
358*7688df22SAndroid Build Coastguard Worker memcpy(enc.vbuf.ptr, ((frame + enc.height * enc.width) + i * enc.width), enc.width);
359*7688df22SAndroid Build Coastguard Worker enc.vbuf.ptr += ALIGN(enc.width, align);
360*7688df22SAndroid Build Coastguard Worker }
361*7688df22SAndroid Build Coastguard Worker
362*7688df22SAndroid Build Coastguard Worker r = amdgpu_bo_cpu_unmap(enc.vbuf.handle);
363*7688df22SAndroid Build Coastguard Worker CU_ASSERT_EQUAL(r, 0);
364*7688df22SAndroid Build Coastguard Worker
365*7688df22SAndroid Build Coastguard Worker len = 0;
366*7688df22SAndroid Build Coastguard Worker memcpy((ib_cpu + len), uve_session_info, sizeof(uve_session_info));
367*7688df22SAndroid Build Coastguard Worker len += sizeof(uve_session_info) / 4;
368*7688df22SAndroid Build Coastguard Worker ib_cpu[len++] = enc.session.addr >> 32;
369*7688df22SAndroid Build Coastguard Worker ib_cpu[len++] = enc.session.addr;
370*7688df22SAndroid Build Coastguard Worker
371*7688df22SAndroid Build Coastguard Worker memcpy((ib_cpu + len), uve_task_info, sizeof(uve_task_info));
372*7688df22SAndroid Build Coastguard Worker len += sizeof(uve_task_info) / 4;
373*7688df22SAndroid Build Coastguard Worker ib_cpu[len++] = 0x000005e0;
374*7688df22SAndroid Build Coastguard Worker ib_cpu[len++] = 0x00000001;
375*7688df22SAndroid Build Coastguard Worker ib_cpu[len++] = 0x00000001;
376*7688df22SAndroid Build Coastguard Worker
377*7688df22SAndroid Build Coastguard Worker memcpy((ib_cpu + len), uve_nalu_buffer_1, sizeof(uve_nalu_buffer_1));
378*7688df22SAndroid Build Coastguard Worker len += sizeof(uve_nalu_buffer_1) / 4;
379*7688df22SAndroid Build Coastguard Worker
380*7688df22SAndroid Build Coastguard Worker memcpy((ib_cpu + len), uve_nalu_buffer_2, sizeof(uve_nalu_buffer_2));
381*7688df22SAndroid Build Coastguard Worker len += sizeof(uve_nalu_buffer_2) / 4;
382*7688df22SAndroid Build Coastguard Worker
383*7688df22SAndroid Build Coastguard Worker memcpy((ib_cpu + len), uve_nalu_buffer_3, sizeof(uve_nalu_buffer_3));
384*7688df22SAndroid Build Coastguard Worker len += sizeof(uve_nalu_buffer_3) / 4;
385*7688df22SAndroid Build Coastguard Worker
386*7688df22SAndroid Build Coastguard Worker memcpy((ib_cpu + len), uve_nalu_buffer_4, sizeof(uve_nalu_buffer_4));
387*7688df22SAndroid Build Coastguard Worker len += sizeof(uve_nalu_buffer_4) / 4;
388*7688df22SAndroid Build Coastguard Worker
389*7688df22SAndroid Build Coastguard Worker memcpy((ib_cpu + len), uve_slice_header, sizeof(uve_slice_header));
390*7688df22SAndroid Build Coastguard Worker len += sizeof(uve_slice_header) / 4;
391*7688df22SAndroid Build Coastguard Worker
392*7688df22SAndroid Build Coastguard Worker ib_cpu[len++] = 0x00000254;
393*7688df22SAndroid Build Coastguard Worker ib_cpu[len++] = 0x00000010;
394*7688df22SAndroid Build Coastguard Worker ib_cpu[len++] = enc.cpb.addr >> 32;
395*7688df22SAndroid Build Coastguard Worker ib_cpu[len++] = enc.cpb.addr;
396*7688df22SAndroid Build Coastguard Worker memcpy((ib_cpu + len), uve_ctx_buffer, sizeof(uve_ctx_buffer));
397*7688df22SAndroid Build Coastguard Worker len += sizeof(uve_ctx_buffer) / 4;
398*7688df22SAndroid Build Coastguard Worker
399*7688df22SAndroid Build Coastguard Worker memcpy((ib_cpu + len), uve_bitstream_buffer, sizeof(uve_bitstream_buffer));
400*7688df22SAndroid Build Coastguard Worker len += sizeof(uve_bitstream_buffer) / 4;
401*7688df22SAndroid Build Coastguard Worker ib_cpu[len++] = 0x00000000;
402*7688df22SAndroid Build Coastguard Worker ib_cpu[len++] = enc.bs.addr >> 32;
403*7688df22SAndroid Build Coastguard Worker ib_cpu[len++] = enc.bs.addr;
404*7688df22SAndroid Build Coastguard Worker ib_cpu[len++] = 0x003f4800;
405*7688df22SAndroid Build Coastguard Worker ib_cpu[len++] = 0x00000000;
406*7688df22SAndroid Build Coastguard Worker
407*7688df22SAndroid Build Coastguard Worker memcpy((ib_cpu + len), uve_feedback_buffer, sizeof(uve_feedback_buffer));
408*7688df22SAndroid Build Coastguard Worker len += sizeof(uve_feedback_buffer) / 4;
409*7688df22SAndroid Build Coastguard Worker ib_cpu[len++] = enc.fb.addr >> 32;
410*7688df22SAndroid Build Coastguard Worker ib_cpu[len++] = enc.fb.addr;
411*7688df22SAndroid Build Coastguard Worker ib_cpu[len++] = 0x00000010;
412*7688df22SAndroid Build Coastguard Worker ib_cpu[len++] = 0x00000028;
413*7688df22SAndroid Build Coastguard Worker
414*7688df22SAndroid Build Coastguard Worker memcpy((ib_cpu + len), uve_feedback_buffer_additional, sizeof(uve_feedback_buffer_additional));
415*7688df22SAndroid Build Coastguard Worker len += sizeof(uve_feedback_buffer_additional) / 4;
416*7688df22SAndroid Build Coastguard Worker
417*7688df22SAndroid Build Coastguard Worker memcpy((ib_cpu + len), uve_intra_refresh, sizeof(uve_intra_refresh));
418*7688df22SAndroid Build Coastguard Worker len += sizeof(uve_intra_refresh) / 4;
419*7688df22SAndroid Build Coastguard Worker
420*7688df22SAndroid Build Coastguard Worker memcpy((ib_cpu + len), uve_layer_select, sizeof(uve_layer_select));
421*7688df22SAndroid Build Coastguard Worker len += sizeof(uve_layer_select) / 4;
422*7688df22SAndroid Build Coastguard Worker
423*7688df22SAndroid Build Coastguard Worker memcpy((ib_cpu + len), uve_rc_layer_init, sizeof(uve_rc_layer_init));
424*7688df22SAndroid Build Coastguard Worker len += sizeof(uve_rc_layer_init) / 4;
425*7688df22SAndroid Build Coastguard Worker
426*7688df22SAndroid Build Coastguard Worker memcpy((ib_cpu + len), uve_layer_select, sizeof(uve_layer_select));
427*7688df22SAndroid Build Coastguard Worker len += sizeof(uve_layer_select) / 4;
428*7688df22SAndroid Build Coastguard Worker
429*7688df22SAndroid Build Coastguard Worker memcpy((ib_cpu + len), uve_rc_per_pic, sizeof(uve_rc_per_pic));
430*7688df22SAndroid Build Coastguard Worker len += sizeof(uve_rc_per_pic) / 4;
431*7688df22SAndroid Build Coastguard Worker
432*7688df22SAndroid Build Coastguard Worker unsigned luma_size = ALIGN(enc.width, align) * ALIGN(enc.height, 16);
433*7688df22SAndroid Build Coastguard Worker luma_offset = enc.vbuf.addr;
434*7688df22SAndroid Build Coastguard Worker chroma_offset = luma_offset + luma_size;
435*7688df22SAndroid Build Coastguard Worker ib_cpu[len++] = 0x00000054;
436*7688df22SAndroid Build Coastguard Worker ib_cpu[len++] = 0x0000000c;
437*7688df22SAndroid Build Coastguard Worker ib_cpu[len++] = 0x00000002;
438*7688df22SAndroid Build Coastguard Worker ib_cpu[len++] = 0x003f4800;
439*7688df22SAndroid Build Coastguard Worker ib_cpu[len++] = luma_offset >> 32;
440*7688df22SAndroid Build Coastguard Worker ib_cpu[len++] = luma_offset;
441*7688df22SAndroid Build Coastguard Worker ib_cpu[len++] = chroma_offset >> 32;
442*7688df22SAndroid Build Coastguard Worker ib_cpu[len++] = chroma_offset;
443*7688df22SAndroid Build Coastguard Worker memcpy((ib_cpu + len), uve_encode_param, sizeof(uve_encode_param));
444*7688df22SAndroid Build Coastguard Worker ib_cpu[len] = ALIGN(enc.width, align);
445*7688df22SAndroid Build Coastguard Worker ib_cpu[len + 1] = ALIGN(enc.width, align);
446*7688df22SAndroid Build Coastguard Worker len += sizeof(uve_encode_param) / 4;
447*7688df22SAndroid Build Coastguard Worker
448*7688df22SAndroid Build Coastguard Worker memcpy((ib_cpu + len), uve_op_speed_enc_mode, sizeof(uve_op_speed_enc_mode));
449*7688df22SAndroid Build Coastguard Worker len += sizeof(uve_op_speed_enc_mode) / 4;
450*7688df22SAndroid Build Coastguard Worker
451*7688df22SAndroid Build Coastguard Worker memcpy((ib_cpu + len), uve_op_encode, sizeof(uve_op_encode));
452*7688df22SAndroid Build Coastguard Worker len += sizeof(uve_op_encode) / 4;
453*7688df22SAndroid Build Coastguard Worker
454*7688df22SAndroid Build Coastguard Worker r = submit(len, AMDGPU_HW_IP_UVD_ENC);
455*7688df22SAndroid Build Coastguard Worker CU_ASSERT_EQUAL(r, 0);
456*7688df22SAndroid Build Coastguard Worker
457*7688df22SAndroid Build Coastguard Worker check_result(&enc);
458*7688df22SAndroid Build Coastguard Worker
459*7688df22SAndroid Build Coastguard Worker free_resource(&enc.fb);
460*7688df22SAndroid Build Coastguard Worker free_resource(&enc.bs);
461*7688df22SAndroid Build Coastguard Worker free_resource(&enc.vbuf);
462*7688df22SAndroid Build Coastguard Worker free_resource(&enc.cpb);
463*7688df22SAndroid Build Coastguard Worker }
464*7688df22SAndroid Build Coastguard Worker
amdgpu_cs_uvd_enc_destroy(void)465*7688df22SAndroid Build Coastguard Worker static void amdgpu_cs_uvd_enc_destroy(void)
466*7688df22SAndroid Build Coastguard Worker {
467*7688df22SAndroid Build Coastguard Worker int len, r;
468*7688df22SAndroid Build Coastguard Worker
469*7688df22SAndroid Build Coastguard Worker num_resources = 0;
470*7688df22SAndroid Build Coastguard Worker resources[num_resources++] = ib_handle;
471*7688df22SAndroid Build Coastguard Worker
472*7688df22SAndroid Build Coastguard Worker len = 0;
473*7688df22SAndroid Build Coastguard Worker memcpy((ib_cpu + len), uve_session_info, sizeof(uve_session_info));
474*7688df22SAndroid Build Coastguard Worker len += sizeof(uve_session_info) / 4;
475*7688df22SAndroid Build Coastguard Worker ib_cpu[len++] = enc.session.addr >> 32;
476*7688df22SAndroid Build Coastguard Worker ib_cpu[len++] = enc.session.addr;
477*7688df22SAndroid Build Coastguard Worker
478*7688df22SAndroid Build Coastguard Worker memcpy((ib_cpu + len), uve_task_info, sizeof(uve_task_info));
479*7688df22SAndroid Build Coastguard Worker len += sizeof(uve_task_info) / 4;
480*7688df22SAndroid Build Coastguard Worker ib_cpu[len++] = 0xffffffff;
481*7688df22SAndroid Build Coastguard Worker ib_cpu[len++] = 0x00000002;
482*7688df22SAndroid Build Coastguard Worker ib_cpu[len++] = 0x00000000;
483*7688df22SAndroid Build Coastguard Worker
484*7688df22SAndroid Build Coastguard Worker memcpy((ib_cpu + len), uve_op_close, sizeof(uve_op_close));
485*7688df22SAndroid Build Coastguard Worker len += sizeof(uve_op_close) / 4;
486*7688df22SAndroid Build Coastguard Worker
487*7688df22SAndroid Build Coastguard Worker r = submit(len, AMDGPU_HW_IP_UVD_ENC);
488*7688df22SAndroid Build Coastguard Worker CU_ASSERT_EQUAL(r, 0);
489*7688df22SAndroid Build Coastguard Worker
490*7688df22SAndroid Build Coastguard Worker free_resource(&enc.session);
491*7688df22SAndroid Build Coastguard Worker }
492