1 files changed, 9 insertions, 4 deletions

diff --git a/drivers/gpu/drm/amd/amdkfd/kfd_crat.c b/drivers/gpu/drm/amd/amdkfd/kfd_crat.c
index 3de5e14c5ae3..d7f67620f57b 100644
--- a/drivers/gpu/drm/amd/amdkfd/kfd_crat.c
+++ b/drivers/gpu/drm/amd/amdkfd/kfd_crat.c
@@ -774,6 +774,7 @@ int kfd_create_crat_image_acpi(void **crat_image, size_t *size)
         struct acpi_table_header *crat_table;
         acpi_status status;
         void *pcrat_image;
+        int rc = 0;
 
         if (!crat_image)
                 return -EINVAL;
@@ -798,14 +799,17 @@ int kfd_create_crat_image_acpi(void **crat_image, size_t *size)
         }
 
         pcrat_image = kvmalloc(crat_table->length, GFP_KERNEL);
-        if (!pcrat_image)
-                return -ENOMEM;
+        if (!pcrat_image) {
+                rc = -ENOMEM;
+                goto out;
+        }
 
         memcpy(pcrat_image, crat_table, crat_table->length);
         *crat_image = pcrat_image;
         *size = crat_table->length;
-
-        return 0;
+out:
+        acpi_put_table(crat_table);
+        return rc;
 }
 
 /* Memory required to create Virtual CRAT.
@@ -988,6 +992,7 @@ static int kfd_create_vcrat_image_cpu(void *pcrat_image, size_t *size)
                                 CRAT_OEMID_LENGTH);
                 memcpy(crat_table->oem_table_id, acpi_table->oem_table_id,
                                 CRAT_OEMTABLEID_LENGTH);
+                acpi_put_table(acpi_table);
         }
         crat_table->total_entries = 0;
         crat_table->num_domains = 0;