summaryrefslogtreecommitdiff
path: root/src/clblast.cpp
diff options
context:
space:
mode:
authorKirill Mavreshko <kimavr@gmail.com>2017-05-26 15:04:19 +0500
committerKirill Mavreshko <kimavr@gmail.com>2017-05-26 15:04:19 +0500
commit628e1e8cce084657df854fa97ba18ad740270b23 (patch)
tree9c62c21fcceed2696d978f90e3ec93ccea3fdfc2 /src/clblast.cpp
parent9c703a6021b4875348d7effb60c58e08fb9566b3 (diff)
Fixes inability to run GEMM on multiple identical GPUs (issue #155)
Diffstat (limited to 'src/clblast.cpp')
-rw-r--r--src/clblast.cpp2
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});
}