diff options
Diffstat (limited to 'src')
| -rw-r--r-- | src/routines/levelx/xgemmbatched.cpp | 2 | ||||
| -rw-r--r-- | src/routines/levelx/xgemmstridedbatched.cpp | 2 |
2 files changed, 2 insertions, 2 deletions
diff --git a/src/routines/levelx/xgemmbatched.cpp b/src/routines/levelx/xgemmbatched.cpp index 6a089b8a..f5ce83e7 100644 --- a/src/routines/levelx/xgemmbatched.cpp +++ b/src/routines/levelx/xgemmbatched.cpp @@ -243,7 +243,7 @@ void XgemmBatched<T>::BatchedGemmIndirect(const size_t m, const size_t n, const // Launches the kernel auto eventKernel = Event(); - auto eventPointer = eventKernel.pointer(); + auto eventPointer = (!c_no_temp) ? eventKernel.pointer() : event_; RunKernel(kernel, queue_, device_, global, local, eventPointer, eventWaitList); // Runs the post-processing kernel if needed diff --git a/src/routines/levelx/xgemmstridedbatched.cpp b/src/routines/levelx/xgemmstridedbatched.cpp index 6165a396..48383cbd 100644 --- a/src/routines/levelx/xgemmstridedbatched.cpp +++ b/src/routines/levelx/xgemmstridedbatched.cpp @@ -199,7 +199,7 @@ void XgemmStridedBatched<T>::BatchedGemmIndirect(const size_t m, const size_t n, // Launches the kernel auto eventKernel = Event(); - auto eventPointer = eventKernel.pointer(); + auto eventPointer = (!c_no_temp) ? eventKernel.pointer() : event_; RunKernel(kernel, queue_, device_, global, local, eventPointer, eventWaitList); // Runs the post-processing kernel if needed |