ZigZag - Deep Learning Hardware Design Space Exploration
This repository presents the novel version of our tried-and-tested hardware Architecture-Mapping Design Space Exploration (DSE) Framework for Deep Learning (DL) accelerators. ZigZag bridges the gap between algorithmic DL decisions and their acceleration cost on specialized accelerators through a fast and accurate hardware cost estimation.
|
Suggested LayerDims to be unrolled for every OADimension. More...
Public Member Functions | |
def | __init__ (self, dict[OADimension, set[LayerDim]] data) |
def | clear_invalid_hits (self, list[LayerDim] valid_layer_dims) |
Check the hints at all contained OADimension. More... | |
def | complete_with_defaults (self, dict[OADimension, int] oa_dim_sizes, list[LayerDim] layer_dims) |
For all OADimensions in oa_dim_sizes that are not already in this SpatialMappingHint, fill the hints for that OADimension with all LayerDims from layer_dims . More... | |
def | __getitem__ (self, OADimension key) |
![]() | |
def | __init__ (self, Any data) |
int | __len__ (self) |
Iterator[Any] | __iter__ (self) |
def | __getitem__ (self, Any key) |
bool | __contains__ (self, Any key) |
def | __str__ (self) |
def | __repr__ (self) |
Any | __jsonrepr__ (self) |
def | __eq__ (self, object other) |
def | __hash__ (self) |
Static Public Member Functions | |
"SpatialMappingHint" | empty () |
Public Attributes | |
data | |
![]() | |
data | |
Suggested LayerDims to be unrolled for every OADimension.
def __init__ | ( | self, | |
dict[OADimension, set[LayerDim]] | data | ||
) |
def __getitem__ | ( | self, | |
OADimension | key | ||
) |
def clear_invalid_hits | ( | self, | |
list[LayerDim] | valid_layer_dims | ||
) |
Check the hints at all contained OADimension.
If the OADimension doesn't contain a single LayerDim that is also present in the given list valid_layer_dims
, remove the OADimension from this instance.
def complete_with_defaults | ( | self, | |
dict[OADimension, int] | oa_dim_sizes, | ||
list[LayerDim] | layer_dims | ||
) |
For all OADimensions in oa_dim_sizes
that are not already in this SpatialMappingHint, fill the hints for that OADimension with all LayerDims from layer_dims
.
|
static |
data |