1 files changed, 2 insertions, 3 deletions

diff --git a/drivers/gpu/drm/i915/i915_gem_dmabuf.c b/drivers/gpu/drm/i915/i915_gem_dmabuf.c
index 773ef77b6c22..defb888ef7f5 100644
--- a/drivers/gpu/drm/i915/i915_gem_dmabuf.c
+++ b/drivers/gpu/drm/i915/i915_gem_dmabuf.c
@@ -276,8 +276,7 @@ struct drm_gem_object *i915_gem_prime_import(struct drm_device *dev,
         if (IS_ERR(attach))
                 return ERR_CAST(attach);
 
-
-        obj = kzalloc(sizeof(*obj), GFP_KERNEL);
+        obj = i915_gem_object_alloc(dev);
         if (obj == NULL) {
                 ret = -ENOMEM;
                 goto fail_detach;
@@ -285,7 +284,7 @@ struct drm_gem_object *i915_gem_prime_import(struct drm_device *dev,
 
         ret = drm_gem_private_object_init(dev, &obj->base, dma_buf->size);
         if (ret) {
-                kfree(obj);
+                i915_gem_object_free(obj);
                 goto fail_detach;
         }