Skip to content

Commit

Permalink
linear: remove mem copy in external call
Browse files Browse the repository at this point in the history
  • Loading branch information
chunyuan-w committed Jun 23, 2022
1 parent 434191a commit a95b140
Showing 1 changed file with 0 additions and 4 deletions.
4 changes: 0 additions & 4 deletions torch/csrc/jit/tensorexpr/external_functions.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -1452,10 +1452,6 @@ void nnc_mkldnn_prepacked_linear_run(
auto context = reinterpret_cast<LinearOpContext*>(buf_data[2]);

at::Tensor output = context->run(x);
// memcpy(
// buf_data[0], output.data_ptr(), output.element_size() *
// output.numel());
// TODO: remove mem copy
context->run(x, buf_data[0]);
}

Expand Down

0 comments on commit a95b140

Please sign in to comment.