ug-cuda 0.5.0

Micro compiler for tensor operations.
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18

use ug::Result;

#[test]
fn lb_cuda_add() -> Result<()> {
    let dev = ug_cuda::Device::new(0)?;
    ug::common_tests::lb::add(&dev)
}

#[test]
fn lb_cuda_mm() -> Result<()> {
    let dev = ug_cuda::Device::new(0)?;
    ug::common_tests::lb::mm(&dev)
}

#[test]
fn lb_cuda_cat() -> Result<()> {
    ug::common_tests::lb::cat(&ug::CpuDevice)
}