|
docc 0.8.3 (0d44999b)
C++ API Reference
|
This is the complete list of members for sdfg::gpu::cuda::CudaMmaSupport, including all inherited members.
| base_support | sdfg::gpu::cuda::CudaMmaSupport | |
| CudaMmaSupport(bool base_support, bool tf32_support, bool fp64_support) | sdfg::gpu::cuda::CudaMmaSupport | inline |
| DEFAULT_BLOCK_SIZE | sdfg::gpu::cuda::CudaMmaSupport | protectedstatic |
| emit_block_frag_type(std::ostream &os, codegen::LanguageExtension &lang_ext, const types::StorageType &storage_type, types::PrimitiveType element_type) const | sdfg::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::CudaMmaSupport | static |
| fp64_support | sdfg::gpu::cuda::CudaMmaSupport | |
| get_accumulator_type(types::PrimitiveType input_type, types::PrimitiveType output_type, types::PrimitiveType desired_acc_type) const override | sdfg::gpu::cuda::CudaMmaSupport | 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::cuda::CudaMmaSupport | 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::cuda::CudaMmaSupport | 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 | sdfg::gpu::GpuMmaSupport | inlinevirtual |
| 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::cuda::CudaMmaSupport | virtual |
| GpuMmaSupport()=default | sdfg::gpu::GpuMmaSupport | |
| is_mma_type(const types::StorageType &storage) | sdfg::gpu::cuda::CudaMmaSupport | static |
| is_valid_block_size(const MmaBlockSize &block_size, types::PrimitiveType input_type, types::PrimitiveType acc_type) const override | sdfg::gpu::cuda::CudaMmaSupport | virtual |
| MMA_STORAGE_TYPE | sdfg::gpu::cuda::CudaMmaSupport | static |
| set_mma_fragment_storage_type(types::StorageType &storage_type, const GpuMmaTiling &size, MmaFragmentType type, MmaFragmentLayout layout) const override | sdfg::gpu::cuda::CudaMmaSupport | virtual |
| supported_types(types::PrimitiveType input_type, types::PrimitiveType output_type) const override | sdfg::gpu::cuda::CudaMmaSupport | virtual |
| tf32_support | sdfg::gpu::cuda::CudaMmaSupport | |
| THREADS_PER_WAVEFRONT | sdfg::gpu::cuda::CudaMmaSupport | protectedstatic |
| ~GpuMmaSupport()=default | sdfg::gpu::GpuMmaSupport | virtual |