Skip to content

Commit cb2ea17

Browse files
jackm321facebook-github-bot
authored andcommitted
Automatic update of fbcode/foxi to 2bcc4064c90e87b9638615c733485f07c47b7558 (pytorch#18070)
Summary: Pull Request resolved: pytorch#18070 Previous import was d1f45b1a2b1585d0e9bc65e15e463db344fc3ff6 Included changes: - **[2bcc406](houseroad/foxi@2bcc406)**: Merge pull request pytorch#7 from jackm321/tracing_fixes <Jack Montgomery> - **[c39033c](houseroad/foxi@c39033c)**: Fixes for tracing events <Jack Montgomery> - **[50912cf](houseroad/foxi@50912cf)**: Merge pull request pytorch#5 from jackm321/add_trace_events <Jack Montgomery> - **[ba2fdcb](houseroad/foxi@ba2fdcb)**: Merge pull request pytorch#5 from jackm321/add_trace_events <Jack Montgomery> - **[7d42b12](houseroad/foxi@7d42b12)**: address comments <Jack Montgomery> - **[dcabd8d](houseroad/foxi@dcabd8d)**: Add trace events interface <Jack Montgomery> Reviewed By: houseroad Differential Revision: D14483201 fbshipit-source-id: f51ed869c9a89521079df89903abc0ac0a45ac7b
1 parent d1843d4 commit cb2ea17

File tree

3 files changed

+5
-3
lines changed

3 files changed

+5
-3
lines changed

caffe2/operators/onnxifi_op.cc

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -172,7 +172,8 @@ bool OnnxifiOp<float, CPUContext>::RunOnDevice() {
172172
input_desc_.data(),
173173
output_desc_.size(),
174174
output_desc_.data(),
175-
&output_fence),
175+
&output_fence,
176+
/* traceEvents */ nullptr),
176177
ONNXIFI_STATUS_SUCCESS);
177178
CAFFE_ENFORCE_EQ(
178179
lib_->onnxWaitEvent(output_fence.event), ONNXIFI_STATUS_SUCCESS);

caffe2/operators/onnxifi_op.h

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -230,7 +230,8 @@ class OnnxifiOp final : public Operator<Context> {
230230
const onnxTensorDescriptorV1*,
231231
uint32_t,
232232
const onnxTensorDescriptorV1*,
233-
onnxMemoryFenceV1*);
233+
onnxMemoryFenceV1*,
234+
onnxTraceEventList*);
234235
#endif
235236

236237
// We bind the op input/output by position while ONNXIFI binds input/output by

0 commit comments

Comments
 (0)