Commit 1f11b525 authored by Daniel Hiltgen's avatar Daniel Hiltgen
Browse files

Refined min memory from testing

parent 526d4eb2
...@@ -30,8 +30,8 @@ type handles struct { ...@@ -30,8 +30,8 @@ type handles struct {
} }
const ( const (
cudaMinimumMemory = 377 * format.MebiByte cudaMinimumMemory = 457 * format.MebiByte
rocmMinimumMemory = 377 * format.MebiByte rocmMinimumMemory = 457 * format.MebiByte
) )
var gpuMutex sync.Mutex var gpuMutex sync.Mutex
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment