We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent e106d09 commit 5d246e8Copy full SHA for 5d246e8
src/qonnx/transformation/create_generic_partitions.py
@@ -202,7 +202,7 @@ def partitioning_func(node):
202
if len(candidates) == 0:
203
return -1
204
assert len(candidates) == 1, f"single node assigned to multiple partitions: {candidates}"
205
- return candidates[0][0] # partition_id
+ return candidates[0][0] # partition_id
206
207
# apply partitioning
208
model = model.transform(PartitionFromLambda(partitioning_func, self.partition_dir))
0 commit comments