misc cleanups to prepare for rebase

Signed-off-by: Sage Moore <sage@neuralmagic.com>
This commit is contained in:
Sage Moore 2025-06-02 18:01:24 +00:00
parent 90e46ee5e3
commit 065816d25f

View File

@ -2122,7 +2122,7 @@ class CUDAGraphRunner(nn.Module):
**kwargs,
) -> torch.Tensor:
attn_metadata: AttentionMetadata = get_forward_context().attn_metadata
# Copy the input tensors to the input buffers.
self.input_buffers["input_ids"].copy_(input_ids, non_blocking=True)
if positions is not None: