nu_plugin_vulkan_matrix 0.1.2

GPU matrix operations for Nushell using Vulkan
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

//! GPU-resident arrays with explicit host transfers.

extern crate self as vulkan_matrix_backend;
include!("../backend/src/lib.rs");

mod convert;
mod elementwise;
#[cfg(target_os = "linux")]
mod file_io;
pub mod gpu;
pub mod layout;
pub mod matmul;
pub mod plugin;
mod reduce_kernel;
pub mod reduction;
mod resident;

#[cfg(test)]
#[path = "../tests/unit/resident.rs"]
mod resident_tests;

#[cfg(test)]
#[path = "../tests/unit/convert.rs"]
mod convert_tests;

#[cfg(test)]
#[path = "../tests/unit/elementwise.rs"]
mod elementwise_tests;

#[cfg(test)]
#[path = "../tests/unit/gpu.rs"]
mod test_gpu;