提交 ed095d40 编写于 作者: Y yejianwu

format code

上级 4ebaa508
...@@ -18,7 +18,7 @@ namespace { ...@@ -18,7 +18,7 @@ namespace {
const bool BufferToImageOpImpl(Tensor *buffer, const bool BufferToImageOpImpl(Tensor *buffer,
Tensor *image, Tensor *image,
std::vector<size_t> &image_shape) { const std::vector<size_t> &image_shape) {
std::unique_ptr<BufferBase> kernel_error; std::unique_ptr<BufferBase> kernel_error;
uint32_t gws[2] = {static_cast<uint32_t>(image_shape[0]), uint32_t gws[2] = {static_cast<uint32_t>(image_shape[0]),
static_cast<uint32_t>(image_shape[1])}; static_cast<uint32_t>(image_shape[1])};
...@@ -35,7 +35,8 @@ const bool BufferToImageOpImpl(Tensor *buffer, ...@@ -35,7 +35,8 @@ const bool BufferToImageOpImpl(Tensor *buffer,
built_options.emplace("-DNON_UNIFORM_WORK_GROUP"); built_options.emplace("-DNON_UNIFORM_WORK_GROUP");
} }
if (buffer->dtype() == image->dtype()) { if (buffer->dtype() == image->dtype()) {
built_options.emplace("-DDATA_TYPE=" + DtToCLDt(DataTypeToEnum<float>::value)); built_options.emplace("-DDATA_TYPE=" +
DtToCLDt(DataTypeToEnum<float>::value));
built_options.emplace("-DCMD_DATA_TYPE=" + built_options.emplace("-DCMD_DATA_TYPE=" +
DtToCLCMDDt(DataTypeToEnum<float>::value)); DtToCLCMDDt(DataTypeToEnum<float>::value));
} else { } else {
...@@ -102,7 +103,8 @@ const bool BufferToImageOpImpl(Tensor *buffer, ...@@ -102,7 +103,8 @@ const bool BufferToImageOpImpl(Tensor *buffer,
bool is_out_of_range = false; bool is_out_of_range = false;
if (runtime->IsOutOfRangeCheckEnabled()) { if (runtime->IsOutOfRangeCheckEnabled()) {
kernel_error->Map(nullptr); kernel_error->Map(nullptr);
is_out_of_range = *(kernel_error->mutable_data<char>()) == '1' ? true : false; is_out_of_range =
*(kernel_error->mutable_data<char>()) == '1' ? true : false;
kernel_error->UnMap(); kernel_error->UnMap();
} }
return is_out_of_range; return is_out_of_range;
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册