return ret;
 }
 
+static u32
+r535_gsp_sr_data_size(struct nvkm_gsp *gsp)
+{
+       GspFwWprMeta *meta = gsp->wpr_meta.data;
+
+       return meta->gspFwWprEnd - meta->gspFwWprStart;
+}
+
 int
 r535_gsp_fini(struct nvkm_gsp *gsp, bool suspend)
 {
        int ret;
 
        if (suspend) {
-               GspFwWprMeta *meta = gsp->wpr_meta.data;
-               u64 len = meta->gspFwWprEnd - meta->gspFwWprStart;
+               u32 len = rm->api->gsp->sr_data_size(gsp);
                GspFwSRMeta *sr;
 
                ret = nvkm_gsp_sg(gsp->subdev.device, len, &gsp->sr.sgt);
        .get_static_info = r535_gsp_get_static_info,
        .xlat_mc_engine_idx = r535_gsp_xlat_mc_engine_idx,
        .drop_send_user_shared_data = r535_gsp_drop_send_user_shared_data,
+       .sr_data_size = r535_gsp_sr_data_size,
 };
 
                int (*get_static_info)(struct nvkm_gsp *);
                bool (*xlat_mc_engine_idx)(u32 mc_engine_idx, enum nvkm_subdev_type *, int *inst);
                void (*drop_send_user_shared_data)(struct nvkm_gsp *);
+               u32 (*sr_data_size)(struct nvkm_gsp *);
        } *gsp;
 
        const struct nvkm_rm_api_rpc {