diff options
author | Kirill Mavreshko <kimavr@gmail.com> | 2017-05-26 15:04:19 +0500 |
---|---|---|
committer | Kirill Mavreshko <kimavr@gmail.com> | 2017-05-26 15:04:19 +0500 |
commit | 628e1e8cce084657df854fa97ba18ad740270b23 (patch) | |
tree | 9c62c21fcceed2696d978f90e3ec93ccea3fdfc2 /src/clblast.cpp | |
parent | 9c703a6021b4875348d7effb60c58e08fb9566b3 (diff) |
Fixes inability to run GEMM on multiple identical GPUs (issue #155)
Diffstat (limited to 'src/clblast.cpp')
-rw-r--r-- | src/clblast.cpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/clblast.cpp b/src/clblast.cpp index 1eb1ddd7..a73b67f3 100644 --- a/src/clblast.cpp +++ b/src/clblast.cpp @@ -2470,7 +2470,7 @@ StatusCode OverrideParameters(const cl_device_id device, const std::string &kern // Clears the existing program & binary cache for routines with the target kernel const auto routine_names = Routine::routines_by_kernel.at(kernel_name); for (const auto &routine_name : routine_names) { - ProgramCache::Instance().RemoveBySubset<1, 2>(ProgramKey{nullptr, precision, routine_name}); + ProgramCache::Instance().RemoveBySubset<1, 2>(ProgramKey{nullptr, device, precision, routine_name}); BinaryCache::Instance().Remove(BinaryKey{precision, routine_name, device_name}); } |