OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-07 12:21:27 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity

Add GlowLoadAOTModel flag (#45189)

Browse Source 
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/45189

Pull Request resolved: https://github.com/pytorch/glow/pull/4902

Test Plan: Test locally

Reviewed By: yinghai

Differential Revision: D23810445

fbshipit-source-id: 56e717d80abbfe76b15d0f4249e1e399a9722753
This commit is contained in:
Jordan Fix 2020-09-23 20:47:51 -07:00 committed by Facebook GitHub Bot
parent 60665ace17
commit c760bc8fb1
2 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
caffe2/opt/onnxifi_op.h
View File

@ -263,10 +263,13 @@ class OnnxifiOp final : public Operator<Context> {
defered_blob_reader = ws->GetBlob("__DEFERRED_BLOB_READER__");
}
onnxGraph graph{nullptr};
static const uint64_t auxPropertiesListAOT[] = {
ONNXIFI_OPTIMIZATION_AOT, ONNXIFI_GRAPH_PROPERTY_NONE};
CAFFE_ENFORCE_EQ(
lib_->onnxInitGraph(
backend,
nullptr,
use_glow_aot_ ? auxPropertiesListAOT : nullptr,
onnx_model_str.size(),
(const void*)(onnx_model_str.c_str()),
weight_descs.size(),

2
third_party/foxi vendored

@ -1 +1 @@
Subproject commit 9ca418d2f4bc8e022d843388afa0fd0a14bd57dc
Subproject commit 4aba696ec8f31794fd42880346dc586486205e0a