projects / platform / upstream / pytorch.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 0c32e1b) | patch
Wrap C10 CUDAStream instead of cudaStream_t in THCPStream
authorShen Li <shenli@fb.com>
Wed, 9 Jan 2019 23:07:07 +0000 (15:07 -0800)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Wed, 9 Jan 2019 23:12:48 +0000 (15:12 -0800)
commit7b9f794580522156b369db04b2fcf96cd8ee5216
tree15ee83f328d0679220d82c9a61a5c699c00c6c3ctree | snapshot
parent0c32e1b43e2fda3f85aca1a213680dcaeda9c5d3commit | diff
Wrap C10 CUDAStream instead of cudaStream_t in THCPStream

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

Differential Revision: D13608337

Pulled By: mrshenli

fbshipit-source-id: 4c66ef89fad0dc14a11ddb69da92907797cd2828
test/test_cuda.py diff | blob | history
torch/csrc/cuda/Stream.cpp diff | blob | history
torch/csrc/cuda/Stream.h diff | blob | history
torch/cuda/streams.py diff | blob | history
Domain: Machine Learning / ML Framework;