diff options
Diffstat (limited to 'src/runtime/GLES_COMPUTE')
-rw-r--r-- | src/runtime/GLES_COMPUTE/functions/GCSoftmaxLayer.cpp | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/src/runtime/GLES_COMPUTE/functions/GCSoftmaxLayer.cpp b/src/runtime/GLES_COMPUTE/functions/GCSoftmaxLayer.cpp index d7d47d2802..1db927c8ff 100644 --- a/src/runtime/GLES_COMPUTE/functions/GCSoftmaxLayer.cpp +++ b/src/runtime/GLES_COMPUTE/functions/GCSoftmaxLayer.cpp @@ -61,6 +61,8 @@ void GCSoftmaxLayer::configure(const IGCTensor *input, IGCTensor *output) void GCSoftmaxLayer::run() { GCScheduler::get().enqueue(_max_kernel, false); + GCScheduler::get().sync(); GCScheduler::get().enqueue(_shift_exp_sum_kernel, false); + GCScheduler::get().sync(); GCScheduler::get().enqueue(_norm_kernel); } |