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

fix(ml-dqn): suppress unused grad_norm_tensor warning without cuda

Browse Source 
Prefix with underscore since it's only consumed in #[cfg(feature = "cuda")] block.

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
This commit is contained in:
jgrusewski
2026-03-10 10:33:18 +01:00
parent 2662d5cda2
commit e6e31305a2
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
crates/ml-dqn/src/dqn.rs
View File

@@ -3348,7 +3348,7 @@ impl DQN {
let result = self.compute_loss_internal(batch)?;
// Backward + clip WITHOUT optimizer step — zero GPU→CPU sync
let (grads, grad_norm_tensor) = if let Some(ref optimizer) = self.optimizer {
let (grads, _grad_norm_tensor) = if let Some(ref optimizer) = self.optimizer {
optimizer.backward_and_clip(&result.loss_tensor, self.gradient_clip_norm)?
} else {
return Err(MLError::TrainingError(
@@ -3358,10 +3358,10 @@ impl DQN {
// Squeeze grad_norm from [1] → rank-0 scalar tensor (GPU path only)
#[cfg(feature = "cuda")]
let grad_norm_scalar_tensor = if grad_norm_tensor.dims().len() > 0 {
grad_norm_tensor.squeeze(0)?
let grad_norm_scalar_tensor = if _grad_norm_tensor.dims().len() > 0 {
_grad_norm_tensor.squeeze(0)?
} else {
grad_norm_tensor
_grad_norm_tensor
};
Ok(GradientResult {