tvm::relax::distributed::transform Namespace Reference

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

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

Typedef Documentation

DataflowBlock

using tvm::relax::distributed::transform::DataflowBlock = typedef tvm::relax::DataflowBlock

Function

using tvm::relax::distributed::transform::Function = typedef tvm::relax::Function

Pass

using tvm::relax::distributed::transform::Pass = typedef tvm::transform::Pass

PassContext

using tvm::relax::distributed::transform::PassContext = typedef tvm::transform::PassContext

PassInfo

using tvm::relax::distributed::transform::PassInfo = typedef tvm::transform::PassInfo

Function Documentation

LegalizeRedistribute()

Pass tvm::relax::distributed::transform::LegalizeRedistribute

(

)

Legalize redistribute op to ccl op.

Returns

The Pass.

LowerDistIR()

Pass tvm::relax::distributed::transform::LowerDistIR

(

)

Lower DistIR to Relax.

Returns

The Pass.

LowerGlobalViewToLocalView()

Pass tvm::relax::distributed::transform::LowerGlobalViewToLocalView

(

)

Lower global view TensorIR into local view.

Returns

The Pass.

PropagateSharding()

Pass tvm::relax::distributed::transform::PropagateSharding

(

)

Propagate sharding information.

Returns

The Pass.