saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-29 20:14:01 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

[Nuphar] Do not handle MatMulInteger with zero-points (#8760)

Browse source 
MatMulInteger can take zero-points as input, and Nuphar
does not handle that yet. Fall back to CPU EP in that case.
This commit is contained in:
KeDengMS 2021-08-18 10:32:42 -07:00 committed by GitHub
parent 00b345eb7b
commit b0c707caa8
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
onnxruntime/core/providers/nuphar/nuphar_execution_provider.cc
View file

@ -321,6 +321,9 @@ NupharExecutionProvider::GetCapability(const onnxruntime::GraphViewer& graph_vie
return false;
}
}
if (node.OpType() == "MatMulInteger" && inputs.size() > 2) {
return false; // do not support MatMulInteger with zero points
}
return true;
};
GraphPartitioner graph_partitioner(is_supported_func);