Skip to content

Commit b17fc49

Browse files
committed
fixup: common: prim_execute: block on wait on gpu properly
1 parent 9f8ada2 commit b17fc49

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

src/common/primitive_iface.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -156,7 +156,7 @@ status_t primitive_execute(
156156
& dnnl::threadpool_interop::threadpool_iface::
157157
ASYNCHRONOUS)
158158
&& stream->engine()->kind() == engine_kind::cpu;
159-
block_on_wait = !async_cpu;
159+
block_on_wait = !is_async_cpu;
160160
#endif
161161
if (block_on_wait) stream->wait();
162162
double start_ms = get_msec();

0 commit comments

Comments
 (0)