Loading...
Searching...
No Matches
C++ API.
More...
#include "group/gemm/api.hpp"
#include "group/gemm/compute_policy.hpp"
Go to the source code of this file.
|
| class | gpu::xetla::group::detail::check_2d_block_pitch_alignment< dtype_a, dtype_b, alignment_a, alignment_b, arch_tag > |
| |
| class | gpu::xetla::group::gemm_selector_t< dtype_a, dtype_b, mem_layout_a, mem_layout_b, mem_space_a, mem_space_b, alignment_a, alignment_b, dtype_acc, tile_shape, k_stride, mma_engine::xmx, arch_tag, stages, sync_freq, std::enable_if_t< detail::check_2d_block_pitch_alignment< dtype_a, dtype_b, alignment_a, alignment_b, arch_tag >::value > > |
| | Selects 2d block && xmx based gemm. More...
|
| |
| class | gpu::xetla::group::gemm_selector_t< dtype_a, dtype_b, mem_layout_a, mem_layout_b, mem_space_a, mem_space_b, alignment_a, alignment_b, dtype_acc, tile_shape, k_stride, mma_engine::xmx, arch_tag, stages, sync_freq, std::enable_if_t<!detail::check_2d_block_pitch_alignment< dtype_a, dtype_b, alignment_a, alignment_b, arch_tag >::value > > |
| | Selects scatter && xmx based brgemm. More...
|
| |
| class | gpu::xetla::group::gemm_selector_t< dtype_a, dtype_b, mem_layout_a, mem_layout_b, mem_space_a, mem_space_b, alignment_a, alignment_b, dtype_acc, tile_shape, k_stride, mma_engine::fpu, arch_tag, stages, sync_freq, std::enable_if_t< detail::check_2d_block_pitch_alignment< dtype_a, dtype_b, alignment_a, alignment_b, arch_tag >::value > > |
| | Selects 2d block && fpu based gemm. More...
|
| |