explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2025-04-16 07:01:58 +03:00
Code Issues Packages Projects Releases Wiki Activity

Make cuda streams non-blocking by default

Browse Source
This commit is contained in:
Matthw Honnibal 2019-10-18 17:21:57 +02:00
parent ca0759b325
commit b2e8f37965
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
spacy/util.py
View File

@ -317,13 +317,13 @@ def is_in_jupyter():
return False
def get_cuda_stream(require=False):
def get_cuda_stream(require=False, non_blocking=True):
if CudaStream is None:
return None
elif isinstance(Model.ops, NumpyOps):
return None
else:
return CudaStream()
return CudaStream(non_blocking=non_blocking)
def get_async(stream, numpy_array):