[PATCH v3] gpudev: manage NULL pointer
eagostini at nvidia.com
eagostini at nvidia.com
Mon Nov 22 19:24:23 CET 2021
From: Elena Agostini <eagostini at nvidia.com>
gpudev free and unregister functions return gracefully if input pointer is NULL
Fixes: 818a067baf90 ("gpudev: manage NULL pointer")
Signed-off-by: Elena Agostini <eagostini at nvidia.com>
---
drivers/gpu/cuda/cuda.c | 6 ------
lib/gpudev/gpudev.c | 6 ++++++
2 files changed, 6 insertions(+), 6 deletions(-)
diff --git a/drivers/gpu/cuda/cuda.c b/drivers/gpu/cuda/cuda.c
index 24ae630d04..47a3bbf256 100644
--- a/drivers/gpu/cuda/cuda.c
+++ b/drivers/gpu/cuda/cuda.c
@@ -764,9 +764,6 @@ cuda_mem_free(struct rte_gpu *dev, void *ptr)
if (dev == NULL)
return -ENODEV;
- if (ptr == NULL)
- return -EINVAL;
-
hk = get_hash_from_ptr((void *)ptr);
mem_item = mem_list_find_item(hk);
@@ -803,9 +800,6 @@ cuda_mem_unregister(struct rte_gpu *dev, void *ptr)
if (dev == NULL)
return -ENODEV;
- if (ptr == NULL)
- return -EINVAL;
-
hk = get_hash_from_ptr((void *)ptr);
mem_item = mem_list_find_item(hk);
diff --git a/lib/gpudev/gpudev.c b/lib/gpudev/gpudev.c
index 2b174d8bd5..b41c43016a 100644
--- a/lib/gpudev/gpudev.c
+++ b/lib/gpudev/gpudev.c
@@ -569,6 +569,9 @@ rte_gpu_mem_free(int16_t dev_id, void *ptr)
{
struct rte_gpu *dev;
+ if (ptr == NULL)
+ return 0;
+
dev = gpu_get_by_id(dev_id);
if (dev == NULL) {
GPU_LOG(ERR, "free mem for invalid device ID %d", dev_id);
@@ -612,6 +615,9 @@ rte_gpu_mem_unregister(int16_t dev_id, void *ptr)
{
struct rte_gpu *dev;
+ if (ptr == NULL)
+ return 0;
+
dev = gpu_get_by_id(dev_id);
if (dev == NULL) {
GPU_LOG(ERR, "unregister mem for invalid device ID %d", dev_id);
--
2.17.1
More information about the dev
mailing list