Home
last modified time | relevance | path

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

/dports/math/onednn/oneDNN-2.5.1/src/gpu/jit/conv/
H A Dkernel_builder.cpp2621 , ab_slm_size_(ab_slm_size) in simple_slm_buffering_injector_t()
2757 slm_buf, non_ab_slm_size + ab_slm_size_ * cfg_.slm_bufs); in inject()
2791 send_t::arg_mem_off(new_args) += ab_slm_size_ * slm_idx; in sub_slm_bufs()
2802 int ab_slm_size_; member in dnnl::impl::gpu::jit::simple_slm_buffering_injector_t
2825 , ab_slm_size_(ab_slm_size) in unrolling_injector_t()
2922 = alloc_mgr_.alloc_size(slm_buf) - ab_slm_size_; in inject()
2924 non_ab_slm_size + ab_slm_size_ * cfg_.slm_bufs); in inject()
3191 int ab_slm_size_; member in dnnl::impl::gpu::jit::unrolling_injector_t
5273 int ab_slm_size() const { return ab_slm_size_; } in ab_slm_size()
5577 ab_slm_size_ += xp_slm_layout.size(); in prepare_gmem_to_slm_impl()
[all …]