transformers
1cdae08e - fix pin_memory for contiguous batching (#44455)

Commit
125 days ago
fix pin_memory for contiguous batching (#44455) * fix pin_memory for contiguous batching Signed-off-by: jiqing-feng <jiqing.feng@intel.com> * use get_available_devices Signed-off-by: jiqing-feng <jiqing.feng@intel.com> * add comment Signed-off-by: jiqing-feng <jiqing.feng@intel.com> --------- Signed-off-by: jiqing-feng <jiqing.feng@intel.com> Co-authored-by: Rémi Ouazan <83456801+remi-or@users.noreply.github.com>
Author
Parents
Loading