nvgpu memtest: put memory ref count

after check failed, need release memory reference

Bug 200531152

Change-Id: I6bfad867eb9bd636a48b5ceb3a4417a80994a3ec
Signed-off-by: Bruce Xu <brucex@nvidia.com>
Reviewed-on: https://git-master.nvidia.com/r/2194025
(cherry picked from commit f85504ae46d65d5346d9e2a5cc84ffb960ba9fb7)
Reviewed-on: https://git-master.nvidia.com/r/2195437
Reviewed-by: Bibek Basu <bbasu@nvidia.com>
Tested-by: Debarshi Dutta <ddutta@nvidia.com>
Reviewed-by: Debarshi Dutta <ddutta@nvidia.com>
Reviewed-by: mobile promotions <svcmobile_promotions@nvidia.com>
Tested-by: mobile promotions <svcmobile_promotions@nvidia.com>

1 files changed, 2 insertions, 0 deletions

diff --git a/drivers/gpu/nvgpu/os/linux/vm.c b/drivers/gpu/nvgpu/os/linux/vm.c
index 97dd0671..27a91cf6 100644
--- a/drivers/gpu/nvgpu/os/linux/vm.c
+++ b/drivers/gpu/nvgpu/os/linux/vm.c
@@ -281,6 +281,7 @@ int nvgpu_vm_map_buffer(struct vm_gk20a *vm,
             (buffer_offset || *map_addr)) {
                 nvgpu_err(g,
                           "Regular map with addr/buf offset is not supported!");
+                dma_buf_put(dmabuf);
                 return -EINVAL;
         }
 
@@ -290,6 +291,7 @@ int nvgpu_vm_map_buffer(struct vm_gk20a *vm,
          */
         if (mapping_size && !(flags & NVGPU_AS_MAP_BUFFER_FLAGS_FIXED_OFFSET)) {
                 nvgpu_err(g, "map_size && non-fixed-mapping!");
+                dma_buf_put(dmabuf);
                 return -EINVAL;
         }