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

style: cargo fmt --workspace (whitespace/wrapping only, no semantic change)

Browse Source 
Whole-workspace rustfmt pass picked up while iterating on Mamba GPU
backward work. Verified formatting-only via diff sampling; no logic
changed.

Co-Authored-By: Claude Sonnet 5 <[email protected]>
This commit is contained in:
osobh
2026-08-10 07:09:36 -07:00
co-authored by Claude Sonnet 5
parent ad6405663f
commit 4aaa36a57a
305 changed files with 25537 additions and 18337 deletions
Download Patch File Download Diff File Expand all files Collapse all files
demos/rtx-model-zoo/src/real_inference.rs
+20 -5
View File
@@ -139,10 +139,22 @@ impl RealInferenceEngine {
let attn_o = Tensor::randn(&[config.hidden_size, config.hidden_size], &device)
.expect("failed to allocate o_proj tensor");
weights.insert(format!("layers.{layer_idx}.attention.q_proj.weight"), attn_q);
weights.insert(format!("layers.{layer_idx}.attention.k_proj.weight"), attn_k);
weights.insert(format!("layers.{layer_idx}.attention.v_proj.weight"), attn_v);
weights.insert(format!("layers.{layer_idx}.attention.o_proj.weight"), attn_o);
weights.insert(
format!("layers.{layer_idx}.attention.q_proj.weight"),
attn_q,
);
weights.insert(
format!("layers.{layer_idx}.attention.k_proj.weight"),
attn_k,
);
weights.insert(
format!("layers.{layer_idx}.attention.v_proj.weight"),
attn_v,
);
weights.insert(
format!("layers.{layer_idx}.attention.o_proj.weight"),
attn_o,
);
let mlp_gate = Tensor::randn(&[config.hidden_size, config.hidden_size * 4], &device)
.expect("failed to allocate gate_proj tensor");
@@ -470,7 +482,10 @@ mod tests {
assert_eq!(info.name, DEMO_MODEL_NAME);
assert_eq!(info.config.vocab_size, VOCAB_SIZE);
assert_eq!(info.config.hidden_size, HIDDEN_SIZE);
assert!(info.parameter_count > 0, "should report real parameter count");
assert!(
info.parameter_count > 0,
"should report real parameter count"
);
assert!(info.memory_usage > 0, "should report real memory usage");
}