Skip to content

Commit 92ad1d2

Browse files
Update README, run Black
1 parent 4ad2ee3 commit 92ad1d2

3 files changed

Lines changed: 9 additions & 7 deletions

File tree

README.md

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,8 @@
11
# AccelForge
22

33
AccelForge is a framework to model and design tensor algebra accelerators. To learn
4-
more, see the [AccelForge website](https://accelergy-project.github.io/accelforge/).
5-
6-
**Source Code**: The AccelForge source code is available on
4+
more, see the [AccelForge website](https://accelergy-project.github.io/accelforge/). The
5+
AccelForge source code is available on
76
[GitHub](https://github.com/Accelergy-Project/accelforge).
87

98
AccelForge uses [HWComponents](https://github.com/accelergy-project/hwcomponents)

accelforge/mapper/FFM/_make_pmappings/make_pmappings_from_templates/make_tile_shapes.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -376,6 +376,7 @@ def try_replace_single_term(
376376
):
377377
return _try_replace_single_term(t, symbols_enumerated & t.free_symbols, bounds)
378378

379+
379380
@lru_cache(maxsize=10000)
380381
def _partition_formula(
381382
f: Expr,
@@ -1202,7 +1203,9 @@ def update_symbol2goal(
12021203
except (TypeError, ValueError):
12031204
pass
12041205

1205-
log_message("Partitioning formula", f"{objective.name}: {objective.formula}")
1206+
log_message(
1207+
"Partitioning formula", f"{objective.name}: {objective.formula}"
1208+
)
12061209
goals = partition_formula(
12071210
objective.formula, sym_enumerated_set, what_tiles_symbol.bounds
12081211
)

accelforge/mapper/FFM/mappings.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -89,9 +89,9 @@ def per_tensor_size(self, return_n_elements: bool = False) -> dict[TensorName, i
8989
dict[TensorName, int]
9090
A dictionary of: {Tensor name: Number of bits} for each tensor in the spec.
9191
"""
92-
assert self.evaluated_specs is not None, (
93-
"Can't get per-tensor size if no Einsums have been mapped."
94-
)
92+
assert (
93+
self.evaluated_specs is not None
94+
), "Can't get per-tensor size if no Einsums have been mapped."
9595
return get_per_tensor_size(
9696
next(iter(self.evaluated_specs.values())).workload,
9797
return_n_elements=return_n_elements,

0 commit comments

Comments
 (0)