void amdgpu_ttm_late_init(struct amdgpu_device *adev)
 {
        void *stolen_vga_buf;
+
        /* return the VGA stolen memory (if any) back to VRAM */
-       amdgpu_bo_free_kernel(&adev->gmc.stolen_vga_memory, NULL, &stolen_vga_buf);
+       if (!adev->gmc.keep_stolen_vga_memory)
+               amdgpu_bo_free_kernel(&adev->gmc.stolen_vga_memory, NULL, &stolen_vga_buf);
 }
 
 /**
  */
 void amdgpu_ttm_fini(struct amdgpu_device *adev)
 {
+       void *stolen_vga_buf;
+
        if (!adev->mman.initialized)
                return;
 
        amdgpu_ttm_training_reserve_vram_fini(adev);
+       /* return the stolen vga memory back to VRAM */
+       if (adev->gmc.keep_stolen_vga_memory)
+               amdgpu_bo_free_kernel(&adev->gmc.stolen_vga_memory, NULL, &stolen_vga_buf);
        /* return the IP Discovery TMR memory back to VRAM */
        amdgpu_bo_free_kernel(&adev->discovery_memory, NULL, NULL);
        amdgpu_ttm_fw_reserve_vram_fini(adev);
 
        struct amdgpu_device *adev = (struct amdgpu_device *)handle;
        int r;
 
-       if (!gmc_v9_0_keep_stolen_memory(adev))
-               amdgpu_bo_late_init(adev);
+       amdgpu_bo_late_init(adev);
 
        r = amdgpu_gmc_allocate_vm_inv_eng(adev);
        if (r)
         * TODO Remove once GART corruption is resolved
         * Check related code in gmc_v9_0_sw_fini
         * */
-       if (gmc_v9_0_keep_stolen_memory(adev))
+       if (adev->gmc.keep_stolen_vga_memory)
                return AMDGPU_VBIOS_VGA_ALLOCATION;
 
        d1vga_control = RREG32_SOC15(DCE, 0, mmD1VGA_CONTROL);
        if (r)
                return r;
 
+       adev->gmc.keep_stolen_vga_memory = gmc_v9_0_keep_stolen_memory(adev);
        adev->gmc.stolen_vga_size = gmc_v9_0_get_vbios_fb_size(adev);
 
        /* Memory manager */
 static int gmc_v9_0_sw_fini(void *handle)
 {
        struct amdgpu_device *adev = (struct amdgpu_device *)handle;
-       void *stolen_vga_buf;
 
        amdgpu_gmc_ras_fini(adev);
        amdgpu_gem_force_release(adev);
        amdgpu_vm_manager_fini(adev);
-
-       if (gmc_v9_0_keep_stolen_memory(adev))
-               amdgpu_bo_free_kernel(&adev->gmc.stolen_vga_memory, NULL, &stolen_vga_buf);
-
        amdgpu_gart_table_vram_free(adev);
        amdgpu_bo_fini(adev);
        amdgpu_gart_fini(adev);