TensorSharp2/TensorSharp.TestMatrix/Matrix/Backend.cs at main · SciSharp/TensorSharp2 · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
// Copyright (c) Zhongkai Fu. All rights reserved.
// Licensed under the BSD-3-Clause license. See LICENSE in the repo root.

using System.Runtime.InteropServices;

namespace TensorSharp.TestMatrix.Matrix;

/// <summary>
/// One TensorSharp compute backend. Mirrors <c>TensorSharp.Server.BackendCatalog</c>
/// (cpu, ggml_cpu, ggml_cuda, ggml_metal, cuda, mlx). Kept as a flat record here so
/// the test runner stays decoupled from internal TensorSharp APIs.
/// </summary>
public sealed record BackendInfo(
    string Id,
    string DisplayName,
    bool RequiresCuda,
    bool RequiresMetal,
    bool RequiresAppleSilicon)
{
    public bool IsAvailableOnHost()
    {
        if (RequiresAppleSilicon && (!RuntimeInformation.IsOSPlatform(OSPlatform.OSX)
            || RuntimeInformation.OSArchitecture != Architecture.Arm64))
        {
            return false;
        }
        if (RequiresMetal && !RuntimeInformation.IsOSPlatform(OSPlatform.OSX))
        {
            return false;
        }
        if (RequiresCuda)
        {
            // Heuristic: assume CUDA is available on non-OSX hosts when not explicitly
            // disabled. The actual check happens when TensorSharp.Cli starts; if CUDA
            // is missing the runner records a 'backend unavailable' error.
            if (RuntimeInformation.IsOSPlatform(OSPlatform.OSX))
            {
                return false;
            }
        }
        return true;
    }
}

public static class BackendCatalog
{
    public static readonly BackendInfo Cpu = new(
        Id: "cpu",
        DisplayName: "CPU (Pure C#)",
        RequiresCuda: false,
        RequiresMetal: false,
        RequiresAppleSilicon: false);

    public static readonly BackendInfo GgmlCpu = new(
        Id: "ggml_cpu",
        DisplayName: "GGML CPU",
        RequiresCuda: false,
        RequiresMetal: false,
        RequiresAppleSilicon: false);

    public static readonly BackendInfo GgmlMetal = new(
        Id: "ggml_metal",
        DisplayName: "GGML Metal (GPU)",
        RequiresCuda: false,
        RequiresMetal: true,
        RequiresAppleSilicon: false);

    public static readonly BackendInfo GgmlCuda = new(
        Id: "ggml_cuda",
        DisplayName: "GGML CUDA (GPU)",
        RequiresCuda: true,
        RequiresMetal: false,
        RequiresAppleSilicon: false);

    public static readonly BackendInfo DirectCuda = new(
        Id: "cuda",
        DisplayName: "CUDA (cuBLAS GPU)",
        RequiresCuda: true,
        RequiresMetal: false,
        RequiresAppleSilicon: false);

    public static readonly BackendInfo Mlx = new(
        Id: "mlx",
        DisplayName: "MLX Metal (GPU)",
        RequiresCuda: false,
        RequiresMetal: true,
        RequiresAppleSilicon: true);

    public static readonly IReadOnlyList<BackendInfo> All = new[]
    {
        Cpu, GgmlCpu, GgmlMetal, GgmlCuda, DirectCuda, Mlx,
    };

    public static BackendInfo? FindById(string id)
    {
        foreach (BackendInfo b in All)
        {
            if (string.Equals(b.Id, id, StringComparison.OrdinalIgnoreCase))
            {
                return b;
            }
        }
        return null;
    }
}