saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-05-18 21:21:17 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Remove InsertFusedRules from CPUExecutionProvider (#573)

Browse source 
Talked with the original author @souptc , this function is not needed anymore.
This commit is contained in:
Changming Sun 2019-03-11 10:45:44 -07:00 committed by GitHub
parent 867eda5262
commit b183df7143
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 0 additions and 22 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
onnxruntime/core/providers/cpu/cpu_execution_provider.cc
View file

@ -644,20 +644,4 @@ std::shared_ptr<KernelRegistry> CPUExecutionProvider::GetKernelRegistry() const
static std::shared_ptr<KernelRegistry> kernel_registry = GetCpuKernelRegistry();
return kernel_registry;
}
std::vector<std::unique_ptr<ComputeCapability>>
CPUExecutionProvider::GetCapability(const onnxruntime::GraphViewer& graph,
const std::vector<const KernelRegistry*>& kernel_registries) const {
std::vector<std::unique_ptr<ComputeCapability>>
result = IExecutionProvider::GetCapability(graph, kernel_registries);
for (auto& rule : fuse_rules_) {
rule(graph, result);
}
return result;
}
void CPUExecutionProvider::InsertFusedRules(FuseRuleFn rule) {
fuse_rules_.push_back(rule);
}
} // namespace onnxruntime

6
onnxruntime/core/providers/cpu/cpu_execution_provider.h
View file

@ -46,11 +46,6 @@ class CPUExecutionProvider : public IExecutionProvider {
#endif
}
std::vector<std::unique_ptr<ComputeCapability>> GetCapability(
const onnxruntime::GraphViewer& graph,
const std::vector<const KernelRegistry*>& kernel_registries) const override;
///requires src.buffer_deleter_ == nullptr
Status CopyTensor(const Tensor&, Tensor&) const override {
return Status(common::ONNXRUNTIME, common::FAIL, "Shouldn't reach here. CPUExecutionProvider doesn't support CopyTensor");
}
@ -62,7 +57,6 @@ class CPUExecutionProvider : public IExecutionProvider {
std::shared_ptr<KernelRegistry> GetKernelRegistry() const override;
void InsertFusedRules(FuseRuleFn rule);
private:
std::vector<FuseRuleFn> fuse_rules_;