ai-hwaccel 1.1.0

Universal AI hardware accelerator detection, capability querying, and workload planning for Rust
Documentation 
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

//! AMD XDNA / Ryzen AI NPU detection via sysfs.

use std::path::Path;

use tracing::debug;

use crate::error::DetectionError;
use crate::hardware::AcceleratorType;
use crate::profile::AcceleratorProfile;

pub(crate) fn detect_amd_xdna(
    profiles: &mut Vec<AcceleratorProfile>,
    _warnings: &mut Vec<DetectionError>,
) {
    let accel_dir = Path::new("/sys/class/accel");
    if !accel_dir.exists() {
        return;
    }
    for entry in std::fs::read_dir(accel_dir).into_iter().flatten().flatten() {
        let driver_link = entry.path().join("device/driver");
        if let Ok(target) = std::fs::read_link(&driver_link) {
            let driver_name = target
                .file_name()
                .map(|n| n.to_string_lossy().to_string())
                .unwrap_or_default();
            if driver_name == "amdxdna" {
                let name = entry.file_name();
                let name_str = name.to_string_lossy();
                let device_id: u32 = name_str
                    .strip_prefix("accel")
                    .and_then(|s| s.parse().ok())
                    .unwrap_or(0);
                debug!(
                    device_id,
                    memory_mb = 2048,
                    "AMD XDNA NPU detected via sysfs"
                );
                profiles.push(AcceleratorProfile {
                    accelerator: AcceleratorType::AmdXdnaNpu { device_id },
                    available: true,
                    memory_bytes: 2 * 1024 * 1024 * 1024,
                    ..Default::default()
                });
            }
        }
    }
}