提交 ffd8baf5 编写于 作者: L liuruilong

update opencl use

上级 63267017
...@@ -55,11 +55,11 @@ class CLScope { ...@@ -55,11 +55,11 @@ class CLScope {
} }
auto program = CLEngine::Instance()->CreateProgramWith(context_.get(), file_name); auto program = CLEngine::Instance()->CreateProgramWith(context_.get(), file_name);
programs_[file_name] = program; programs_[file_name] = std::move(program);
status_ = clBuildProgram(program, 0, 0, 0, 0, 0); status_ = clBuildProgram(program, 0, 0, 0, 0, 0);
CL_CHECK_ERRORS(status_); CL_CHECK_ERRORS(status_);
return program; return program.get();
} }
private: private:
......
...@@ -18,7 +18,6 @@ limitations under the License. */ ...@@ -18,7 +18,6 @@ limitations under the License. */
#include <string> #include <string>
#include <vector> #include <vector>
#include "common/enforce.h" #include "common/enforce.h"
#include "common/type_define.h" #include "common/type_define.h"
#include "common/types.h" #include "common/types.h"
...@@ -170,7 +169,6 @@ class OpKernelBase { ...@@ -170,7 +169,6 @@ class OpKernelBase {
CLHelper cl_helper_; CLHelper cl_helper_;
}; };
#define DEFINE_OP_CONSTRUCTOR(cls, parent_cls) \ #define DEFINE_OP_CONSTRUCTOR(cls, parent_cls) \
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册