diff --git a/pyverbs/dmabuf_alloc.c b/pyverbs/dmabuf_alloc.c
index e3ea0a44f..68f25928a 100644
--- a/pyverbs/dmabuf_alloc.c
+++ b/pyverbs/dmabuf_alloc.c
@@ -43,7 +43,7 @@ static int i915_alloc(struct drm *drm, uint64_t size, uint32_t *handle, int gtt)
 	return 0;
 }
 
-static int amdgpu_alloc(struct drm *drm, size_t size, uint32_t *handle, int gtt)
+static int amdgpu_alloc(struct drm *drm, uint64_t size, uint32_t *handle, int gtt)
 {
 	union drm_amdgpu_gem_create gem_create = {{}};
 	int err;