Skip to content

Commit 2141cb7

Browse files
Marat Dukhanfacebook-github-bot
Marat Dukhan
authored andcommitted
Update OnnxifiOp to reflect onnx/onnx#1256
Summary: Pull Request resolved: #10230 Reviewed By: yinghai Differential Revision: D9174527 Pulled By: Maratyszcza fbshipit-source-id: 753493e67446b528d65b146e89ea9f874b469ead
1 parent 5df8547 commit 2141cb7

File tree

2 files changed

+2
-1
lines changed

2 files changed

+2
-1
lines changed

caffe2/operators/onnxifi_op.h

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -95,6 +95,7 @@ class OnnxifiOp final : public Operator<Context> {
9595
CAFFE_ENFORCE_EQ(
9696
lib_->onnxInitGraph(
9797
backend_,
98+
nullptr,
9899
onnx_model_str.size(),
99100
(void*)(onnx_model_str.c_str()),
100101
weight_descs.size(),

0 commit comments

Comments
 (0)