Fixed an event synchronisation issue in the batched gemm routines

pull/246/head
Cedric Nugteren 2018-01-26 20:37:04 +01:00
parent 3651b51664
commit caebe8a9d5
2 changed files with 2 additions and 2 deletions

View File

@ -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

View File

@ -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