tvm
Public Member Functions | Public Attributes | List of all members
tvm::tirx::ActiveSet Struct Reference

Active thread set A. The source of truth is layout: shard = active axes with extents offset = per-axis lower bound, possibly a selector PrimExpr. More...

#include <exec_context.h>

Collaboration diagram for tvm::tirx::ActiveSet:

Public Member Functions

int64_t size () const
 
bool GetAxis (const std::string &axis, AxisRange *out) const
 
bool HasAxis (const std::string &axis) const
 
ActiveSet WithAxis (const std::string &axis, const AxisRange &range) const
 
std::vector< std::string > AxisNames () const
 

Public Attributes

TileLayout layout
 

Detailed Description

Active thread set A. The source of truth is layout: shard = active axes with extents offset = per-axis lower bound, possibly a selector PrimExpr.

Member Function Documentation

◆ AxisNames()

std::vector<std::string> tvm::tirx::ActiveSet::AxisNames ( ) const

◆ GetAxis()

bool tvm::tirx::ActiveSet::GetAxis ( const std::string &  axis,
AxisRange out 
) const

◆ HasAxis()

bool tvm::tirx::ActiveSet::HasAxis ( const std::string &  axis) const

◆ size()

int64_t tvm::tirx::ActiveSet::size ( ) const

◆ WithAxis()

ActiveSet tvm::tirx::ActiveSet::WithAxis ( const std::string &  axis,
const AxisRange range 
) const

Member Data Documentation

◆ layout

TileLayout tvm::tirx::ActiveSet::layout

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