scirs2-linalg 0.4.0

Linear algebra module for SciRS2 (scirs2-linalg)
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
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

//! ROCm backend implementation for AMD GPU acceleration
//!
//! This module provides a ROCm backend for GPU acceleration on AMD devices.
//! ROCm (Radeon Open Compute) is AMD's open-source GPU computing platform.

use super::common::*;

/// Placeholder ROCm backend (requires ROCm feature and runtime)
#[cfg(feature = "rocm")]
pub mod rocm_impl {
    use super::*;

    pub struct RocmBackend {
        #[allow(dead_code)]
        devices: Vec<String>,
    }

    impl RocmBackend {
        pub fn new() -> LinalgResult<Self> {
            // In a real implementation, this would initialize ROCm/HIP
            Ok(Self {
                devices: Vec::new(),
            })
        }
    }

    impl GpuBackend for RocmBackend {
        fn name(&self) -> &str {
            "ROCm"
        }

        fn is_available(&self) -> bool {
            // In a real implementation, check ROCm availability
            false
        }

        fn list_devices(&self) -> LinalgResult<Vec<GpuDeviceInfo>> {
            // In a real implementation, enumerate ROCm devices
            Ok(vec![])
        }

        fn create_context(&self, _device_id: usize) -> LinalgResult<Box<dyn GpuContext>> {
            Err(LinalgError::ComputationError(
                "ROCm backend not fully implemented".to_string(),
            ))
        }
    }
}

// Re-export the ROCm backend when the feature is enabled
#[cfg(feature = "rocm")]
pub use rocm_impl::*;

// Provide a stub when ROCm is not available
#[cfg(not(feature = "rocm"))]
pub struct RocmBackend;

#[cfg(not(feature = "rocm"))]
impl RocmBackend {
    pub fn new() -> LinalgResult<Self> {
        Err(LinalgError::ComputationError(
            "ROCm support not compiled in".to_string(),
        ))
    }
}

#[cfg(not(feature = "rocm"))]
impl GpuBackend for RocmBackend {
    fn name(&self) -> &str {
        "ROCm (not available)"
    }

    fn is_available(&self) -> bool {
        false
    }

    fn list_devices(&self) -> LinalgResult<Vec<GpuDeviceInfo>> {
        Ok(vec![])
    }

    fn create_context(&self, _device_id: usize) -> LinalgResult<Box<dyn GpuContext>> {
        Err(LinalgError::ComputationError(
            "ROCm support not compiled in".to_string(),
        ))
    }
}