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 39 of file ShardOps.h.

Constructor & Destructor Documentation

Sharding() [1/2]

Sharding::Sharding

(

::mlir::FlatSymbolRefAttr

grid_ = nullptr

)

Definition at line 777 of file ShardOps.cpp.

Referenced by equalHaloAndShardSizes(), equalHaloSizes(), equalShardSizes(), equalSplitAxes(), get(), operator!=(), operator==(), and Sharding().

Sharding() [2/2]

Sharding::Sharding

(

Value

rhs

)

Definition at line 779 of file ShardOps.cpp.

References get(), and Sharding().

Member Function Documentation

equalHaloAndShardSizes()

bool Sharding::equalHaloAndShardSizes

(

const Sharding &

rhs

)

const

Definition at line 710 of file ShardOps.cpp.

References equalHaloSizes(), equalShardSizes(), and Sharding().

Referenced by operator==(), and operator==().

equalHaloSizes()

bool Sharding::equalHaloSizes

(

const Sharding &

rhs

)

const

Definition at line 730 of file ShardOps.cpp.

References getDynamicHaloSizes(), getStaticHaloSizes(), and Sharding().

Referenced by equalHaloAndShardSizes(), and mlir::shard::UpdateHaloPattern::tryApply().

equalShardSizes()

bool Sharding::equalShardSizes

(

const Sharding &

rhs

)

const

Definition at line 714 of file ShardOps.cpp.

References getDynamicShardedDimsOffsets(), getStaticShardedDimsOffsets(), and Sharding().

Referenced by equalHaloAndShardSizes().

equalSplitAxes()

bool Sharding::equalSplitAxes

(

const Sharding &

rhs

)

const

Definition at line 691 of file ShardOps.cpp.

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

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

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 795 of file ShardOps.cpp.

References mlir::clone(), mlir::detail::DenseArrayAttrImpl< int16_t >::get(), and Sharding().

Referenced by getSharding(), getSharding(), and Sharding().

getDynamicHaloSizes()

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

inline

Definition at line 64 of file ShardOps.h.

Referenced by equalHaloSizes(), and mlir::shard::UpdateHaloPattern::tryApply().

getDynamicShardedDimsOffsets()

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

inline

Definition at line 65 of file ShardOps.h.

Referenced by equalShardSizes().

getGrid()

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

inline

Definition at line 58 of file ShardOps.h.

Referenced by equalSplitAxes(), and mlir::shard::operator<<().

getGridAttr()

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

inline

Definition at line 57 of file ShardOps.h.

Referenced by mlir::shard::detail::defaultGetShardingOption().

getSplitAxes()

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

inline

Definition at line 59 of file ShardOps.h.

Referenced by mlir::shard::detail::defaultGetShardingOption(), equalSplitAxes(), mlir::shard::isFullReplication(), mlir::shard::operator<<(), mlir::shard::shardShapedType(), and mlir::shard::UpdateHaloPattern::tryApply().

getStaticHaloSizes()

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

inline

Definition at line 60 of file ShardOps.h.

Referenced by equalHaloSizes(), mlir::shard::ReshardingPattern::hasStaticOffsetsOrHalos(), mlir::shard::operator<<(), mlir::shard::shardShapedType(), and mlir::shard::UpdateHaloPattern::tryApply().

getStaticShardedDimsOffsets()

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

inline

Definition at line 61 of file ShardOps.h.

Referenced by equalShardSizes(), mlir::shard::ReshardingPattern::hasStaticOffsets(), mlir::shard::operator<<(), and mlir::shard::shardShapedType().

operator bool()

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

inline

Definition at line 68 of file ShardOps.h.

operator!=() [1/2]

bool Sharding::operator!=

(

const Sharding &

rhs

)

const

Definition at line 752 of file ShardOps.cpp.

References Sharding().

operator!=() [2/2]

bool Sharding::operator!=

(

Value

rhs

)

const

Definition at line 746 of file ShardOps.cpp.

operator==() [1/2]

bool Sharding::operator==

(

const Sharding &

rhs

)

const

Definition at line 748 of file ShardOps.cpp.

References equalHaloAndShardSizes(), equalSplitAxes(), and Sharding().

operator==() [2/2]

bool Sharding::operator==

(

Value

rhs

)

const

Definition at line 742 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