mirror of
https://github.com/ollama/ollama.git
synced 2026-04-26 10:45:57 +02:00
Fix Unit Test (Add Vulkan Library)
This commit is contained in:
@@ -11,7 +11,7 @@ import (
|
||||
func TestBasicGetGPUInfo(t *testing.T) {
|
||||
info := GetGPUInfo()
|
||||
assert.NotEmpty(t, len(info))
|
||||
assert.Contains(t, "cuda rocm cpu metal", info[0].Library)
|
||||
assert.Contains(t, "cuda rocm cpu metal vulkan", info[0].Library)
|
||||
if info[0].Library != "cpu" {
|
||||
assert.Greater(t, info[0].TotalMemory, uint64(0))
|
||||
assert.Greater(t, info[0].FreeMemory, uint64(0))
|
||||
|
||||
Reference in New Issue
Block a user