gpu_mma_fragment.h File Reference

#include <nlohmann/json_fwd.hpp>
#include <ostream>
#include <string>
#include "sdfg/data_flow/library_node.h"
#include "sdfg/symbolic/symbolic.h"
#include "sdfg/types/type.h"

Go to the source code of this file.

Classes
struct	sdfg::gpu::MmaBlockSize
struct	sdfg::gpu::BlockCounts
	Grouping of MMA blocks. More...
struct	sdfg::gpu::GpuMmaTiling
struct	sdfg::gpu::GpuMmaSupport
struct	sdfg::gpu::GpuMmaFromMemoryLayout
	Describes how a single MMA fragment maps to memory relative to a base pointer. More...

Namespaces
namespace	sdfg
namespace	sdfg::gpu

Enumerations
enum class	sdfg::gpu::MmaFragmentType { sdfg::gpu::A = 1 , sdfg::gpu::B = 2 , sdfg::gpu::C = 3 }
enum	sdfg::gpu::MmaFragmentLayout { sdfg::gpu::MMA_LAYOUT_UNSPECIFIED = 0 , sdfg::gpu::MMA_LAYOUT_ROW_MAJOR = 1 , sdfg::gpu::MMA_LAYOUT_COL_MAJOR = 2 }

Functions
std::ostream &	sdfg::gpu::operator<< (std::ostream &os, const MmaBlockSize &block_size)
constexpr const char *	sdfg::gpu::mma_fragment_layout_to_string (MmaFragmentLayout layout)
std::ostream &	sdfg::gpu::operator<< (std::ostream &os, const GpuMmaFromMemoryLayout &layout)
void	sdfg::gpu::serialize_mma_block_size (nlohmann::json &j, const MmaBlockSize &block_size)
MmaBlockSize	sdfg::gpu::deserialize_mma_block_size (const nlohmann::json &j)
void	sdfg::gpu::serialize_mma_from_memory_layout (nlohmann::json &j, const GpuMmaFromMemoryLayout &layout)
GpuMmaFromMemoryLayout	sdfg::gpu::deserialize_mma_from_memory_layout (const nlohmann::json &j)