Skip to content

Commit c8979e1

Browse files
committed
Fix lint issue
1 parent e018afb commit c8979e1

File tree

2 files changed

+4
-5
lines changed

2 files changed

+4
-5
lines changed

python/tvm/relay/op/contrib/dnnl.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -41,7 +41,7 @@
4141
from tvm.relay.expr_functor import ExprMutator, ExprVisitor
4242

4343
from ... import _ffi_api
44-
from ...dataflow_pattern import wildcard, is_op, is_constant, is_expr, rewrite, DFPatternCallback
44+
from ...dataflow_pattern import wildcard, is_op, is_expr, rewrite, DFPatternCallback
4545
from .register import register_pattern_table
4646

4747
logger = logging.getLogger("DNNL")

src/runtime/contrib/dnnl/dnnl_json_runtime.cc

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -473,7 +473,6 @@ class DNNLJSONRuntime : public JSONRuntimeBase {
473473
dnnl::normalization_flags::use_scale_shift);
474474

475475
auto lnorm_prim_desc = dnnl::layer_normalization_forward::primitive_desc(lnorm_desc, engine_);
476-
auto lnorm_prim = dnnl::layer_normalization_forward(lnorm_prim_desc);
477476

478477
// Concatenate scale and shift tensors
479478
auto scale_shift_tr = TensorRequisite::AsIs(lnorm_prim_desc.weights_desc(), GenUniqueEid());
@@ -493,9 +492,9 @@ class DNNLJSONRuntime : public JSONRuntimeBase {
493492
register_copy(gamma_tr, scale_tr);
494493
register_copy(beta_tr, shift_tr);
495494

496-
Submit(dnnl::layer_normalization_forward(lnorm_prim_desc), {{DNNL_ARG_SRC, src_tr},
497-
{DNNL_ARG_DST, dst_tr},
498-
{DNNL_ARG_SCALE_SHIFT, scale_shift_tr}});
495+
Submit(
496+
dnnl::layer_normalization_forward(lnorm_prim_desc),
497+
{{DNNL_ARG_SRC, src_tr}, {DNNL_ARG_DST, dst_tr}, {DNNL_ARG_SCALE_SHIFT, scale_shift_tr}});
499498
}
500499

501500
void Pooling(const size_t& nid, dnnl::algorithm algo) {

0 commit comments

Comments
 (0)