projects / platform / upstream / pytorch.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 50b914a) | patch
Move stream functions from CUDAContext to CUDAStream (#14110)
authorEdward Yang <ezyang@fb.com>
Tue, 20 Nov 2018 01:01:34 +0000 (17:01 -0800)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Tue, 20 Nov 2018 01:05:48 +0000 (17:05 -0800)
commit7d1db89ef9edd3ea844c08d682fd634e0c34d813
tree858f0f102fd59c40a8f515d6795329af83ea8926tree | snapshot
parent50b914aeebdcfbb379be59423995d65b51b09cdacommit | diff
Move stream functions from CUDAContext to CUDAStream (#14110)

Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/14110

I'm planning to move CUDAStream to c10/cuda, without also moving
CUDAContext, and so it's most convenient if these definitions
are in the actual header file in question.

Reviewed By: smessmer

Differential Revision: D13104693

fbshipit-source-id: 23ce492003091adadaa5ca6a17124213005046c2
aten/src/ATen/cuda/CUDAContext.cpp diff | blob | history
aten/src/ATen/cuda/CUDAContext.h diff | blob | history
aten/src/ATen/cuda/CUDAStream.cpp diff | blob | history
aten/src/ATen/cuda/CUDAStream.h diff | blob | history
Domain: Machine Learning / ML Framework;