diff options
author | Cedric Nugteren <web@cedricnugteren.nl> | 2018-01-26 20:37:04 +0100 |
---|---|---|
committer | Cedric Nugteren <web@cedricnugteren.nl> | 2018-01-26 20:37:04 +0100 |
commit | caebe8a9d52712d86bd7adcc7cf2448beab0fe0e (patch) | |
tree | 55920c1ad013ae83ac7b4c32f0426895051a4816 /src | |
parent | 3651b5166406527b2085d5b6ed100ddfd4ea2c5b (diff) |
Fixed an event synchronisation issue in the batched gemm routines
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 |