test/amdgpu: add proper error handling v2
Otherwise the calling function won't notice that something is wrong. v2: check map result as well Signed-off-by: Christian König <christian.koenig@amd.com> Reviewed-by: Alex Deucher <alexander.deucher@amd.com>main
parent
c40bd3cbf8
commit
864d4501b9
|
@ -219,17 +219,34 @@ static inline amdgpu_bo_handle gpu_mem_alloc(
|
||||||
|
|
||||||
r = amdgpu_bo_alloc(device_handle, &req, &buf_handle);
|
r = amdgpu_bo_alloc(device_handle, &req, &buf_handle);
|
||||||
CU_ASSERT_EQUAL(r, 0);
|
CU_ASSERT_EQUAL(r, 0);
|
||||||
|
if (r)
|
||||||
|
return NULL;
|
||||||
|
|
||||||
r = amdgpu_va_range_alloc(device_handle,
|
r = amdgpu_va_range_alloc(device_handle,
|
||||||
amdgpu_gpu_va_range_general,
|
amdgpu_gpu_va_range_general,
|
||||||
size, alignment, 0, vmc_addr,
|
size, alignment, 0, vmc_addr,
|
||||||
va_handle, 0);
|
va_handle, 0);
|
||||||
CU_ASSERT_EQUAL(r, 0);
|
CU_ASSERT_EQUAL(r, 0);
|
||||||
|
if (r)
|
||||||
|
goto error_free_bo;
|
||||||
|
|
||||||
r = amdgpu_bo_va_op(buf_handle, 0, size, *vmc_addr, 0, AMDGPU_VA_OP_MAP);
|
r = amdgpu_bo_va_op(buf_handle, 0, size, *vmc_addr, 0,
|
||||||
|
AMDGPU_VA_OP_MAP);
|
||||||
CU_ASSERT_EQUAL(r, 0);
|
CU_ASSERT_EQUAL(r, 0);
|
||||||
|
if (r)
|
||||||
|
goto error_free_va;
|
||||||
|
|
||||||
return buf_handle;
|
return buf_handle;
|
||||||
|
|
||||||
|
error_free_va:
|
||||||
|
r = amdgpu_va_range_free(*va_handle);
|
||||||
|
CU_ASSERT_EQUAL(r, 0);
|
||||||
|
|
||||||
|
error_free_bo:
|
||||||
|
r = amdgpu_bo_free(buf_handle);
|
||||||
|
CU_ASSERT_EQUAL(r, 0);
|
||||||
|
|
||||||
|
return NULL;
|
||||||
}
|
}
|
||||||
|
|
||||||
static inline int gpu_mem_free(amdgpu_bo_handle bo,
|
static inline int gpu_mem_free(amdgpu_bo_handle bo,
|
||||||
|
@ -239,16 +256,23 @@ static inline int gpu_mem_free(amdgpu_bo_handle bo,
|
||||||
{
|
{
|
||||||
int r;
|
int r;
|
||||||
|
|
||||||
|
if (!bo)
|
||||||
|
return 0;
|
||||||
|
|
||||||
r = amdgpu_bo_va_op(bo, 0, size, vmc_addr, 0, AMDGPU_VA_OP_UNMAP);
|
r = amdgpu_bo_va_op(bo, 0, size, vmc_addr, 0, AMDGPU_VA_OP_UNMAP);
|
||||||
CU_ASSERT_EQUAL(r, 0);
|
CU_ASSERT_EQUAL(r, 0);
|
||||||
|
if (r)
|
||||||
|
return r;
|
||||||
|
|
||||||
r = amdgpu_va_range_free(va_handle);
|
r = amdgpu_va_range_free(va_handle);
|
||||||
CU_ASSERT_EQUAL(r, 0);
|
CU_ASSERT_EQUAL(r, 0);
|
||||||
|
if (r)
|
||||||
|
return r;
|
||||||
|
|
||||||
r = amdgpu_bo_free(bo);
|
r = amdgpu_bo_free(bo);
|
||||||
CU_ASSERT_EQUAL(r, 0);
|
CU_ASSERT_EQUAL(r, 0);
|
||||||
|
|
||||||
return 0;
|
return r;
|
||||||
}
|
}
|
||||||
|
|
||||||
static inline int
|
static inline int
|
||||||
|
|
Loading…
Reference in New Issue