Update cuda_malloc.py

This commit is contained in:
patientx 2024-08-30 12:19:50 +03:00 committed by GitHub
parent a8652a052f
commit 99789c6109
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -62,8 +62,8 @@ def cuda_malloc_supported():
for b in blacklist: for b in blacklist:
if b in x: if b in x:
return False return False
return True return False
#returned to older version. #We don't need malloc at all with amd gpu's. So disabling all together.
if not args.cuda_malloc: if not args.cuda_malloc: