From 8210e20939cc941682b926bd69f64916091146d9 Mon Sep 17 00:00:00 2001
From: IonutParau <ionut.alex.parau@gmail.com>
Date: Sat, 28 Jun 2025 18:03:18 +0200
Subject: [PATCH] small fix to bad ref counting

---
 src/components/gpu.c | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/src/components/gpu.c b/src/components/gpu.c
index 7a9cd57..bc1b6b7 100644
--- a/src/components/gpu.c
+++ b/src/components/gpu.c
@@ -90,10 +90,11 @@ void nni_gpu_bind(nni_gpu *gpu, void *_, nn_component *component, nn_computer *c
         return;
     }
 
-    if(gpu->currentScreen != NULL) {
-        nn_destroyScreen(gpu->currentScreen);
-    }
     nn_screen *screen = nn_getComponentUserdata(c);
+    nn_retainScreen(screen);
+    if(gpu->currentScreen != NULL) nn_destroyScreen(gpu->currentScreen);
+    gpu->currentScreen = screen;
+    
 
     if(reset) {
         for(size_t i = 0; i < screen->width; i++) {