1 files changed, 5 insertions, 3 deletions

diff --git a/drivers/gpu/nvgpu/gk20a/pmu_gk20a.h b/drivers/gpu/nvgpu/gk20a/pmu_gk20a.h
index 85403767..0682e4b8 100644
--- a/drivers/gpu/nvgpu/gk20a/pmu_gk20a.h
+++ b/drivers/gpu/nvgpu/gk20a/pmu_gk20a.h
@@ -680,9 +680,11 @@ struct pmu_pg_cmd_eng_buf_load_v1 {
         u8 engine_id;
         u8 buf_idx;
         u8 pad;
-        u16 buf_size;
-        struct falc_dma_addr dma_addr;  /* 256-byte block address */
-        u8 dma_idx;
+        struct flcn_mem_desc {
+                struct falc_u64 dma_addr;
+                u16 dma_size;
+                u8 dma_idx;
+        } dma_desc;
 };
 
 enum {