rocm_arch.h File Reference

#include <string>
#include <vector>
#include "sdfg/codegen/dispatchers/block_dispatcher.h"
#include "sdfg/structured_control_flow/structured_loop.h"
#include "sdfg/targets/gpu/gpu_arch.h"
#include "sdfg/types/array.h"

Go to the source code of this file.

Classes
struct	sdfg::gpu::rocm::RocmMmaSupport
class	sdfg::gpu::rocm::RocmArch
struct	sdfg::gpu::rocm::RocmDeviceInfo
	A ROCm/HIP GPU device as reported by rocminfo: its LLVM/ROCm ISA (gfx) name together with the marketing name(s) of the device(s) that report it and the wavefront size. More...

Namespaces
namespace	sdfg
namespace	sdfg::gpu
namespace	sdfg::gpu::rocm

Enumerations
enum class	sdfg::gpu::rocm::RocmMmaVariant { sdfg::gpu::rocm::Fragment = 1 , sdfg::gpu::rocm::Bare = 2 }

Functions
const std::vector< RocmDeviceInfo > &	sdfg::gpu::rocm::query_rocm_devices ()
	Queries the ROCm GPUs available on this machine via rocminfo.
const RocmArch *	sdfg::gpu::rocm::rocm_arch_for_device (const RocmDeviceInfo &device)
	Resolves a queried device to a RocmArch.
const RocmArch *	sdfg::gpu::rocm::rocm_arch_from_schedule_type (const structured_control_flow::ScheduleType &schedule)
	Wrapper, that allows us to store this inside the ScheduleType in the future.
const RocmArch *	sdfg::gpu::rocm::rocm_arch_from_env ()
	Use the DOCC_ROCM_ARCH env var or if unset, query the available cards.
const RocmArch *	sdfg::gpu::rocm::rocm_arch_from_available_hardware ()
const RocmArch *	sdfg::gpu::rocm::rocm_arch_parse (const std::string &name)

Variables
RocmArch	sdfg::gpu::rocm::ROCM_ARCH_GFX1201
RocmArch	sdfg::gpu::rocm::ROCM_ARCH_GFX90A
RocmArch	sdfg::gpu::rocm::ROCM_ARCH_GFX942