diff options
| author | Sam Gross <sgross@fb.com> | 2016-10-17 18:49:23 +0300 |
|---|---|---|
| committer | Sam Gross <sgross@fb.com> | 2016-10-17 18:49:23 +0300 |
| commit | c848e7d07913c7f5981f7fceae85f3bb89b3141b (patch) | |
| tree | 73b9b22fc8e81291469019455b9889a5adf38336 | |
| parent | 9d35ccb9c3690f6b98acdaaec73b142da47f6665 (diff) | |
Fix cutorch.getStream()
state->numUserStreams does not include the NULL stream, which is stored
in res->streams[i]
| -rw-r--r-- | lib/THC/THCGeneral.c | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/lib/THC/THCGeneral.c b/lib/THC/THCGeneral.c index 1d4622b..2d478de 100644 --- a/lib/THC/THCGeneral.c +++ b/lib/THC/THCGeneral.c @@ -559,7 +559,7 @@ int THCState_getCurrentStreamIndex(THCState *state) int device; THCudaCheck(cudaGetDevice(&device)); THCCudaResourcesPerDevice* res = THCState_getDeviceResourcePtr(state, device); - for (int i = 0; i < state->numUserStreams; ++i) { + for (int i = 0; i <= state->numUserStreams; ++i) { if (res->streams[i] == stream) { return i; } |