From 2edfcd61418624c1a5dc2dc46b93a79445d93932 Mon Sep 17 00:00:00 2001
From: patientx <patient.x@gmail.com>
Date: Mon, 22 Dec 2025 02:18:29 +0300
Subject: [PATCH] Update GPU detection logic in cuda_malloc.py

Refactor GPU check logic to return True for NVIDIA GPUs.
---
 cuda_malloc.py | 7 ++-----
 1 file changed, 2 insertions(+), 5 deletions(-)

diff --git a/cuda_malloc.py b/cuda_malloc.py
index 7450c12d3..ee2bc4b69 100644
--- a/cuda_malloc.py
+++ b/cuda_malloc.py
@@ -56,14 +56,11 @@ def cuda_malloc_supported():
     except:
         names = set()
     for x in names:
-        if "AMD" in x:
-            return False
-        elif "NVIDIA" in x:
+        if "NVIDIA" in x:
             for b in blacklist:
                 if b in x:
                     return False
-    return False
-#We don't need malloc at all with amd gpu's. So disabling all together.
+    return True
 
 
 version = ""