From 6bc6534a9fb591f976e7c5d252ebce2a995a2db4 Mon Sep 17 00:00:00 2001
From: James Benton <james.benton2@gmail.com>
Date: Fri, 2 Jun 2017 14:37:13 +0100
Subject: [PATCH] whb: Fix GPU hang reset.

---
 src/libwhb/src/gfx.c | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

diff --git a/src/libwhb/src/gfx.c b/src/libwhb/src/gfx.c
index d1159da0..a0c20209 100644
--- a/src/libwhb/src/gfx.c
+++ b/src/libwhb/src/gfx.c
@@ -435,6 +435,7 @@ WHBGfxShutdown()
    }
 
    GfxProcCallbackReleased(NULL);
+   GX2RSetAllocator(NULL, NULL);
    GX2Shutdown();
 
    if (sTvContextState) {
@@ -451,8 +452,6 @@ WHBGfxShutdown()
       GfxHeapFreeMEM2(sCommandBufferPool);
       sCommandBufferPool = NULL;
    }
-
-   GX2RSetAllocator(NULL, NULL);
 }
 
 void
@@ -466,7 +465,6 @@ WHBGfxBeginRender()
       GX2GetSwapStatus(&swapCount, &flipCount, &lastFlip, &lastVsync);
 
       if (flipCount >= swapCount) {
-         sGpuTimedOut = FALSE;
          break;
       }