Fix LINT errors

apache · Jun 26, 2024 · 778557e · 778557e
1 parent 78eb4e6
commit 778557e
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/python/tvm/relax/op/manipulate.py b/python/tvm/relax/op/manipulate.py
@@ -162,7 +162,9 @@ def layout_transform(
     if input_axis_separators is None:
         input_axis_separators = []
 
-    return _ffi_api.layout_transform(x, index_map, pad_value, axis_separators, input_axis_separators)  # type: ignore
+    return _ffi_api.layout_transform(
+        x, index_map, pad_value, axis_separators, input_axis_separators
+    )
 
 
 def permute_dims(x: Expr, axes: Optional[List[int]] = None) -> Expr:

diff --git a/src/relax/transform/alter_op_impl.cc b/src/relax/transform/alter_op_impl.cc
@@ -145,7 +145,7 @@ class AlterOpImplMutator : public ExprMutator {
 
     auto call_tir_inputs_tuple = GetRef<Tuple>(call->args[1].as<TupleNode>());
     Tuple updated_inputs = UpdateInputs(call_tir_inputs_tuple, buffer_transforms, axis_separators,
-                                        replacement_func, buffer_pad_values, input_axis_separators);
+                                        input_axis_separators);
 
     ICHECK_EQ(call->sinfo_args.size(), 1) << "call_tir sinfo_args.size() is expected to be 1";
     StructInfo updated_ret_sinfo = UpdateStructInfo(call->sinfo_args[0], buffer_transforms);
@@ -355,7 +355,6 @@ class AlterOpImplMutator : public ExprMutator {
   Expr TransformOutputs(const Expr& expr, const Array<IndexMap>& buffer_transforms,
                         const StructInfo& old_struct_info,
                         const Optional<Array<Array<IntImm>>>& axis_separators,
-                        const Optional<Array<IntImm>>& buffer_pad_values,
                         const Optional<Array<Array<IntImm>>>& input_axis_separators) {
     if (buffer_transforms.empty()) return expr;