highperfocused/ollama
1
0
Fork 0
mirror of https://github.com/ollama/ollama.git synced 2025-09-29 20:33:59 +02:00
Code Issues Packages Projects Releases Wiki Activity

Rename gpu package discover (#7143)

Browse Source 
Cleaning up go package naming
This commit is contained in:
Daniel Hiltgen
2024-10-16 17:45:00 -07:00
committed by GitHub
parent 7d6eb0d4c3
commit 05cd82ef94
33 changed files with 94 additions and 94 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
llm/memory_test.go
View File

@@ -10,7 +10,7 @@ import (
"github.com/stretchr/testify/require"
"github.com/ollama/ollama/api"
"github.com/ollama/ollama/gpu"
"github.com/ollama/ollama/discover"
)
func TestEstimateGPULayers(t *testing.T) {
@@ -50,7 +50,7 @@ func TestEstimateGPULayers(t *testing.T) {
}
// Simple CPU scenario
gpus := []gpu.GpuInfo{
gpus := []discover.GpuInfo{
{
Library: "cpu",
},
@@ -72,7 +72,7 @@ func TestEstimateGPULayers(t *testing.T) {
// Dual CUDA scenario with assymetry
gpuMinimumMemory := uint64(2048)
gpus = []gpu.GpuInfo{
gpus = []discover.GpuInfo{
{
Library: "cuda",
MinimumMemory: gpuMinimumMemory,