pytorch

2718a540 - Stash streams on all visible devices when targeting rocm