transform.h File Reference

Relax distributed specific transformation passes. More...

#include <tvm/ir/transform.h>
#include <tvm/relax/dataflow_pattern.h>
#include <tvm/relax/expr.h>
#include <tvm/relax/transform.h>
#include <tvm/tirx/function.h>
#include <tvm/tirx/index_map.h>

Include dependency graph for transform.h:

Go to the source code of this file.

Namespaces
namespace	tvm
	An object that builds and maintains block scope and StmtSref mapping for Dependence analysis.
namespace	tvm::relax
namespace	tvm::relax::distributed
namespace	tvm::relax::distributed::transform

Typedefs
using	tvm::relax::distributed::transform::Pass = tvm::transform::Pass
using	tvm::relax::distributed::transform::PassInfo = tvm::transform::PassInfo
using	tvm::relax::distributed::transform::PassContext = tvm::transform::PassContext
using	tvm::relax::distributed::transform::Function = tvm::relax::Function
using	tvm::relax::distributed::transform::DataflowBlock = tvm::relax::DataflowBlock

Functions
Pass	tvm::relax::distributed::transform::PropagateSharding ()
	Propagate sharding information.
Pass	tvm::relax::distributed::transform::LowerGlobalViewToLocalView ()
	Lower global view TensorIR into local view.
Pass	tvm::relax::distributed::transform::LegalizeRedistribute ()
	Legalize redistribute op to ccl op.
Pass	tvm::relax::distributed::transform::LowerDistIR ()
	Lower DistIR to Relax.

Detailed Description

Relax distributed specific transformation passes.