Searched refs:bmnk_mapper_ (Results 1 – 3 of 3) sorted by relevance
| /dports/math/onednn/oneDNN-2.5.1/src/gpu/jit/conv/ |
| H A D | gemm_schedule.hpp | 124 : bmnk_mapper_(bmnk_mapper) {} in bmnk_block_mapper_t()
151 const auto &var = bmnk_mapper_.var(p.first, b.dim_idx); in create_prb_blocks()
152 b.dim_idx = bmnk_mapper_.dim_idx(abc_kind, var); in create_prb_blocks()
161 bmnk_mapper_t bmnk_mapper_; member in dnnl::impl::gpu::jit::bmnk_block_mapper_t
383 const bmnk_mapper_t &bmnk_mapper() const { return bmnk_mapper_; } in bmnk_mapper()
412 bmnk_mapper_.set_a_vars(a_view_.vvars()); in set_a_view()
417 bmnk_mapper_.set_b_vars(b_view_.vvars()); in set_b_view()
422 bmnk_mapper_.set_c_vars(c_view_.vvars()); in set_c_view()
588 for (int i = 0; i < bmnk_mapper_.ndims(abc_kind_t::a); i++) { in with_thread_group_k_slicing()
733 bmnk_mapper_.set_bmnk_kind(var, kind); in set_bmnk_kind()
[all …]
|
| H A D | gemm_schedule.cpp | 50 auto bmnk_kind = bmnk_mapper_.bmnk_kind(abc_kind, b.dim_idx); in push_block()
93 bmnk_layout.type(), bmnk_mapper_.ndims(abc_kind), 0, blocks); in map_from_bmnk()
|
| H A D | kernel_builder.cpp | 4253 , bmnk_mapper_(bmnk_mapper) in multiply_builder_t()
4327 bmnk_block_mapper_t from_bmnk_mapper(bmnk_mapper_); in try_build_dpas()
4332 a_layout = bmnk_mapper_.map_to_bmnk( in try_build_dpas()
4334 b_layout = bmnk_mapper_.map_to_bmnk( in try_build_dpas()
4425 auto &var = bmnk_mapper_.var(abc_kind, i); in create_loop_nest()
4437 loop.a_idx = bmnk_mapper_.dim_idx(abc_kind_t::a, loop.var); in create_loop_nest()
4438 loop.b_idx = bmnk_mapper_.dim_idx(abc_kind_t::b, loop.var); in create_loop_nest()
4439 loop.c_idx = bmnk_mapper_.dim_idx(abc_kind_t::c, loop.var); in create_loop_nest()
4576 layout_t c_layout(c_type, bmnk_mapper_.ndims(abc_kind_t::c), 0, {}); in compute_mad_c_layout()
4578 int c_dim_idx = bmnk_mapper_.dim_idx(abc_kind_t::c, block_loop.var); in compute_mad_c_layout()
[all …]
|