[Offloading] Support full disk offloading (#4083)
* force hooks for disk
Signed-off-by: Kyle Sayers <kylesayrs@gmail.com>
* better asserts, add test
Signed-off-by: Kyle Sayers <kylesayrs@gmail.com>
* handle shared tensors
Signed-off-by: Kyle Sayers <kylesayrs@gmail.com>
* Apply suggestions from code review
Co-authored-by: Marc Sun <57196510+SunMarc@users.noreply.github.com>
* use expanded device map
Signed-off-by: Kyle Sayers <kylesayrs@gmail.com>
* add comment
Signed-off-by: Kyle Sayers <kylesayrs@gmail.com>
---------
Signed-off-by: Kyle Sayers <kylesayrs@gmail.com>
Co-authored-by: Marc Sun <57196510+SunMarc@users.noreply.github.com>
