1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22

use crate::ComplexField;
use num_complex::Complex32 as c32;
use num_complex::Complex64 as c64;

pub trait Taxpy: ComplexField{
    fn axpy(n: i32, alpha: Self, x: &[Self], incx: i32, y: &mut [Self], incy: i32);
}

macro_rules! impl_taxpy(
    ($N: ty, $taxpy: path) => (
        impl Taxpy for $N{
            fn axpy(n: i32, alpha: Self, x: &[Self], incx: i32, y: &mut [Self], incy: i32) {
                unsafe{ $taxpy(n, alpha, x, incx, y, incy) }
            }
        }
    )
);

impl_taxpy!(f32, cblas::saxpy);
impl_taxpy!(f64, cblas::daxpy);
impl_taxpy!(c32, cblas::caxpy);
impl_taxpy!(c64, cblas::zaxpy);