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

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

base_supportsdfg::gpu::cuda::CudaMmaSupport
CudaMmaSupport(bool base_support, bool tf32_support, bool fp64_support)sdfg::gpu::cuda::CudaMmaSupportinline
DEFAULT_BLOCK_SIZEsdfg::gpu::cuda::CudaMmaSupportprotectedstatic
emit_block_frag_type(std::ostream &os, codegen::LanguageExtension &lang_ext, const types::StorageType &storage_type, types::PrimitiveType element_type) constsdfg::gpu::cuda::CudaMmaSupport
emit_block_frag_type(std::ostream &os, codegen::LanguageExtension &lang_ext, MmaFragmentType type, const MmaBlockSize &dims, MmaFragmentLayout layout, types::PrimitiveType scalar_type, std::optional< std::pair< int, int > > coop_dims)sdfg::gpu::cuda::CudaMmaSupportstatic
fp64_supportsdfg::gpu::cuda::CudaMmaSupport
get_accumulator_type(types::PrimitiveType input_type, types::PrimitiveType output_type, types::PrimitiveType desired_acc_type) const overridesdfg::gpu::cuda::CudaMmaSupportvirtual
sdfg::gpu::GpuMmaSupport::get_accumulator_type(types::PrimitiveType input_type, types::PrimitiveType output_type) constsdfg::gpu::GpuMmaSupportinline
get_ceil_block_count(const symbolic::Expression &size, uint16_t block_size)sdfg::gpu::GpuMmaSupportstatic
get_integer_block_count(const symbolic::Expression &size, uint16_t block_size)sdfg::gpu::GpuMmaSupportstatic
get_mma_impl_type(const GpuMmaTiling &tiling) const overridesdfg::gpu::cuda::CudaMmaSupportvirtual
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 overridesdfg::gpu::cuda::CudaMmaSupportvirtual
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) constsdfg::gpu::GpuMmaSupportinlinevirtual
get_storage_type_arg_as_int(const types::StorageType &storage, int idx)sdfg::gpu::GpuMmaSupportprotectedstatic
get_supported_block_sizes(types::PrimitiveType input_type, types::PrimitiveType acc_type) const overridesdfg::gpu::cuda::CudaMmaSupportvirtual
GpuMmaSupport()=defaultsdfg::gpu::GpuMmaSupport
is_mma_type(const types::StorageType &storage)sdfg::gpu::cuda::CudaMmaSupportstatic
is_valid_block_size(const MmaBlockSize &block_size, types::PrimitiveType input_type, types::PrimitiveType acc_type) const overridesdfg::gpu::cuda::CudaMmaSupportvirtual
MMA_STORAGE_TYPEsdfg::gpu::cuda::CudaMmaSupportstatic
set_mma_fragment_storage_type(types::StorageType &storage_type, const GpuMmaTiling &size, MmaFragmentType type, MmaFragmentLayout layout) const overridesdfg::gpu::cuda::CudaMmaSupportvirtual
supported_types(types::PrimitiveType input_type, types::PrimitiveType output_type) const overridesdfg::gpu::cuda::CudaMmaSupportvirtual
tf32_supportsdfg::gpu::cuda::CudaMmaSupport
THREADS_PER_WAVEFRONTsdfg::gpu::cuda::CudaMmaSupportprotectedstatic
~GpuMmaSupport()=defaultsdfg::gpu::GpuMmaSupportvirtual