sdfg::gpu::GpuMmaExpander Member List

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

arch_	sdfg::gpu::GpuMmaExpander	protected
block_size_hint_	sdfg::gpu::GpuMmaExpander	protected
code() const	sdfg::passes::CodeLibNodeExpander< math::tensor::MatMulNode >	inline
CodeLibNodeExpander(const data_flow::LibraryNodeCode &code)	sdfg::passes::CodeLibNodeExpander< math::tensor::MatMulNode >	inline
create_eltwise_add_block(AccessNodeExpand &standalone, const MmaBlockSize &mma_block_size, types::PrimitiveType acc_type, types::PrimitiveType output_type, const data_flow::ImplementationType &impl_type, const DebugInfo &debug_info, builder::StructuredSDFGBuilder &builder, const std::string &c_frag_name, const std::string &acc_frag_name, const std::string &store_frag_name, Block &block)	sdfg::gpu::GpuMmaExpander	static
create_fragment_load(passes::LibNodeExpander::AccessNodeExpand &standalone, types::PrimitiveType input_type, const data_flow::ImplementationType &impl_type, const DebugInfo &org_debug_info, builder::StructuredSDFGBuilder &builder, std::string frag_name, const MmaBlockSize &mma_block_size, MmaFragmentType frag_type, const GpuMmaFromMemoryLayout &from_memory, unsigned input_index, const char *operand_label, Block &block)	sdfg::gpu::GpuMmaExpander	static
create_fragment_mma(LibNodeExpander::AccessNodeExpand &standalone, const GpuArch &arch, GpuMmaTiling &mma_tiling, types::PrimitiveType input_type, const std::string &frag_a, const std::string &frag_b, const std::string &frag_acc, const data_flow::ImplementationType &impl_type, const DebugInfo &org_debug_info, builder::StructuredSDFGBuilder &builder, structured_control_flow::Block &block)	sdfg::gpu::GpuMmaExpander	static
create_fragment_store(passes::LibNodeExpander::AccessNodeExpand &standalone, types::PrimitiveType output_type, const data_flow::ImplementationType &impl_type, const DebugInfo &org_debug_info, builder::StructuredSDFGBuilder &builder, std::string frag_name, const MmaBlockSize &mma_block_size, MmaFragmentType frag_type, const GpuMmaFromMemoryLayout &to_memory, unsigned input_index, const char *operand_label, Block &block)	sdfg::gpu::GpuMmaExpander	static
expand_mma_standalone(LibNodeExpander::AccessNodeExpand &standalone, const GpuArch &arch, GpuMmaTiling &mma_tiling, const math::tensor::TensorLayout &layout_a, const math::tensor::TensorLayout &layout_b, const math::tensor::TensorLayout &layout_y, types::PrimitiveType input_type, types::PrimitiveType output_type, bool include_c_add, const DebugInfo &org_debug_info, const std::array< int, 3 > &args_order, bool ragged=false, int k_block_multiple=1)	sdfg::gpu::GpuMmaExpander	static
for_lib_node(const data_flow::LibraryNode &node) const override	sdfg::gpu::GpuMmaExpander	virtual
GpuMmaExpander(const GpuArch *arch, const MmaBlockSize *block_size_hint=nullptr, const BlockCounts *wave_tile_hint=nullptr)	sdfg::gpu::GpuMmaExpander	inline
handle_expand(LibNodeExpander::ExpandContext &context, structured_control_flow::Block &block, math::tensor::MatMulNode &node) const override	sdfg::gpu::GpuMmaExpander
sdfg::passes::CodeLibNodeExpander< math::tensor::MatMulNode >::handle_expand(LibNodeExpander::ExpandContext &context, structured_control_flow::Block &block, math::tensor::MatMulNode &node) const=0	sdfg::passes::TypedLibNodeExpander< math::tensor::MatMulNode >	pure virtual
sdfg::passes::LibNodeExpander::handle_expand(ExpandContext &context, structured_control_flow::Block &block, data_flow::LibraryNode &node) const =0	sdfg::passes::LibNodeExpander	pure virtual
InputUse enum name	sdfg::passes::LibNodeExpander
matched_code() const override	sdfg::passes::CodeLibNodeExpander< math::tensor::MatMulNode >	inlinevirtual
matches_possible_mma_pattern(const math::tensor::MatMulNode &node) const	sdfg::gpu::GpuMmaExpander	protectedvirtual
wave_tile_hint_	sdfg::gpu::GpuMmaExpander	protected
~GpuMmaExpander()=default	sdfg::gpu::GpuMmaExpander	virtual
~LibNodeExpander()=default	sdfg::passes::LibNodeExpander	virtual