Add Device context manager for temporary device switching#1597
Draft
Andy-Jost wants to merge 1 commit intoNVIDIA:mainfrom
Draft
Add Device context manager for temporary device switching#1597Andy-Jost wants to merge 1 commit intoNVIDIA:mainfrom
Andy-Jost wants to merge 1 commit intoNVIDIA:mainfrom