SIGN IN SIGN UP

nixlbench: change to range lookup in Neuron allocation tracking (#1833)

Today in nixlbench, when data is copied from Neuron device to host, it
checks the address of an IOV within the existing Neuron memory
allocation. The latter is done once with a single large buffer, which
has the base address different from the IOV's base address. Not found
neuron allocation leads to nixlbench failure.

## What?
Change to range lookup when searching in Neuron allocation tracking.




<!-- This is an auto-generated comment: release notes by coderabbit.ai
-->
## Summary by CodeRabbit

* **Bug Fixes**
* Fixed device copy and memset to operate on the correct region within
an existing allocation, using the full VA range (not just the base).
* Improved device allocation tracking to reliably resolve tensors for
any VA within a mapped range.
* Added stronger bounds checking for copy and fill operations to prevent
out-of-range reads and writes.
* Enhanced diagnostics with clearer messages when allocations are
missing, access is out of bounds, or device read/write calls fail.
<!-- end of auto-generated comment: release notes by coderabbit.ai -->

---------

Co-authored-by: Adit Ranadive <aranadive@nvidia.com>
F
fengjica committed
cdb42525c7be1ce57ae6c6fd6e836703df2400d2
Parent: c17d52f
Committed by GitHub <noreply@github.com> on 7/24/2026, 7:03:12 AM