sdfg::gpu::rocm::GpuMmaTransform Class Reference

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

Inheritance diagram for sdfg::gpu::rocm::GpuMmaTransform:

Public Member Functions
	GpuMmaTransform (sdfg::math::tensor::MatMulNode &node, const sdfg::gpu::GpuArch *arch)
std::string	name () const override
	Get the name of this transformation.
bool	can_be_applied (sdfg::builder::StructuredSDFGBuilder &, sdfg::analysis::AnalysisManager &) override
	Check if this transformation can be safely applied.
void	apply (sdfg::builder::StructuredSDFGBuilder &builder, sdfg::analysis::AnalysisManager &) override
	Apply this transformation to the SDFG.
bool	expanded () const
void	to_json (nlohmann::json &j) const override
	Serialize this transformation to JSON.
Public Member Functions inherited from sdfg::transformations::Transformation
virtual	~Transformation ()=default
virtual void	set_report (PassReportConsumer *report)
	Set the report consumer for transformation feedback.
virtual bool	try_apply (builder::StructuredSDFGBuilder &builder, analysis::AnalysisManager &analysis_manager)
	Attempt to apply the transformation if possible.

Static Public Member Functions
static GpuMmaTransform	from_json (builder::StructuredSDFGBuilder &builder, const nlohmann::json &desc)

Additional Inherited Members
Protected Attributes inherited from sdfg::transformations::Transformation
PassReportConsumer *	report_ = nullptr

Constructor & Destructor Documentation

GpuMmaTransform()

sdfg::gpu::rocm::GpuMmaTransform::GpuMmaTransform	(	sdfg::math::tensor::MatMulNode &	node,
		const sdfg::gpu::GpuArch *	arch )

Member Function Documentation

apply()

void sdfg::gpu::rocm::GpuMmaTransform::apply ( sdfg::builder::StructuredSDFGBuilder & builder, sdfg::analysis::AnalysisManager & analysis_manager )

overridevirtual

Apply this transformation to the SDFG.

Parameters

builder	The SDFG builder
analysis_manager	The analysis manager (may be invalidated)

Implements sdfg::transformations::Transformation.

can_be_applied()

bool sdfg::gpu::rocm::GpuMmaTransform::can_be_applied ( sdfg::builder::StructuredSDFGBuilder & builder, sdfg::analysis::AnalysisManager & analysis_manager )

overridevirtual

Check if this transformation can be safely applied.

Parameters

builder	The SDFG builder
analysis_manager	The analysis manager providing necessary analyses

Returns

true if all preconditions are met

Implements sdfg::transformations::Transformation.

expanded()

bool sdfg::gpu::rocm::GpuMmaTransform::expanded

(

)

const

from_json()

GpuMmaTransform sdfg::gpu::rocm::GpuMmaTransform::from_json ( builder::StructuredSDFGBuilder & builder, const nlohmann::json & desc )

static

name()

std::string sdfg::gpu::rocm::GpuMmaTransform::name ( ) const

overridevirtual

Get the name of this transformation.

Returns

The transformation name (e.g., "LoopInterchange")

Implements sdfg::transformations::Transformation.

to_json()

void sdfg::gpu::rocm::GpuMmaTransform::to_json ( nlohmann::json & j ) const

overridevirtual

Serialize this transformation to JSON.

Parameters

j	JSON object to populate with transformation description

Exceptions

std::logic_error

if not implemented

Implements sdfg::transformations::Transformation.

---

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

• opt/include/sdfg/targets/rocm/rocm_mma_transform.h