tblis 0.2.5

TBLIS wrapper in 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

//! TBLIS trait for floating point types.

// use crate::prelude::*;
use duplicate::duplicate_item;
use num::complex::ComplexFloat;
use num::Complex;

/// Float trait for TBLIS operations (f32, f64, c32, c64).
///
/// Note that this trait only inplements for [`f32`], [`f64`], [`Complex<f32>`], [`Complex<f64>`].
pub trait TblisFloatAPI: ComplexFloat {
    /// TBLIS constant indicating type in FFI.
    const TYPE: tblis_ffi::tblis::type_t;
    /// Convert to scalar type of TBLIS FFI.
    fn to_ffi_scalar(&self) -> tblis_ffi::tblis::tblis_scalar;
}

#[duplicate_item(
     T              FLOAT_TYPE                        FLOAT_FIELD ;
    [f32         ] [tblis_ffi::tblis::TYPE_FLOAT   ] [s          ];
    [f64         ] [tblis_ffi::tblis::TYPE_DOUBLE  ] [d          ];
    [Complex<f32>] [tblis_ffi::tblis::TYPE_SCOMPLEX] [c          ];
    [Complex<f64>] [tblis_ffi::tblis::TYPE_DCOMPLEX] [z          ];
)]
impl TblisFloatAPI for T {
    const TYPE: tblis_ffi::tblis::type_t = FLOAT_TYPE;

    fn to_ffi_scalar(&self) -> tblis_ffi::tblis::tblis_scalar {
        tblis_ffi::tblis::tblis_scalar {
            data: tblis_ffi::tblis::tblis_scalar_scalar { FLOAT_FIELD: *self },
            type_: FLOAT_TYPE,
        }
    }
}