cuda-rust-wasm 0.1.7

CUDA to Rust transpiler with WebGPU/WASM support
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

//! Backend abstraction layer

pub mod backend_trait;
pub mod native_gpu;
pub mod webgpu;
pub mod webgpu_optimized;
pub mod wasm_runtime;

pub use backend_trait::{BackendTrait, BackendCapabilities};

// Re-export BackendTrait as Backend for backward compatibility
pub use backend_trait::BackendTrait as Backend;

/// Get the current backend implementation
pub fn get_backend() -> Box<dyn Backend> {
    #[cfg(target_arch = "wasm32")]
    {
        Box::new(webgpu::WebGPUBackend::new())
    }
    
    #[cfg(not(target_arch = "wasm32"))]
    {
        #[cfg(feature = "cuda-backend")]
        {
            if native_gpu::is_cuda_available() {
                return Box::new(native_gpu::NativeGPUBackend::new());
            }
        }
        
        // Fallback to CPU backend
        Box::new(wasm_runtime::WasmRuntime::new())
    }
}