Skip to content

Commit a7018ad

Browse files
committed
Fix LINT errors
1 parent 78eb4e6 commit a7018ad

File tree

2 files changed

+4
-2
lines changed

2 files changed

+4
-2
lines changed

python/tvm/relax/op/manipulate.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -162,7 +162,9 @@ def layout_transform(
162162
if input_axis_separators is None:
163163
input_axis_separators = []
164164

165-
return _ffi_api.layout_transform(x, index_map, pad_value, axis_separators, input_axis_separators) # type: ignore
165+
return _ffi_api.layout_transform(
166+
x, index_map, pad_value, axis_separators, input_axis_separators
167+
)
166168

167169

168170
def permute_dims(x: Expr, axes: Optional[List[int]] = None) -> Expr:

src/relax/transform/alter_op_impl.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -145,7 +145,7 @@ class AlterOpImplMutator : public ExprMutator {
145145

146146
auto call_tir_inputs_tuple = GetRef<Tuple>(call->args[1].as<TupleNode>());
147147
Tuple updated_inputs = UpdateInputs(call_tir_inputs_tuple, buffer_transforms, axis_separators,
148-
replacement_func, buffer_pad_values, input_axis_separators);
148+
input_axis_separators);
149149

150150
ICHECK_EQ(call->sinfo_args.size(), 1) << "call_tir sinfo_args.size() is expected to be 1";
151151
StructInfo updated_ret_sinfo = UpdateStructInfo(call->sinfo_args[0], buffer_transforms);

0 commit comments

Comments
 (0)