From 92c26a35d425d4e8ca1b805ea776ea10f5ded3df Mon Sep 17 00:00:00 2001
From: Matthew Honnibal <honnibal+gh@gmail.com>
Date: Tue, 27 Mar 2018 16:42:00 +0000
Subject: [PATCH] Update get_cuda_stream

---
 spacy/util.py | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/spacy/util.py b/spacy/util.py
index 73d314e27..0d22198f6 100644
--- a/spacy/util.py
+++ b/spacy/util.py
@@ -14,6 +14,7 @@ from collections import OrderedDict
 import inspect
 import warnings
 from thinc.neural._classes.model import Model
+from thinc.neural.ops import NumpyOps
 import functools
 import cytoolz
 import itertools
@@ -241,7 +242,12 @@ def is_in_jupyter():
 
 
 def get_cuda_stream(require=False):
-    return CudaStream() if CudaStream is not None else None
+    if CudaStream is None:
+        return None
+    elif isinstance(Model.ops, NumpyOps):
+        return None
+    else:
+        return CudaStream()
 
 
 def get_async(stream, numpy_array):