Searched refs:grf_perm (Results 1 – 3 of 3) sorted by relevance
| /dports/math/onednn/oneDNN-2.5.1/src/gpu/jit/conv/ |
| H A D | reorder_support.hpp | 112 const std::shared_ptr<grf_permutator_t> &grf_perm = nullptr) { in IR_DECL_DERIVED_TYPE_ID()
113 return func_t(new reorder_t(src_layout, dst_layout, grf_perm)); in IR_DECL_DERIVED_TYPE_ID()
122 && (grf_perm.get() == other.grf_perm.get()); in is_equal()
126 return ir_utils::get_hash(src_layout, dst_layout, grf_perm.get()); in get_hash()
141 std::shared_ptr<grf_permutator_t> grf_perm; member in dnnl::impl::gpu::jit::reorder_t
145 const std::shared_ptr<grf_permutator_t> &grf_perm) in reorder_t() argument
146 : src_layout(src_layout), dst_layout(dst_layout), grf_perm(grf_perm) {} in reorder_t()
|
| H A D | conv_kernel.hpp | 2356 if (grf_perm.is_empty()) return ret; in get_subregister()
2367 int new_base = grf_perm.map(base0); in get_subregister()
2370 ir_assert(grf_perm.map(base0 + i) == new_base + i) in get_subregister()
2431 auto s = get_subregister(hw, grf_perm, src, i, esize, in emit_reorder_1d_tile()
2433 auto d = get_subregister(hw, grf_perm, dst, i, esize, in emit_reorder_1d_tile()
2508 const grf_permutator_t &grf_perm) in reorder_impl_t() argument
2512 , grf_perm_(grf_perm) { in reorder_impl_t()
3093 grf_permutator_t grf_perm(hw); in reorder() local
3094 if (reorder_func.grf_perm) { in reorder()
3097 grf_perm = *reorder_func.grf_perm; in reorder()
[all …]
|
| H A D | kernel_builder.cpp | 44 permutation_injector_t(const grf_permutator_t &grf_perm) in permutation_injector_t() argument
45 : grf_perm_(new grf_permutator_t(grf_perm)) {} in permutation_injector_t()
81 grf_permutator_t grf_perm(hw_, c_buf_); in inject() local
89 if (try_convert_to_dpasw(a, b, grf_perm)) { in inject()
95 if (try_convert_to_dpasw(dpas_infos_[i], grf_perm)) { in inject()
343 dpas_info_t &a, dpas_info_t &b, grf_permutator_t &grf_perm) { in try_convert_to_dpasw() argument
387 grf_perm.set_permute(a_old, grf_new); in try_convert_to_dpasw()
388 grf_perm.set_permute(b_old, grf_new + grf_size * rcount / 2); in try_convert_to_dpasw()
448 grf_perm.set_permute(grf, grf); in try_convert_to_dpasw()
462 const stmt_t &stmt, const grf_permutator_t &grf_perm) { in apply_permutation_to_reorder() argument
[all …]
|