rustyverse/rustytorch
2
0
Fork 0
Code Issues Pull Requests Actions 46 Packages Projects Releases Wiki Activity

rtx-cfd legacy GPU tests: the current CudaKernelManager API (Arc-held manager, allocate-on-copy, copy_from_device returning the host vector), MatrixOpsKernel::tridiagonal_matvec added over the existing kernel, PoissonKernel::solve_jacobi_2d kept as the older name; CfdConfig literals take ..Default
CI / Build (macos-latest) (push) Waiting to run
Details
CI / Test (macos-latest) (push) Blocked by required conditions
Details
CI / Test (ubuntu-latest) (push) Blocked by required conditions
Details
CI / Python Bindings (maturin) (macos-latest) (push) Blocked by required conditions
Details
CI / Python Bindings (maturin) (ubuntu-latest) (push) Blocked by required conditions
Details
CI / WASM Build + Size Check (push) Blocked by required conditions
Details
CI / Distributed Training Tests (push) Blocked by required conditions
Details
CI / CI Success (push) Blocked by required conditions
Details
CI / Build CPU-Only (Explicit) (push) Failing after 6s
Details
CI / Build (ubuntu-latest) (push) Failing after 6s
Details
Documentation / Build API Documentation (push) Failing after 6s
Details
Documentation / Build User Guide (push) Successful in 7s
Details
CI / Format Check (push) Failing after 13s
Details
CI / Clippy Check (push) Failing after 37s
Details
Performance Benchmarks / Run Benchmarks (push) Successful in 3m23s
Details

Browse Source
This commit is contained in:
Omar Sobh
2026-09-16 07:18:34 -05:00
parent af19860c04
commit 1381b5b633
6 changed files with 109 additions and 48 deletions
Download Patch File Download Diff File Expand all files Collapse all files
crates/specialized/rtx-cfd/tests/gpu_kernel_comprehensive_tests.rs
+14 -7
View File
@@ -37,9 +37,10 @@ mod gpu_kernel_tests {
lz: 0.0,
dt: 0.001,
..Default::default()
..CfdConfig::default()
};
let manager = CudaKernelManager::new(&config)?;
let manager = std::sync::Arc::new(CudaKernelManager::new(&config)?);
// Test all advection schemes
let schemes = vec![
@@ -117,9 +118,10 @@ mod gpu_kernel_tests {
lz: 0.0,
dt: 0.001,
..Default::default()
..CfdConfig::default()
};
let manager = CudaKernelManager::new(&config)?;
let manager = std::sync::Arc::new(CudaKernelManager::new(&config)?);
let kernel = AdvectionKernel::new(&manager, AdvectionScheme::Upwind)?;
let nx = 64;
@@ -202,9 +204,10 @@ mod gpu_kernel_tests {
lz: 0.0,
dt: 0.0001,
..Default::default()
..CfdConfig::default()
};
let manager = CudaKernelManager::new(&config)?;
let manager = std::sync::Arc::new(CudaKernelManager::new(&config)?);
// Test all diffusion schemes
let schemes = vec![
@@ -288,9 +291,10 @@ mod gpu_kernel_tests {
lz: 0.0,
dt: 0.0001,
..Default::default()
..CfdConfig::default()
};
let manager = CudaKernelManager::new(&config)?;
let manager = std::sync::Arc::new(CudaKernelManager::new(&config)?);
let kernel = DiffusionKernel::new(&manager, DiffusionScheme::Explicit)?;
let nx = 64;
@@ -366,9 +370,10 @@ mod gpu_kernel_tests {
lz: 0.0,
dt: 0.001,
..Default::default()
..CfdConfig::default()
};
let manager = CudaKernelManager::new(&config)?;
let manager = std::sync::Arc::new(CudaKernelManager::new(&config)?);
let kernel = PoissonKernel::new(&manager)?;
let nx = 32;
@@ -443,9 +448,10 @@ mod gpu_kernel_tests {
lz: 0.0,
dt: 0.001,
..Default::default()
..CfdConfig::default()
};
let manager = CudaKernelManager::new(&config)?;
let manager = std::sync::Arc::new(CudaKernelManager::new(&config)?);
// Test allocation and deallocation of large arrays
let sizes = vec![1024, 65536, 262144];
@@ -491,9 +497,10 @@ mod gpu_kernel_tests {
lz: 0.0,
dt: 0.001,
..Default::default()
..CfdConfig::default()
};
let manager = CudaKernelManager::new(&config)?;
let manager = std::sync::Arc::new(CudaKernelManager::new(&config)?);
let kernel = AdvectionKernel::new(&manager, AdvectionScheme::Upwind)?;
let n = 512 * 512;