diff options
Diffstat (limited to 'drivers/gpu/nvgpu/common/linux')
-rw-r--r-- | drivers/gpu/nvgpu/common/linux/vm.c | 8 |
1 files changed, 7 insertions, 1 deletions
diff --git a/drivers/gpu/nvgpu/common/linux/vm.c b/drivers/gpu/nvgpu/common/linux/vm.c index 4fa01855..f356fee2 100644 --- a/drivers/gpu/nvgpu/common/linux/vm.c +++ b/drivers/gpu/nvgpu/common/linux/vm.c | |||
@@ -25,6 +25,7 @@ | |||
25 | 25 | ||
26 | #include "gk20a/gk20a.h" | 26 | #include "gk20a/gk20a.h" |
27 | #include "gk20a/mm_gk20a.h" | 27 | #include "gk20a/mm_gk20a.h" |
28 | #include "gk20a/kind_gk20a.h" | ||
28 | 29 | ||
29 | #include "vm_priv.h" | 30 | #include "vm_priv.h" |
30 | 31 | ||
@@ -237,7 +238,12 @@ u64 nvgpu_vm_map(struct vm_gk20a *vm, | |||
237 | goto clean_up; | 238 | goto clean_up; |
238 | } | 239 | } |
239 | 240 | ||
240 | bfr.kind_v = kind; | 241 | if (kind >= NV_KIND_ATTR_SIZE) { |
242 | err = -EINVAL; | ||
243 | goto clean_up; | ||
244 | } else { | ||
245 | bfr.kind_v = kind; | ||
246 | } | ||
241 | bfr.size = dmabuf->size; | 247 | bfr.size = dmabuf->size; |
242 | sgl = bfr.sgt->sgl; | 248 | sgl = bfr.sgt->sgl; |
243 | 249 | ||