* PGRAPH engine/subdev functions
  ******************************************************************************/
 
+static int
+gf100_gr_fecs_discover_zcull_image_size(struct gf100_gr *gr, u32 *psize)
+{
+       struct nvkm_device *device = gr->base.engine.subdev.device;
+
+       nvkm_wr32(device, 0x409840, 0xffffffff);
+       nvkm_wr32(device, 0x409500, 0x00000000);
+       nvkm_wr32(device, 0x409504, 0x00000016);
+       nvkm_msec(device, 2000,
+               if ((*psize = nvkm_rd32(device, 0x409800)))
+                       return 0;
+       );
+
+       return -ETIMEDOUT;
+}
+
 static int
 gf100_gr_fecs_discover_image_size(struct gf100_gr *gr, u32 *psize)
 {
        if (ret)
                return ret;
 
-       nvkm_wr32(device, 0x409840, 0xffffffff);
-       nvkm_wr32(device, 0x409500, 0x00000000);
-       nvkm_wr32(device, 0x409504, 0x00000016);
-       if (nvkm_msec(device, 2000,
-               if (nvkm_rd32(device, 0x409800))
-                       break;
-       ) < 0)
-               return -EBUSY;
+       /* Determine how much memory is required to store ZCULL image. */
+       ret = gf100_gr_fecs_discover_zcull_image_size(gr, &gr->size_zcull);
+       if (ret)
+               return ret;
 
        nvkm_wr32(device, 0x409840, 0xffffffff);
        nvkm_wr32(device, 0x409500, 0x00000000);