diff options
Diffstat (limited to 'src/runtime/GLES_COMPUTE/functions/GCSoftmaxLayer.cpp')
| -rw-r--r-- | src/runtime/GLES_COMPUTE/functions/GCSoftmaxLayer.cpp | 10 |
1 files changed, 5 insertions, 5 deletions
diff --git a/src/runtime/GLES_COMPUTE/functions/GCSoftmaxLayer.cpp b/src/runtime/GLES_COMPUTE/functions/GCSoftmaxLayer.cpp index 34464ff057..5221c5cc5d 100644 --- a/src/runtime/GLES_COMPUTE/functions/GCSoftmaxLayer.cpp +++ b/src/runtime/GLES_COMPUTE/functions/GCSoftmaxLayer.cpp @@ -63,9 +63,9 @@ void GCSoftmaxLayer::configure(const IGCTensor *input, IGCTensor *output, float 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); + GCScheduler::get().dispatch(_max_kernel, false); + GCScheduler::get().memory_barrier(); + GCScheduler::get().dispatch(_shift_exp_sum_kernel, false); + GCScheduler::get().memory_barrier(); + GCScheduler::get().dispatch(_norm_kernel); } |