summaryrefslogtreecommitdiff
path: root/src
diff options
context:
space:
mode:
authorCedric Nugteren <web@cedricnugteren.nl>2018-06-03 10:41:57 +0200
committerCedric Nugteren <web@cedricnugteren.nl>2018-06-03 10:41:57 +0200
commitbd1715aff99a80d0efe83c896e604377d8fd061b (patch)
tree53abfdc9d971acf1fb7507bb3d4eef3b6637d007 /src
parent4f594e39319912f0fbbc145d330ad9e09a66bbee (diff)
Fixes for CUDA version of CLBlast
Diffstat (limited to 'src')
-rw-r--r--src/cupp11.hpp4
1 files changed, 2 insertions, 2 deletions
diff --git a/src/cupp11.hpp b/src/cupp11.hpp
index 509ae3e8..a1cb1614 100644
--- a/src/cupp11.hpp
+++ b/src/cupp11.hpp
@@ -678,8 +678,8 @@ public:
}
// Regular constructor with memory management
- explicit Kernel(const Program &program, const std::string &name): name_(name) {
- CheckError(cuModuleGetFunction(&kernel_, program.GetModule(), name.c_str()));
+ explicit Kernel(const std::shared_ptr<Program> program, const std::string &name): name_(name) {
+ CheckError(cuModuleGetFunction(&kernel_, program->GetModule(), name.c_str()));
}
// Sets a kernel argument at the indicated position. This stores both the value of the argument