diff --git a/crates/specialized/rtx-cfd/tests/kernels_tests.rs b/crates/specialized/rtx-cfd/tests/kernels_tests.rs index a8b8f29..bb1ff10 100644 --- a/crates/specialized/rtx-cfd/tests/kernels_tests.rs +++ b/crates/specialized/rtx-cfd/tests/kernels_tests.rs @@ -40,13 +40,13 @@ mod cuda_tests { // Allocate GPU memory let mut d_phi = kernel_manager.allocate_f32(nx)?; - let d_phi_new = kernel_manager.allocate_f32(nx)?; + let mut d_phi_new = kernel_manager.allocate_f32(nx)?; // Copy to GPU d_phi = kernel_manager.copy_to_device(&phi)?; // Run advection kernel - advection_kernel.apply(&d_phi, &d_phi_new, velocity as f32, dt as f32, dx as f32)?; + advection_kernel.apply(&d_phi, &mut d_phi_new, velocity as f32, dt as f32, dx as f32)?; // Copy result back let mut result = vec![0.0f32; nx]; @@ -98,10 +98,10 @@ mod cuda_tests { } let mut d_phi = kernel_manager.allocate_f32(nx)?; - let d_phi_new = kernel_manager.allocate_f32(nx)?; + let mut d_phi_new = kernel_manager.allocate_f32(nx)?; d_phi = kernel_manager.copy_to_device(&phi)?; - advection_kernel.apply(&d_phi, &d_phi_new, velocity as f32, dt as f32, dx as f32)?; + advection_kernel.apply(&d_phi, &mut d_phi_new, velocity as f32, dt as f32, dx as f32)?; let mut result = vec![0.0f32; nx]; result = kernel_manager.copy_from_device(&d_phi_new)?; @@ -133,10 +133,10 @@ mod cuda_tests { } let mut d_temp = kernel_manager.allocate_f32(nx)?; - let d_temp_new = kernel_manager.allocate_f32(nx)?; + let mut d_temp_new = kernel_manager.allocate_f32(nx)?; d_temp = kernel_manager.copy_to_device(&temperature)?; - diffusion_kernel.apply(&d_temp, &d_temp_new, alpha as f32, dt as f32, dx as f32)?; + diffusion_kernel.apply(&d_temp, &mut d_temp_new, alpha as f32, dt as f32, dx as f32)?; let mut result = vec![0.0f32; nx]; result = kernel_manager.copy_from_device(&d_temp_new)?;