docc 0.8.3 (0d44999b)
C++ API Reference
Loading...
Searching...
No Matches
sdfg::gpu::rocm::RocmArch Class Reference

#include <sdfg/targets/rocm/rocm_arch.h>

Inheritance diagram for sdfg::gpu::rocm::RocmArch:
[legend]

Public Member Functions

 RocmArch (const std::string &name, int per_cu_threads, bool mma_base_support, bool mma_f32_support, uint16_t mma_int8_block_k=0)
std::string unique_id () const override
std::string name () const override
int per_cu_threads () const override
const RocmMmaSupport * mma_support () const override
structured_control_flow::ScheduleType create_schedule_type () const override
Public Member Functions inherited from sdfg::gpu::GpuArch
 GpuArch ()
virtual ~GpuArch ()=default

Additional Inherited Members

Static Public Member Functions inherited from sdfg::gpu::GpuArch
static const GpuArch * get_from_schedule_type (const structured_control_flow::ScheduleType &schedule)
static const GpuArch * parse_from_name (const std::string &raw_name)
Static Public Attributes inherited from sdfg::gpu::GpuArch
static constexpr const char * ARCH_PROPERTY = "ARCH"

Constructor & Destructor Documentation

◆ RocmArch()

sdfg::gpu::rocm::RocmArch::RocmArch ( const std::string & name,
int per_cu_threads,
bool mma_base_support,
bool mma_f32_support,
uint16_t mma_int8_block_k = 0 )
inline

Member Function Documentation

◆ create_schedule_type()

structured_control_flow::ScheduleType sdfg::gpu::rocm::RocmArch::create_schedule_type ( ) const
overridevirtual

Implements sdfg::gpu::GpuArch.

◆ mma_support()

const RocmMmaSupport * sdfg::gpu::rocm::RocmArch::mma_support ( ) const
inlineoverridevirtual

Implements sdfg::gpu::GpuArch.

◆ name()

std::string sdfg::gpu::rocm::RocmArch::name ( ) const
overridevirtual

Implements sdfg::gpu::GpuArch.

◆ per_cu_threads()

int sdfg::gpu::rocm::RocmArch::per_cu_threads ( ) const
inlineoverridevirtual

Implements sdfg::gpu::GpuArch.

◆ unique_id()

std::string sdfg::gpu::rocm::RocmArch::unique_id ( ) const
overridevirtual

Implements sdfg::gpu::GpuArch.


The documentation for this class was generated from the following file: