diff --git a/agent/gpu_nvml.go b/agent/gpu_nvml.go index 83bbe12e..5515a770 100644 --- a/agent/gpu_nvml.go +++ b/agent/gpu_nvml.go @@ -1,3 +1,5 @@ +//go:build (linux || windows) && (amd64 || arm64) + package agent import ( diff --git a/agent/gpu_nvml_linux.go b/agent/gpu_nvml_linux.go index a9bf4573..3307586c 100644 --- a/agent/gpu_nvml_linux.go +++ b/agent/gpu_nvml_linux.go @@ -1,14 +1,14 @@ -//go:build linux +//go:build linux && (amd64 || arm64) package agent import ( + "log/slog" "os" "path/filepath" "strings" "github.com/ebitengine/purego" - "log/slog" ) func openLibrary(name string) (uintptr, error) { diff --git a/agent/gpu_nvml_unsupported.go b/agent/gpu_nvml_unsupported.go index ac5502a3..96bf9135 100644 --- a/agent/gpu_nvml_unsupported.go +++ b/agent/gpu_nvml_unsupported.go @@ -1,9 +1,21 @@ -//go:build !linux && !windows +//go:build (!linux && !windows) || (!amd64 && !arm64) package agent import "fmt" +type nvmlCollector struct { + gm *GPUManager +} + +func (c *nvmlCollector) init() error { + return fmt.Errorf("nvml not supported on this platform") +} + +func (c *nvmlCollector) start() {} + +func (c *nvmlCollector) collect() {} + func openLibrary(name string) (uintptr, error) { return 0, fmt.Errorf("nvml not supported on this platform") } diff --git a/agent/gpu_nvml_windows.go b/agent/gpu_nvml_windows.go index 67318ec9..5405ae20 100644 --- a/agent/gpu_nvml_windows.go +++ b/agent/gpu_nvml_windows.go @@ -1,4 +1,4 @@ -//go:build windows +//go:build windows && (amd64 || arm64) package agent