search

NVIDIA / apex

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

Add GPUDirect Storage #1774

Closed Aidyn-A closed 4 months ago

Aidyn-A commented 5 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 5 months ago

CC @crcrpar