From 628e1e8cce084657df854fa97ba18ad740270b23 Mon Sep 17 00:00:00 2001 From: Kirill Mavreshko Date: Fri, 26 May 2017 15:04:19 +0500 Subject: Fixes inability to run GEMM on multiple identical GPUs (issue #155) --- src/clblast.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'src/clblast.cpp') 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}); } -- cgit v1.2.3