This commit is contained in:
+1
@@ -65,6 +65,7 @@ std::optional<ProducerValueRef> getProducerValueRef(Value value) {
|
||||
if (!op)
|
||||
return std::nullopt;
|
||||
|
||||
//TODO Extract Slice is not the only global non compute operation. There are other legal op
|
||||
while (auto extract = dyn_cast<tensor::ExtractSliceOp>(op)) {
|
||||
value = extract.getSource();
|
||||
op = value.getDefiningOp();
|
||||
|
||||
Reference in New Issue
Block a user