37 #include "compute_kernel_writer/include/ckw/KernelWriter.h"
44 namespace experimental
46 namespace dynamic_fusion
70 "Only the weights tensor can be exported to cl_image");
84 const bool using_bias = _bia !=
nullptr;
101 const auto stride_x =
static_cast<int32_t
>(_attributes.
stride().x());
102 const auto stride_y =
static_cast<int32_t
>(_attributes.
stride().y());
103 const auto pad_x =
static_cast<int32_t
>(_attributes.
pad().left);
104 const auto pad_y =
static_cast<int32_t
>(_attributes.
pad().top);
105 const auto kernel_size = kernel_width * kernel_height;
110 auto const_kernel_w_i32 = writer->declare_constant_tile(ckw::ConstantData({{kernel_width}}, ckw::DataType::Int32));
111 auto const_kernel_size_i32 =
112 writer->declare_constant_tile(ckw::ConstantData({{kernel_size}}, ckw::DataType::Int32));
113 auto const_src_c_i32 = writer->declare_constant_tile(ckw::ConstantData({{src_c}}, ckw::DataType::Int32));
114 auto const_src_w_i32 = writer->declare_constant_tile(ckw::ConstantData({{src_w}}, ckw::DataType::Int32));
115 auto const_src_h_i32 = writer->declare_constant_tile(ckw::ConstantData({{src_h}}, ckw::DataType::Int32));
116 auto const_dst_w_i32 = writer->declare_constant_tile(ckw::ConstantData({{dst_w}}, ckw::DataType::Int32));
117 auto const_stride_x_i32 = writer->declare_constant_tile(ckw::ConstantData({{stride_x}}, ckw::DataType::Int32));
118 auto const_stride_y_i32 = writer->declare_constant_tile(ckw::ConstantData({{stride_y}}, ckw::DataType::Int32));
119 auto const_pad_x_i32 = writer->declare_constant_tile(ckw::ConstantData({{pad_x}}, ckw::DataType::Int32));
120 auto const_pad_y_i32 = writer->declare_constant_tile(ckw::ConstantData({{pad_y}}, ckw::DataType::Int32));
121 auto const_k0_i32 = writer->declare_constant_tile(ckw::ConstantData({{k0}}, ckw::DataType::Int32));
122 auto const_0_i32 = writer->declare_constant_tile(ckw::ConstantData({{0}}, ckw::DataType::Int32));
123 auto const_pos_1_i32 = writer->declare_constant_tile(ckw::ConstantData({{1}}, ckw::DataType::Int32));
124 auto const_neg_1_i32 = writer->declare_constant_tile(ckw::ConstantData({{-1}}, ckw::DataType::Int32));
125 auto const_0_fp = writer->declare_constant_tile(ckw::ConstantData({{0.0f}}, dst_dt));
126 auto const_src_c_i32_minus_k0_i32 =
127 writer->declare_constant_tile(ckw::ConstantData({{src_c - k0}}, ckw::DataType::Int32));
138 const int32_t dst_n0 = root_window.
x().
step();
139 const int32_t dst_m0 = root_window.y().step();
142 const int32_t dst_n0_partial = _dst->
dimension(0) % dst_n0;
146 const int32_t dst_shift_back = (dst_n0 - dst_n0_partial) % dst_n0;
148 ckw::TensorSampler sampler_dst;
149 sampler_dst.format(ckw::TensorSamplerFormat::Dim0_Dim1xDim2_1);
150 if (dst_n0_partial == 0)
156 sampler_dst.address_mode_x(ckw::TensorSamplerAddressModeX::OverlappingMin);
159 if (dst_m0_partial == 0)
165 sampler_dst.address_mode_y(ckw::TensorSamplerAddressModeY::ClampToBorderMaxOnly);
169 sampler_dst.storage(ckw::TensorStorageType::BufferUint8Ptr);
172 auto tile_dst = writer->declare_tile(
"dst", ckw::TileInfo(dst_dt, dst_m0, dst_n0));
175 writer->op_assign(tile_dst, const_0_fp);
178 dst->init_virtual_tensor(tile_dst, sampler_dst);
184 auto const_dst_n0_i32 = writer->declare_constant_tile(ckw::ConstantData({{dst_n0}}, ckw::DataType::Int32));
185 auto const_dst_m0_i32 = writer->declare_constant_tile(ckw::ConstantData({{dst_m0}}, ckw::DataType::Int32));
186 auto const_shift_back_dst_n0_i32 =
187 writer->declare_constant_tile(ckw::ConstantData({{dst_shift_back}}, ckw::DataType::Int32));
207 auto address_mode_y_src =
209 ckw::TensorSampler sampler_src;
210 sampler_src.format(ckw::TensorSamplerFormat::Dim0_Dim1xDim2_1);
212 sampler_src.address_mode_y(address_mode_y_src);
214 sampler_src.storage(ckw::TensorStorageType::BufferUint8Ptr);
218 ckw::TensorSampler sampler_wei;
219 sampler_wei.format(ckw::TensorSamplerFormat::Dim0_Dim1xDim2_1);
225 sampler_wei.storage(ckw::TensorStorageType::Texture2dReadOnly);
229 sampler_wei.storage(ckw::TensorStorageType::BufferUint8Ptr);
233 ckw::TensorSampler sampler_bia;
237 sampler_bia.format(ckw::TensorSamplerFormat::Dim0_Dim1xDim2_1);
238 sampler_bia.address_mode_x(sampler_dst.address_mode_x());
241 sampler_bia.storage(ckw::TensorStorageType::BufferUint8Ptr);
253 auto tile_gid_0 = writer->declare_tile(
"gid_0", ckw::TileInfo(ckw::DataType::Int32));
254 auto tile_gid_1 = writer->declare_tile(
"gid_1", ckw::TileInfo(ckw::DataType::Int32));
255 auto tile_gid_2 = writer->declare_tile(
"gid_2", ckw::TileInfo(ckw::DataType::Int32));
257 writer->op_get_global_id(tile_gid_0, 0);
258 writer->op_get_global_id(tile_gid_1, 1);
259 writer->op_get_global_id(tile_gid_2, 2);
261 auto tile_cout = writer->declare_tile(
"cout", ckw::TileInfo(ckw::DataType::Int32));
262 auto tile_mout = writer->declare_tile(
"mout", ckw::TileInfo(ckw::DataType::Int32));
263 auto tile_bout = writer->declare_tile(
"bout", ckw::TileInfo(ckw::DataType::Int32));
267 const_shift_back_dst_n0_i32, const_0_i32);
276 for (int32_t m = 0; m < dst_m0; ++m)
278 std::vector<int32_t> idx{m};
282 const auto &const_idxs = writer->declare_constant_tile(ckw::ConstantData(it, ckw::DataType::Int32));
284 auto tile_xi = writer->declare_tile(
"xi", ckw::TileInfo(ckw::DataType::Int32, dst_m0, 1));
285 auto tile_yi = writer->declare_tile(
"yi", ckw::TileInfo(ckw::DataType::Int32, dst_m0, 1));
290 writer->op_binary(tile_xi, ckw::BinaryOp::Add, tile_mout, const_idxs);
291 writer->op_binary(tile_yi, ckw::BinaryOp::Add, tile_mout, const_idxs);
292 writer->op_binary(tile_xi, ckw::BinaryOp::Mod, tile_xi, const_dst_w_i32);
293 writer->op_binary(tile_yi, ckw::BinaryOp::Div, tile_yi, const_dst_w_i32);
294 writer->op_binary(tile_xi, ckw::BinaryOp::Mul, tile_xi, const_stride_x_i32);
295 writer->op_binary(tile_yi, ckw::BinaryOp::Mul, tile_yi, const_stride_y_i32);
296 writer->op_binary(tile_xi, ckw::BinaryOp::Sub, tile_xi, const_pad_x_i32);
297 writer->op_binary(tile_yi, ckw::BinaryOp::Sub, tile_yi, const_pad_y_i32);
299 auto tile_y_b = writer->declare_tile(
"y_b", ckw::TileInfo(ckw::DataType::Int32));
300 writer->op_binary(tile_y_b, ckw::BinaryOp::Mul, tile_cout, const_kernel_size_i32);
302 auto tile_i = writer->declare_tile(
"i", ckw::TileInfo(ckw::DataType::Int32));
303 writer->op_assign(tile_i, const_0_i32);
306 writer->op_for_loop(tile_i, ckw::BinaryOp::Less, const_kernel_size_i32, tile_i, ckw::AssignmentOp::Increment, const_pos_1_i32, [&]()
308 auto tile_x_k = writer->declare_tile(
"x_k", ckw::TileInfo(ckw::DataType::Int32));
309 auto tile_y_k = writer->declare_tile(
"y_k", ckw::TileInfo(ckw::DataType::Int32));
311 writer->op_binary(tile_x_k, ckw::BinaryOp::Mod, tile_i, const_kernel_w_i32);
312 writer->op_binary(tile_y_k, ckw::BinaryOp::Div, tile_i, const_kernel_w_i32);
314 auto tile_ck = writer->declare_tile(
"ck", ckw::TileInfo(ckw::DataType::Int32));
315 writer->op_assign(tile_ck, const_0_i32);
325 auto tile_xs = writer->declare_tile(
"xs", ckw::TileInfo(ckw::DataType::Int32, dst_m0, 1));
326 auto tile_ys = writer->declare_tile(
"ys", ckw::TileInfo(ckw::DataType::Int32, dst_m0, 1));
327 auto tile_mi = writer->declare_tile(
"mi", ckw::TileInfo(ckw::DataType::Int32, dst_m0, 1));
329 auto tile_xs_gte_0 = writer->declare_tile(
"xs_gte_0", ckw::TileInfo(ckw::DataType::Int32, dst_m0, 1));
330 auto tile_ys_gte_0 = writer->declare_tile(
"ys_gte_0", ckw::TileInfo(ckw::DataType::Int32, dst_m0, 1));
331 auto tile_xs_lt_w = writer->declare_tile(
"xs_lt_w", ckw::TileInfo(ckw::DataType::Int32, dst_m0, 1));
332 auto tile_ys_lt_h = writer->declare_tile(
"ys_lt_h", ckw::TileInfo(ckw::DataType::Int32, dst_m0, 1));
334 writer->op_binary(tile_xs, ckw::BinaryOp::Add, tile_xi, tile_x_k);
335 writer->op_binary(tile_ys, ckw::BinaryOp::Add, tile_yi, tile_y_k);
336 writer->op_binary(tile_mi, ckw::BinaryOp::Mul, tile_ys, const_src_w_i32);
337 writer->op_binary(tile_mi, ckw::BinaryOp::Add, tile_mi, tile_xs);
338 writer->op_binary(tile_xs_gte_0, ckw::BinaryOp::GreaterEqual, tile_xs, const_0_i32);
339 writer->op_binary(tile_ys_gte_0, ckw::BinaryOp::GreaterEqual, tile_ys, const_0_i32);
340 writer->op_binary(tile_xs_lt_w, ckw::BinaryOp::Less, tile_xs, const_src_w_i32);
341 writer->op_binary(tile_ys_lt_h, ckw::BinaryOp::Less, tile_ys, const_src_h_i32);
342 writer->op_ternary(tile_mi, ckw::TernaryOp::Select, const_neg_1_i32, tile_mi, tile_xs_gte_0);
343 writer->op_ternary(tile_mi, ckw::TernaryOp::Select, const_neg_1_i32, tile_mi, tile_ys_gte_0);
344 writer->op_ternary(tile_mi, ckw::TernaryOp::Select, const_neg_1_i32, tile_mi, tile_xs_lt_w);
345 writer->op_ternary(tile_mi, ckw::TernaryOp::Select, const_neg_1_i32, tile_mi, tile_ys_lt_h);
347 writer->op_for_loop(tile_ck, ckw::BinaryOp::LessEqual, const_src_c_i32_minus_k0_i32, tile_ck, ckw::AssignmentOp::Increment, const_k0_i32, [&]()
349 auto tile_lhs = writer->declare_tile(
"lhs", ckw::TileInfo(
to_ckw(_src->
data_type()), dst_m0, k0));
350 auto tile_rhs = writer->declare_tile(
"rhs", ckw::TileInfo(
to_ckw(_wei->
data_type()), dst_n0, k0));
351 writer->op_assign(tile_lhs, const_0_fp);
352 writer->op_assign(tile_rhs, const_0_fp);
354 writer->op_load_indirect(tile_lhs,
src->tensor(), sampler_src, tile_ck, tile_mi, const_0_i32, tile_bout);
355 writer->op_load_dilated(tile_rhs, wei->
tensor(), sampler_wei, tile_ck, tile_y_b, const_0_i32, const_0_i32, const_pos_1_i32, const_kernel_size_i32);
357 writer->op_binary(tile_dst, ckw::BinaryOp::MatMul_Nt_T, tile_lhs, tile_rhs);
361 if(((src_c % k0) != 0))
363 writer->op_for_loop(tile_ck, ckw::BinaryOp::Less, const_src_c_i32, tile_ck, ckw::AssignmentOp::Increment, const_pos_1_i32, [&]()
365 auto tile_lhs = writer->declare_tile(
"lhs_leftover", ckw::TileInfo(
to_ckw(_src->
data_type()), dst_m0, 1));
366 auto tile_rhs = writer->declare_tile(
"rhs_leftover", ckw::TileInfo(
to_ckw(_wei->
data_type()), dst_n0, 1));
367 writer->op_assign(tile_lhs, const_0_fp);
368 writer->op_assign(tile_rhs, const_0_fp);
370 writer->op_load_indirect(tile_lhs,
src->tensor(), sampler_src, tile_ck, tile_mi, const_0_i32, tile_bout);
371 writer->op_load_dilated(tile_rhs, wei->
tensor(), sampler_wei, tile_ck, tile_y_b, const_0_i32, const_0_i32, const_pos_1_i32, const_kernel_size_i32);
373 writer->op_binary(tile_dst, ckw::BinaryOp::MatMul_Nt_T, tile_lhs, tile_rhs);
377 writer->op_binary(tile_y_b, ckw::BinaryOp::Add, tile_y_b, const_pos_1_i32);
387 auto tile_bia = writer->declare_tile(
"bia", ckw::TileInfo(
to_ckw(_src->
data_type()), 1, dst_n0));
388 writer->op_load(tile_bia, bia->
tensor(), sampler_bia, tile_cout, const_0_i32, const_0_i32, const_0_i32);
391 auto &tile_bia = bia->
tile();
393 writer->op_binary(tile_dst, ckw::BinaryOp::Add, tile_dst, tile_bia);
422 return "direct_conv2d";