mlir::shard::Sharding Class Reference

#include "mlir/Dialect/Shard/IR/ShardOps.h"

Public Member Functions
	Sharding (::mlir::FlatSymbolRefAttr grid_=nullptr)
	Sharding (Value rhs)
::mlir::FlatSymbolRefAttr	getGridAttr () const
::llvm::StringRef	getGrid () const
ArrayRef< GridAxesAttr >	getSplitAxes () const
ArrayRef< int64_t >	getStaticHaloSizes () const
ArrayRef< int64_t >	getStaticShardedDimsOffsets () const
ArrayRef< Value >	getDynamicHaloSizes () const
ArrayRef< Value >	getDynamicShardedDimsOffsets () const
	operator bool () const
bool	operator== (Value rhs) const
bool	operator!= (Value rhs) const
bool	operator== (const Sharding &rhs) const
bool	operator!= (const Sharding &rhs) const
bool	equalSplitAxes (const Sharding &rhs) const
bool	equalHaloAndShardSizes (const Sharding &rhs) const
bool	equalHaloSizes (const Sharding &rhs) const
bool	equalShardSizes (const Sharding &rhs) const

Static Public Member Functions
static Sharding	get (::mlir::FlatSymbolRefAttr grid_, ArrayRef< GridAxesAttr > split_axes_, ArrayRef< int64_t > static_halo_sizes_={}, ArrayRef< int64_t > static_sharded_dims_offsets_={}, ArrayRef< Value > dynamic_halo_sizes_={}, ArrayRef< Value > dynamic_sharded_dims_offsets_={})

Detailed Description

Definition at line 42 of file ShardOps.h.

Constructor & Destructor Documentation

Sharding() [1/2]

Sharding::Sharding

(

::mlir::FlatSymbolRefAttr

grid_ = nullptr

)

Definition at line 752 of file ShardOps.cpp.

Referenced by Sharding().

Sharding() [2/2]

Sharding::Sharding

(

Value

rhs

)

Definition at line 754 of file ShardOps.cpp.

References get(), mlir::Value::getDefiningOp(), and Sharding().

Member Function Documentation

equalHaloAndShardSizes()

bool Sharding::equalHaloAndShardSizes

(

const Sharding &

rhs

)

const

Definition at line 708 of file ShardOps.cpp.

References equalHaloSizes(), and equalShardSizes().

Referenced by operator==().

equalHaloSizes()

bool Sharding::equalHaloSizes

(

const Sharding &

rhs

)

const

Definition at line 728 of file ShardOps.cpp.

References getDynamicHaloSizes(), and getStaticHaloSizes().

Referenced by equalHaloAndShardSizes(), and mlir::shard::tryUpdateHaloInResharding().

equalShardSizes()

bool Sharding::equalShardSizes

(

const Sharding &

rhs

)

const

Definition at line 712 of file ShardOps.cpp.

References getDynamicShardedDimsOffsets(), and getStaticShardedDimsOffsets().

Referenced by equalHaloAndShardSizes().

equalSplitAxes()

bool Sharding::equalSplitAxes

(

const Sharding &

rhs

)

const

Definition at line 689 of file ShardOps.cpp.

References getGrid(), getSplitAxes(), and min().

Referenced by operator==(), and mlir::shard::tryUpdateHaloInResharding().

get()

Sharding Sharding::get ( ::mlir::FlatSymbolRefAttr  grid_, ArrayRef< GridAxesAttr >  split_axes_, ArrayRef< int64_t >  static_halo_sizes_ = {}, ArrayRef< int64_t >  static_sharded_dims_offsets_ = {}, ArrayRef< Value >  dynamic_halo_sizes_ = {}, ArrayRef< Value >  dynamic_sharded_dims_offsets_ = {}  )

static

Definition at line 770 of file ShardOps.cpp.

References mlir::clone(), copy(), mlir::detail::enumerate(), and mlir::detail::DenseArrayAttrImpl< T >::get().

Referenced by getSharding(), Sharding(), mlir::shard::targetShardingInMoveLastAxis(), mlir::shard::targetShardingInSplitLastAxis(), and mlir::shard::targetShardingInUnsplitLastAxis().

getDynamicHaloSizes()

ArrayRef<Value> mlir::shard::Sharding::getDynamicHaloSizes ( ) const

inline

Definition at line 67 of file ShardOps.h.

Referenced by equalHaloSizes(), and mlir::shard::tryUpdateHaloInResharding().

getDynamicShardedDimsOffsets()

ArrayRef<Value> mlir::shard::Sharding::getDynamicShardedDimsOffsets ( ) const

inline

Definition at line 68 of file ShardOps.h.

Referenced by equalShardSizes().

getGrid()

::llvm::StringRef mlir::shard::Sharding::getGrid ( ) const

inline

Definition at line 61 of file ShardOps.h.

References mlir::FlatSymbolRefAttr::getValue().

Referenced by equalSplitAxes().

getGridAttr()

::mlir::FlatSymbolRefAttr mlir::shard::Sharding::getGridAttr ( ) const

inline

Definition at line 60 of file ShardOps.h.

Referenced by mlir::shard::detail::defaultGetShardingOption(), mlir::shard::targetShardingInMoveLastAxis(), mlir::shard::targetShardingInSplitLastAxis(), and mlir::shard::targetShardingInUnsplitLastAxis().

getSplitAxes()

ArrayRef<GridAxesAttr> mlir::shard::Sharding::getSplitAxes ( ) const

inline

Definition at line 62 of file ShardOps.h.

Referenced by mlir::shard::detail::defaultGetShardingOption(), mlir::shard::detectMoveLastSplitAxisInResharding(), mlir::shard::detectSplitLastAxisInResharding(), mlir::shard::detectUnsplitLastAxisInResharding(), equalSplitAxes(), mlir::shard::isFullReplication(), mlir::shard::shardShapedType(), mlir::shard::targetShardingInMoveLastAxis(), mlir::shard::targetShardingInSplitLastAxis(), mlir::shard::targetShardingInUnsplitLastAxis(), and mlir::shard::tryUpdateHaloInResharding().

getStaticHaloSizes()

ArrayRef<int64_t> mlir::shard::Sharding::getStaticHaloSizes ( ) const

inline

Definition at line 63 of file ShardOps.h.

Referenced by equalHaloSizes(), mlir::shard::reshardOn1DGrid(), mlir::shard::shardShapedType(), and mlir::shard::tryUpdateHaloInResharding().

getStaticShardedDimsOffsets()

ArrayRef<int64_t> mlir::shard::Sharding::getStaticShardedDimsOffsets ( ) const

inline

Definition at line 64 of file ShardOps.h.

Referenced by equalShardSizes(), mlir::shard::reshardOn1DGrid(), mlir::shard::shardShapedType(), and mlir::shard::tryUpdateHaloInResharding().

operator bool()

mlir::shard::Sharding::operator bool ( ) const

inline

Definition at line 71 of file ShardOps.h.

operator!=() [1/2]

bool Sharding::operator!=

(

const Sharding &

rhs

)

const

Definition at line 750 of file ShardOps.cpp.

operator!=() [2/2]

bool Sharding::operator!=

(

Value

rhs

)

const

Definition at line 744 of file ShardOps.cpp.

operator==() [1/2]

bool Sharding::operator==

(

const Sharding &

rhs

)

const

Definition at line 746 of file ShardOps.cpp.

References equalHaloAndShardSizes(), and equalSplitAxes().

operator==() [2/2]

bool Sharding::operator==

(

Value

rhs

)

const

Definition at line 740 of file ShardOps.cpp.

References equalHaloAndShardSizes(), and equalSplitAxes().

---

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

• include/mlir/Dialect/Shard/IR/ShardOps.h
• lib/Dialect/Shard/IR/ShardOps.cpp