pytorch

07e595e8 - Add `device_idx` to `free_fn` in `CUDAPluggableAllocator` (#91398)

Add `device_idx` to `free_fn` in `CUDAPluggableAllocator` (#91398) This was requested by nvidia folks, track also the device_id in the free function. Pull Request resolved: https://github.com/pytorch/pytorch/pull/91398 Approved by: https://github.com/albanD