From d7f28efe8dc023780b0be43357bb3f58d8627768 Mon Sep 17 00:00:00 2001 From: patientx Date: Mon, 26 Aug 2024 14:42:12 +0300 Subject: [PATCH] Update cuda_malloc.py --- cuda_malloc.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/cuda_malloc.py b/cuda_malloc.py index 11bf7c039..595af2ffe 100644 --- a/cuda_malloc.py +++ b/cuda_malloc.py @@ -56,12 +56,14 @@ def cuda_malloc_supported(): except: names = set() for x in names: - if "NVIDIA" in x: + if "AMD" in x: + return False + elif "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 +#returned to older version. if not args.cuda_malloc: