[CRIU] [RFC PATCH 14/20] criu/plugin: Re-adjust doorbell offset for queues

[Felix Kuehling]

From: David Yat Sin <david.yatsin at amd.com>

The doorbell offset depends on the gpu_id. When restoring on a different
node, the gpu_id may be different, so we need to re-adjust the doorbell
offsets.

Signed-off-by: David Yat Sin <david.yatsin at amd.com>
---
 test/others/ext-kfd/kfd_plugin.c | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/test/others/ext-kfd/kfd_plugin.c b/test/others/ext-kfd/kfd_plugin.c
index 78c8c4f3b..86f77f9ff 100644
--- a/test/others/ext-kfd/kfd_plugin.c
+++ b/test/others/ext-kfd/kfd_plugin.c
@@ -1013,7 +1013,8 @@ int kfd_plugin_restore_file(int id)
 		if (e->bo_info_test[i]->bo_alloc_flags &
 			(KFD_IOC_ALLOC_MEM_FLAGS_VRAM |
 			 KFD_IOC_ALLOC_MEM_FLAGS_GTT |
-			 KFD_IOC_ALLOC_MEM_FLAGS_MMIO_REMAP)) {
+			 KFD_IOC_ALLOC_MEM_FLAGS_MMIO_REMAP |
+			 KFD_IOC_ALLOC_MEM_FLAGS_DOORBELL)) {
 
 			struct vma_metadata *vma_md;
 			vma_md = xmalloc(sizeof(*vma_md));
-- 
2.17.1