}
 
 static void
-r535_instmem_resume(struct nvkm_instmem *imem)
+r535_fbsr_resume(struct nvkm_gsp *gsp)
 {
        /* RM has restored VRAM contents already, so just need to free the sysmem buffer. */
-       if (imem->rm.fbsr_valid) {
-               nvkm_gsp_sg_free(imem->subdev.device, &imem->rm.fbsr);
-               imem->rm.fbsr_valid = false;
-       }
+       nvkm_gsp_sg_free(gsp->subdev.device, &gsp->sr.fbsr);
 }
 
 static int
-r535_instmem_suspend(struct nvkm_instmem *imem)
+r535_fbsr_suspend(struct nvkm_gsp *gsp)
 {
-       struct nvkm_subdev *subdev = &imem->subdev;
+       struct nvkm_subdev *subdev = &gsp->subdev;
        struct nvkm_device *device = subdev->device;
-       struct nvkm_gsp *gsp = device->gsp;
+       struct nvkm_instmem *imem = device->imem;
        struct nvkm_instobj *iobj;
        struct fbsr fbsr = {};
        struct fbsr_item *item, *temp;
        fbsr.size += gsp->fb.bios.vga_workspace.size;
        nvkm_debug(subdev, "fbsr: size: 0x%llx bytes\n", fbsr.size);
 
-       ret = nvkm_gsp_sg(gsp->subdev.device, fbsr.size, &imem->rm.fbsr);
+       ret = nvkm_gsp_sg(gsp->subdev.device, fbsr.size, &gsp->sr.fbsr);
        if (ret)
                goto done;
 
        if (ret)
                goto done_sgt;
 
-       ret = fbsr_init(&fbsr, &imem->rm.fbsr, items_size);
+       ret = fbsr_init(&fbsr, &gsp->sr.fbsr, items_size);
        if (WARN_ON(ret))
                goto done_sgt;
 
                        goto done_sgt;
        }
 
-       imem->rm.fbsr_valid = true;
-
        /* Cleanup everything except the sysmem backup, which will be removed after resume. */
 done_sgt:
        if (ret) /* ... unless we failed already. */
-               nvkm_gsp_sg_free(device, &imem->rm.fbsr);
+               nvkm_gsp_sg_free(device, &gsp->sr.fbsr);
 done:
        list_for_each_entry_safe(item, temp, &fbsr.items, head) {
                list_del(&item->head);
        return ret;
 }
 
+const struct nvkm_rm_api_fbsr
+r535_fbsr = {
+       .suspend = r535_fbsr_suspend,
+       .resume = r535_fbsr_resume,
+};
+
 static void *
 r535_instmem_dtor(struct nvkm_instmem *imem)
 {
 
        rm->dtor = r535_instmem_dtor;
        rm->fini = hw->fini;
-       rm->suspend = r535_instmem_suspend;
-       rm->resume  = r535_instmem_resume;
        rm->memory_new = hw->memory_new;
        rm->memory_wrap = hw->memory_wrap;
        rm->zero = false;
 
 int
 r535_gsp_fini(struct nvkm_gsp *gsp, bool suspend)
 {
+       struct nvkm_rm *rm = gsp->rm;
        int ret;
 
        if (suspend) {
                sr->revision = GSP_FW_SR_META_REVISION;
                sr->sysmemAddrOfSuspendResumeData = gsp->sr.radix3.lvl0.addr;
                sr->sizeOfSuspendResumeData = len;
+
+               ret = rm->api->fbsr->suspend(gsp);
+               if (ret) {
+                       nvkm_gsp_mem_dtor(&gsp->sr.meta);
+                       nvkm_gsp_radix3_dtor(gsp, &gsp->sr.radix3);
+                       nvkm_gsp_sg_free(gsp->subdev.device, &gsp->sr.sgt);
+                       return ret;
+               }
        }
 
        ret = r535_gsp_rpc_unloading_guest_driver(gsp, suspend);
 
 done:
        if (gsp->sr.meta.data) {
+               gsp->rm->api->fbsr->resume(gsp);
+
                nvkm_gsp_mem_dtor(&gsp->sr.meta);
                nvkm_gsp_radix3_dtor(gsp, &gsp->sr.radix3);
                nvkm_gsp_sg_free(gsp->subdev.device, &gsp->sr.sgt);
 
                } event;
        } *device;
 
+       const struct nvkm_rm_api_fbsr {
+               int (*suspend)(struct nvkm_gsp *);
+               void (*resume)(struct nvkm_gsp *);
+       } *fbsr;
+
        const struct nvkm_rm_api_disp {
                int (*get_static_info)(struct nvkm_disp *);
 
 extern const struct nvkm_rm_api_alloc r535_alloc;
 extern const struct nvkm_rm_api_client r535_client;
 extern const struct nvkm_rm_api_device r535_device;
+extern const struct nvkm_rm_api_fbsr r535_fbsr;
 extern const struct nvkm_rm_api_disp r535_disp;
 extern const struct nvkm_rm_api_fifo r535_fifo;
 extern const struct nvkm_rm_api_engine r535_ce;