search

NVIDIA / apex

A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch
BSD 3-Clause "New" or "Revised" License
8.42k stars 1.4k forks source link

Add GPUDirect Storage #1774

Closed Aidyn-A closed 9 months ago

Aidyn-A commented 9 months ago

This is a temporary placeholder for future PyTorch-native GPU Direct Storage until we push it to the upstream.

cc @eqy @Fuzzkatt

eqy commented 9 months ago

CC @crcrpar