2 files changed, 4 insertions, 0 deletions

diff --git a/drivers/gpu/nvgpu/gk20a/gk20a_allocator_bitmap.c b/drivers/gpu/nvgpu/gk20a/gk20a_allocator_bitmap.c
index 29f52081..2291a953 100644
--- a/drivers/gpu/nvgpu/gk20a/gk20a_allocator_bitmap.c
+++ b/drivers/gpu/nvgpu/gk20a/gk20a_allocator_bitmap.c
@@ -42,6 +42,7 @@ static int gk20a_bitmap_alloc_inited(struct gk20a_allocator *a)
 {
         struct gk20a_bitmap_allocator *ba = a->priv;
 
+        rmb();
         return ba->inited;
 }
 
@@ -405,6 +406,7 @@ int gk20a_bitmap_allocator_init(struct gk20a_allocator *__a,
         if (!a->bitmap)
                 goto fail;
 
+        wmb();
         a->inited = true;
 
         gk20a_init_alloc_debug(__a);
diff --git a/drivers/gpu/nvgpu/gk20a/gk20a_allocator_buddy.c b/drivers/gpu/nvgpu/gk20a/gk20a_allocator_buddy.c
index 70a10c24..ee5e69d2 100644
--- a/drivers/gpu/nvgpu/gk20a/gk20a_allocator_buddy.c
+++ b/drivers/gpu/nvgpu/gk20a/gk20a_allocator_buddy.c
@@ -1078,6 +1078,7 @@ static int gk20a_buddy_alloc_inited(struct gk20a_allocator *a)
 {
         struct gk20a_buddy_allocator *ba = a->priv;
 
+        rmb();
         return ba->initialized;
 }
 
@@ -1284,6 +1285,7 @@ int __gk20a_buddy_allocator_init(struct gk20a_allocator *__a,
         if (err)
                 goto fail;
 
+        wmb();
         a->initialized = 1;
 
         gk20a_init_alloc_debug(__a);