提交 ffd8baf5 编写于 作者: L liuruilong

update opencl use

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