diff --git a/crates/neuron/src/harness/device_worker/dispatch.rs b/crates/neuron/src/harness/device_worker/dispatch.rs index d83379d..f1b1586 100644 --- a/crates/neuron/src/harness/device_worker/dispatch.rs +++ b/crates/neuron/src/harness/device_worker/dispatch.rs @@ -1015,6 +1015,10 @@ fn drain_poisoned(job: Job, device_index: u32) { Job::TpForwardLogits { reply, .. } => { let _ = reply.send(Err(err())); } + #[cfg(feature = "cuda")] + Job::TpForwardLogitsWithImages { reply, .. } => { + let _ = reply.send(Err(err())); + } Job::Shutdown => { // Filtered by the matches!() guard in run(); reaching // here would be a logic error.