sdfg::gpu::rocm::RocmMmaSupport Member List

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

base_support	sdfg::gpu::rocm::RocmMmaSupport
CDNA_BLOCK_SIZE	sdfg::gpu::rocm::RocmMmaSupport	protectedstatic
DEFAULT_BLOCK_SIZE	sdfg::gpu::rocm::RocmMmaSupport	protectedstatic
emit_bare_fragment_decl_type(std::ostream &os, MmaFragmentType type, const MmaBlockSize &dims, MmaFragmentLayout layout, types::PrimitiveType scalar_type, std::optional< std::pair< int, int > > coop_dims)	sdfg::gpu::rocm::RocmMmaSupport	static
emit_block_frag_type(std::ostream &os, const types::StorageType &storage_type, types::PrimitiveType element_type) const	sdfg::gpu::rocm::RocmMmaSupport
emit_block_frag_type(std::ostream &os, MmaFragmentType type, const MmaBlockSize &dims, MmaFragmentLayout layout, types::PrimitiveType scalar_type, std::optional< std::pair< int, int > > coop_dims)	sdfg::gpu::rocm::RocmMmaSupport	static
f32_support	sdfg::gpu::rocm::RocmMmaSupport
get_accumulator_type(types::PrimitiveType input_type, types::PrimitiveType output_type, types::PrimitiveType desired_acc_type) const override	sdfg::gpu::rocm::RocmMmaSupport	virtual
sdfg::gpu::GpuMmaSupport::get_accumulator_type(types::PrimitiveType input_type, types::PrimitiveType output_type) const	sdfg::gpu::GpuMmaSupport	inline
get_ceil_block_count(const symbolic::Expression &size, uint16_t block_size)	sdfg::gpu::GpuMmaSupport	static
get_integer_block_count(const symbolic::Expression &size, uint16_t block_size)	sdfg::gpu::GpuMmaSupport	static
get_mma_impl_type(const GpuMmaTiling &tiling) const override	sdfg::gpu::rocm::RocmMmaSupport	virtual
get_mma_tiling(const symbolic::MultiExpression &res_shape, types::PrimitiveType input_type, types::PrimitiveType acc_type, const MmaBlockSize *block_size_hint, const BlockCounts *wave_tile_hint) const override	sdfg::gpu::rocm::RocmMmaSupport	virtual
get_ragged_mma_tiling(const symbolic::MultiExpression &padded_shape, types::PrimitiveType input_type, types::PrimitiveType output_type, types::PrimitiveType acc_type, const MmaBlockSize *block_size_hint, const BlockCounts *wave_tile_hint) const override	sdfg::gpu::rocm::RocmMmaSupport	virtual
get_storage_type_arg_as_int(const types::StorageType &storage, int idx)	sdfg::gpu::GpuMmaSupport	protectedstatic
get_supported_block_sizes(types::PrimitiveType input_type, types::PrimitiveType acc_type) const override	sdfg::gpu::rocm::RocmMmaSupport	virtual
GpuMmaSupport()=default	sdfg::gpu::GpuMmaSupport
int8_block_k	sdfg::gpu::rocm::RocmMmaSupport
int8_block_size() const	sdfg::gpu::rocm::RocmMmaSupport	inline
is_mma_type(const types::StorageType &storage)	sdfg::gpu::rocm::RocmMmaSupport	static
is_valid_block_size(const MmaBlockSize &block_size, types::PrimitiveType input_type, types::PrimitiveType acc_type) const override	sdfg::gpu::rocm::RocmMmaSupport	virtual
MMA_FRAGMENT_STORAGE_TYPE	sdfg::gpu::rocm::RocmMmaSupport	static
MMA_REGS_STORAGE_TYPE	sdfg::gpu::rocm::RocmMmaSupport	static
RocmMmaSupport(bool base_support, bool f32_support, uint16_t threads, uint16_t int8_block_k=0)	sdfg::gpu::rocm::RocmMmaSupport	inline
select_variant(const GpuMmaTiling &tiling)	sdfg::gpu::rocm::RocmMmaSupport	static
set_mma_fragment_storage_type(types::StorageType &storage_type, const GpuMmaTiling &size, MmaFragmentType type, MmaFragmentLayout layout) const override	sdfg::gpu::rocm::RocmMmaSupport	virtual
supported_types(types::PrimitiveType input_type, types::PrimitiveType output_type) const override	sdfg::gpu::rocm::RocmMmaSupport	virtual
threads_per_mma_block	sdfg::gpu::rocm::RocmMmaSupport
try_get_mma_tiling(const MmaBlockSize &block_size, const BlockCounts *wave_tile_hint, const symbolic::MultiExpression &res_shape, types::PrimitiveType acc_type, bool ceil_counts=false) const	sdfg::gpu::rocm::RocmMmaSupport
~GpuMmaSupport()=default	sdfg::gpu::GpuMmaSupport	virtual