DeepSpeed
b36d1b6e - Route contiguous activation offload pins through accelerator pin_memory

Commit
3 days ago
Route contiguous activation offload pins through accelerator pin_memory Honor DS_PIN_MEMORY_BACKEND for the common contiguous checkpoint-input path. Strided views keep ATen empty_strided pin because native pin cannot preserve arbitrary strides. Document pin on/off vs backend; DeepCompile compile.offload_activation_pin_memory already landed in #8258. Signed-off-by: Olatunji Ruwase <tunji.ruwase@snowflake.com> Co-authored-by: Cursor <cursoragent@cursor.com>
Parents
Loading