From 1f11b525111ea89bf325ad6fda3e3a13d6396a50 Mon Sep 17 00:00:00 2001 From: Daniel Hiltgen Date: Mon, 1 Apr 2024 16:47:33 -0700 Subject: [PATCH] Refined min memory from testing --- gpu/gpu.go | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/gpu/gpu.go b/gpu/gpu.go index cf2f3b7f..708fad10 100644 --- a/gpu/gpu.go +++ b/gpu/gpu.go @@ -30,8 +30,8 @@ type handles struct { } const ( - cudaMinimumMemory = 377 * format.MebiByte - rocmMinimumMemory = 377 * format.MebiByte + cudaMinimumMemory = 457 * format.MebiByte + rocmMinimumMemory = 457 * format.MebiByte ) var gpuMutex sync.Mutex