Skip to content

Commit

Permalink
fix dnnl runtime
Browse files Browse the repository at this point in the history
  • Loading branch information
rebel-jangys committed Mar 30, 2023
1 parent 3b1c52a commit 391e344
Showing 1 changed file with 1 addition and 6 deletions.
7 changes: 1 addition & 6 deletions src/runtime/contrib/dnnl/dnnl_json_runtime.cc
Original file line number Diff line number Diff line change
Expand Up @@ -424,15 +424,10 @@ class DNNLJSONRuntime : public JSONRuntimeBase {
// Minus one for DNNL representation. No dilation for DNNL is 0, for relay is 1.
for (auto& d : dilates) d--;

// TODO(@apeskov): WA. conv3dTranspose uses wrong layout specifier. IO instead of OI.
auto wgh_logic_layout = TensorRequisite::DefaultLogicLayoutFor(wgh_layout);
if (wgh_logic_layout == "OIDHW") wgh_logic_layout = "IODHW";
if (wgh_logic_layout == "GOIDHW") wgh_logic_layout = "GIODHW";

// Take into account provided layout strings
src_tr = src_tr.TreatAs(src_layout);
dst_tr = dst_tr.TreatAs(dst_layout);
wgh_tr = wgh_tr.TreatAs(wgh_layout, wgh_logic_layout);
wgh_tr = wgh_tr.TreatAs(wgh_layout);

// Should support G mixed with O. Like { G*O, I, H, W }
if (wgh_layout.find("G") == std::string::npos) {
Expand Down

0 comments on commit 391e344

Please sign in to comment.