utils.h File Reference

#include <tvm/s_tir/sblock_scope.h>
#include <tvm/tirx/stmt.h>
#include <unordered_map>

Include dependency graph for utils.h:

Go to the source code of this file.

Namespaces
	tvm
	An object that builds and maintains block scope and StmtSref mapping for Dependence analysis.
	tvm::tirx

Macros
#define	TVM_SREF_AS_OR_ERR(Result, SRef, Type)
	A helper macro to convert an sref to the statement it points to, then check if the downcasting succeeded. More...
#define	TVM_SREF_TO_SBLOCK(SRef)
	A helper macro to convert an sref to the block it points to,. More...
#define	TVM_SREF_TO_FOR(SRef)
	A helper macro to convert an sref to the for-loop it points to. More...
#define	TVM_TYPE_AS_OR_ERR(Result, From, Type)
	Downcast a TVM ffi::ObjectRef to its corresponding container using ffi::ObjectRef::as<Type>, then check if the downcasting succeeded. More...
#define	TVM_TYPE_AS(From, Type)
	Downcast a TVM ffi::ObjectRef to its corresponding container using ffi::ObjectRef::as<Type>, throwing an internal error if downcast fails. More...

Functions
void	tvm::tirx::SetSeqIndex (std::unordered_map< const StmtNode *, StmtSRef > &stmt2ref, const Stmt &stmt, int seq_index, bool include_loops=true)
	Set the StmtSRefNode::seq_index field for stmt. More...
void	tvm::tirx::SetSeqIndexInChildren (std::unordered_map< const StmtNode *, StmtSRef > &stmt2ref, const SeqStmtNode *seq_stmt, bool include_loops=true)
	Update seq_index of the children of a SeqStmt. More...

Macro Definition Documentation

TVM_SREF_AS_OR_ERR

#define TVM_SREF_AS_OR_ERR	(		Result,
			SRef,
			Type
	)

Value:

  SRef->StmtAs<Type>();                        \
  TVM_FFI_CHECK(Result, TypeError)

A helper macro to convert an sref to the statement it points to, then check if the downcasting succeeded.

Parameters

Result	The result variable, used for checking
SRef	The SRef to be cast
Type	The type to be cast to, can be SBlock or For

TVM_SREF_TO_FOR

#define TVM_SREF_TO_FOR

(

SRef

)

Value:

  [&]() {                                                                              \
    auto result = TVM_SREF_AS_OR_ERR(result, (SRef), ::tvm::tirx::ForNode)             \
                  << "Expects StmtSRef `" << #SRef << "` points to `Loop`, but gets: " \
                  << ((SRef)->stmt ? (SRef)->stmt->GetTypeKey() : "None");             \
    return result;                                                                     \
  }()

A helper macro to convert an sref to the for-loop it points to.

Throws an internal error if downcasting fails. The variable name in the parent scope is used for the error message.

Parameters

SRef	The SRef to be cast

TVM_SREF_TO_SBLOCK

#define TVM_SREF_TO_SBLOCK

(

SRef

)

Value:

  [&]() {                                                                               \
    auto result = TVM_SREF_AS_OR_ERR(result, (SRef), ::tvm::tirx::SBlockNode)           \
                  << "Expects StmtSRef `" << #SRef << "` points to `Block`, but gets: " \
                  << ((SRef)->stmt ? (SRef)->stmt->GetTypeKey() : "None");              \
    return result;                                                                      \
  }()

A helper macro to convert an sref to the block it points to,.

Throws an internal error if downcasting fails. The variable name in the parent scope is used for the error message.

Parameters

SRef	The SRef to be cast

TVM_TYPE_AS

#define TVM_TYPE_AS	(		From,
			Type
	)

Value:

  [&]() {                                                                          \
    auto result = TVM_TYPE_AS_OR_ERR(result, (From), Type)                         \
                  << "Expects `" << #From << "` to have type `" << Type::_type_key \
                  << "`, but gets: " << ((From).GetTypeKey());                     \
    return result;                                                                 \
  }()

Downcast a TVM ffi::ObjectRef to its corresponding container using ffi::ObjectRef::as<Type>, throwing an internal error if downcast fails.

Parameters

From	The ffi::ObjectRef to be downcast
Type	The type to be downcast to

TVM_TYPE_AS_OR_ERR

#define TVM_TYPE_AS_OR_ERR	(		Result,
			From,
			Type
	)

Value:

  From.as<Type>();                             \
  TVM_FFI_CHECK(Result, TypeError)

Downcast a TVM ffi::ObjectRef to its corresponding container using ffi::ObjectRef::as<Type>, then check if the downcasting succeeded.

Parameters

Result	The result variable, used for checking
From	The ffi::ObjectRef to be downcast
Type	The type to be downcast to