Home
last modified time | relevance | path

Searched refs:bmnk_kinds (Results 1 – 3 of 3) sorted by relevance

/dports/math/onednn/oneDNN-2.5.1/src/gpu/jit/conv/
H A Dgemm_schedule.cpp25 const std::vector<bmnk_kind_t> &bmnk_kinds, const view_t &view) const { in map_to_bmnk() argument
27 return map_to_bmnk(abc_kind, bmnk_kinds, layout); in map_to_bmnk()
31 const std::vector<bmnk_kind_t> &bmnk_kinds, in map_to_bmnk() argument
37 for (int i = 0; i < int(bmnk_kinds.size()); i++) { in map_to_bmnk()
38 if (bmnk_kinds[i] == b_bmnk_kind) { in map_to_bmnk()
46 return layout_t(layout.type(), int(bmnk_kinds.size()), 0, blocks); in map_to_bmnk()
60 const std::vector<bmnk_kind_t> &bmnk_kinds, in map_from_bmnk() argument
74 auto &bmnk_blocks = tmp_blocks[static_cast<int>(bmnk_kinds[b.dim_idx])]; in map_from_bmnk()
79 for (auto bmnk_kind : bmnk_kinds) { in map_from_bmnk()
H A Dgemm_schedule.hpp52 bmnk_mapper_t(const object_map_t<expr_t, bmnk_kind_t> &bmnk_kinds) in bmnk_mapper_t() argument
53 : bmnk_kinds_(bmnk_kinds) {} in bmnk_mapper_t()
91 const std::vector<bmnk_kind_t> &bmnk_kinds,
95 const std::vector<bmnk_kind_t> &bmnk_kinds,
134 const std::vector<bmnk_kind_t> &bmnk_kinds,
H A Dkernel_builder.cpp4670 std::vector<bmnk_kind_t> bmnk_kinds; in convert_to_fma_friendly_layout() local
4672 bmnk_kinds.push_back(bmnk_kind_t::m); in convert_to_fma_friendly_layout()
4673 bmnk_kinds.push_back(bmnk_kind_t::k); in convert_to_fma_friendly_layout()
4675 bmnk_kinds.push_back(bmnk_kind_t::k); in convert_to_fma_friendly_layout()
4676 bmnk_kinds.push_back(bmnk_kind_t::n); in convert_to_fma_friendly_layout()
4679 auto bmnk_layout = bmnk_mapper.map_to_bmnk(abc_kind, bmnk_kinds, layout); in convert_to_fma_friendly_layout()
4691 = from_bmnk_mapper.map_from_bmnk(abc_kind, bmnk_kinds, dpas_layout); in convert_to_fma_friendly_layout()