Skip to content
Snippets Groups Projects
Unverified Commit 8dcb821c authored by Yaman Umuroglu's avatar Yaman Umuroglu Committed by GitHub
Browse files

[LabelSel] assertion to check no folding for now

parent 33874f17
No related branches found
No related tags found
No related merge requests found
......@@ -69,6 +69,7 @@ class LabelSelect_Batch(HLSCustomOp):
pe = self.get_nodeattr("PE")
vecs = list(self.get_nodeattr("numInputVectors"))
assert nlabels % pe == 0, "PE must divide Labels"
assert nlabels == pe, "LabelSelect currently fails with folding"
folds = int(nlabels / pe)
folded_ishape = tuple(vecs + [folds, pe])
return folded_ishape
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment