Static resolver for ScopeIdDef values. Replaces the former ScopeIdResolveTable runtime registry with a closed-enum switch. More...

#include <exec_scope.h>

Collaboration diagram for tvm::tirx::ScopeIdResolve:

Public Types
using	LaunchParams = std::unordered_map< ffi::String, IterVar >

Static Public Member Functions
static ffi::Array< PrimExpr >	Resolve (ScopeBinding binding, const ffi::Optional< ffi::Array< PrimExpr >> &extents, int out_dim, const ffi::String &target_kind, const LaunchParams &params)
	Resolve a ScopeIdDef for a given canonical binding + target. More...

static PrimExpr	ComputeWarpIdInCta (const LaunchParams &params)
	Compute the warp_id_in_cta shuffle expression from threadIdx in launch params. More...

Detailed Description

Static resolver for ScopeIdDef values. Replaces the former ScopeIdResolveTable runtime registry with a closed-enum switch.

Member Typedef Documentation

static PrimExpr tvm::tirx::ScopeIdResolve::ComputeWarpIdInCta ( const LaunchParams & params )

static

Compute the warp_id_in_cta shuffle expression from threadIdx in launch params.

static ffi::Array<PrimExpr> tvm::tirx::ScopeIdResolve::Resolve	(	ScopeBinding	binding,
		const ffi::Optional< ffi::Array< PrimExpr >> &	extents,
		int	out_dim,
		const ffi::String &	target_kind,
		const LaunchParams &	params
	)

static

Resolve a ScopeIdDef for a given canonical binding + target.

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