3 files changed, 9 insertions, 4 deletions

diff --git a/drivers/gpu/drm/amd/amdkfd/kfd_kernel_queue.c b/drivers/gpu/drm/amd/amdkfd/kfd_kernel_queue.c
index 9abac48de499..935071410724 100644
--- a/drivers/gpu/drm/amd/amdkfd/kfd_kernel_queue.c
+++ b/drivers/gpu/drm/amd/amdkfd/kfd_kernel_queue.c
@@ -221,8 +221,14 @@ static int acquire_packet_buffer(struct kernel_queue *kq,
                                                         queue_size_dwords;
 
         if (packet_size_in_dwords >= queue_size_dwords ||
-                        packet_size_in_dwords >= available_size)
+                        packet_size_in_dwords >= available_size) {
+                /*
+                 * make sure calling functions know
+                 * acquire_packet_buffer() failed
+                 */
+                *buffer_ptr = NULL;
                 return -ENOMEM;
+        }
 
         if (wptr + packet_size_in_dwords >= queue_size_dwords) {
                 while (wptr > 0) {
diff --git a/drivers/gpu/drm/amd/amdkfd/kfd_pasid.c b/drivers/gpu/drm/amd/amdkfd/kfd_pasid.c
index 2458ab7c0c6e..71699ad97d74 100644
--- a/drivers/gpu/drm/amd/amdkfd/kfd_pasid.c
+++ b/drivers/gpu/drm/amd/amdkfd/kfd_pasid.c
@@ -32,8 +32,7 @@ int kfd_pasid_init(void)
 {
         pasid_limit = max_num_of_processes;
 
-        pasid_bitmap = kzalloc(DIV_ROUND_UP(pasid_limit, BITS_PER_BYTE),
-                                GFP_KERNEL);
+        pasid_bitmap = kzalloc(BITS_TO_LONGS(pasid_limit), GFP_KERNEL);
         if (!pasid_bitmap)
                 return -ENOMEM;
 
diff --git a/drivers/gpu/drm/amd/amdkfd/kfd_process.c b/drivers/gpu/drm/amd/amdkfd/kfd_process.c
index b4f49ac13334..b85eb0b830b4 100644
--- a/drivers/gpu/drm/amd/amdkfd/kfd_process.c
+++ b/drivers/gpu/drm/amd/amdkfd/kfd_process.c
@@ -196,7 +196,7 @@ static void kfd_process_destroy_delayed(struct rcu_head *rcu)
         mmdrop(p->mm);
 
         work = (struct kfd_process_release_work *)
-                kmalloc(sizeof(struct kfd_process_release_work), GFP_KERNEL);
+                kmalloc(sizeof(struct kfd_process_release_work), GFP_ATOMIC);
 
         if (work) {
                 INIT_WORK((struct work_struct *) work, kfd_process_wq_release);