docc 0.8.3 (0d44999b)
C++ API Reference
Loading...
Searching...
No Matches
sdfg::gpu::GpuMmaEinsumReplacer Member List

This is the complete list of members for sdfg::gpu::GpuMmaEinsumReplacer, including all inherited members.

analyze(const einsum::EinsumCluster &cluster, EinsumMmaAnalysis &result) constsdfg::gpu::GpuMmaEinsumReplacer
sdfg::einsum::Einsum2MatMul::analyze(const EinsumCluster &cluster, MatMulAnalysis &result) constsdfg::einsum::Einsum2MatMul
build_tensor_layout(const EinsumIndexing &indexing, bool swapped, const symbolic::Expression &outer_dim, const symbolic::Expression &inner_dim, const symbolic::Expression &outer_offset, const symbolic::Expression &inner_offset, const symbolic::Expression &external_offset) constsdfg::einsum::Einsum2MatMul
can_be_applied(const einsum::EinsumCluster &cluster) const overridesdfg::gpu::GpuMmaEinsumReplacervirtual
GpuMmaEinsumReplacer(const GpuArch *arch, const MmaBlockSize *block_size_hint=nullptr, const BlockCounts *wave_tile_hint=nullptr, int k_block_multiple=1)sdfg::gpu::GpuMmaEinsumReplacer
matches_possible_mma_pattern(EinsumMmaAnalysis &analysis) constsdfg::gpu::GpuMmaEinsumReplacerprotected
name() const overridesdfg::einsum::Einsum2MatMulinlinevirtual
replace(einsum::EinsumReplacementContext &context, const einsum::EinsumCluster &cluster) const overridesdfg::gpu::GpuMmaEinsumReplacervirtual
select_tiling(EinsumMmaAnalysis &analysis) constsdfg::gpu::GpuMmaEinsumReplacerprotected
~EinsumReplacer()=defaultsdfg::einsum::EinsumReplacervirtual