hanzo-rocm-kernels 0.10.2

ROCm/HIP kernels for Hanzo
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

use thiserror::Error;

#[derive(Debug, Error)]
pub enum KernelError {
    #[error("ROCm error: {0}")]
    Rocm(String),
    #[error("IO error: {0}")]
    Io(String),
    #[error("Kernel compilation failed: {0}")]
    Compilation(String),
    #[error("Kernel launch failed: {0}")]
    Launch(String),
    #[error("Internal error: {0}")]
    Internal(String),
    #[error("Unsupported dtype: {0}")]
    UnsupportedDType(String),
    #[error("Unsupported operation: {0}")]
    UnsupportedOperation(String),
}

impl From<rocm_rs::hip::Error> for KernelError {
    fn from(e: rocm_rs::hip::Error) -> Self {
        KernelError::Rocm(e.to_string())
    }
}