trueno-gpu 0.4.29

Pure Rust PTX generation for NVIDIA CUDA - no LLVM, no nvcc
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22

use super::*;

#[test]
fn test_gemm_kernel_builds() {
    let kernel = GemmKernel::naive(1024, 1024, 1024);
    let ptx = kernel.emit_ptx();
    assert!(ptx.contains(".visible .entry"));
    assert!(ptx.contains("gemm"));
}

#[test]
fn test_softmax_kernel_builds() {
    let kernel = SoftmaxKernel::new(4096);
    let ptx = kernel.emit_ptx();
    assert!(ptx.contains(".visible .entry"));
    assert!(ptx.contains("softmax"));
}

mod barrier_safety;
mod coverage_tests;
mod fusion_contract_falsify;
mod property_tests;