gitadmin/foxhunt
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: update GPU profile test assertions to match current TOML values

Browse Source 
Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
This commit is contained in:
jgrusewski
2026-04-04 01:04:46 +02:00
parent 047b420b55
commit 514635a9d2
1 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
crates/ml-core/src/gpu/profile.rs
View File

@@ -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);
}