From 514635a9d2d7d0d17c79ec7569a30c2eb72aa650 Mon Sep 17 00:00:00 2001 From: jgrusewski Date: Sat, 4 Apr 2026 01:04:46 +0200 Subject: [PATCH] fix: update GPU profile test assertions to match current TOML values Co-Authored-By: Claude Sonnet 4.6 --- crates/ml-core/src/gpu/profile.rs | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/crates/ml-core/src/gpu/profile.rs b/crates/ml-core/src/gpu/profile.rs index c83b17a14..d19b0135d 100644 --- a/crates/ml-core/src/gpu/profile.rs +++ b/crates/ml-core/src/gpu/profile.rs @@ -291,9 +291,10 @@ mod tests { #[test] fn test_embedded_a100_parses() { let profile: GpuProfile = toml::from_str(A100_TOML).unwrap(); - assert_eq!(profile.training.batch_size, 512); + assert_eq!(profile.training.batch_size, 2048); assert_eq!(profile.training.num_atoms, 51); - assert_eq!(profile.training.buffer_size, 200_000); + assert_eq!(profile.training.buffer_size, 0); // 0 = auto from VRAM + assert_eq!(profile.experience.gpu_timesteps_per_episode, 500); assert_eq!(profile.cuda.cuda_stack_bytes, 65536); } @@ -302,7 +303,7 @@ mod tests { let profile: GpuProfile = toml::from_str(DEFAULT_TOML).unwrap(); assert_eq!(profile.training.batch_size, 256); assert_eq!(profile.training.num_atoms, 21); - assert_eq!(profile.training.buffer_size, 50_000); + assert_eq!(profile.training.buffer_size, 0); // 0 = auto from VRAM assert_eq!(profile.experience.gpu_timesteps_per_episode, 200); assert_eq!(profile.cuda.cuda_stack_bytes, 32768); }