Skip to content

Commit 4896b66

Browse files
committed
fix lint
1 parent 1e8cad2 commit 4896b66

File tree

2 files changed

+8
-3
lines changed

2 files changed

+8
-3
lines changed

include/tvm/relax/attrs/sort.h

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -39,9 +39,11 @@ struct SortAttrs : public tvm::AttrsNode<SortAttrs> {
3939
TVM_ATTR_FIELD(axis).set_default(-1).describe(
4040
"Axis along which the sort is computed."
4141
"The default the last axis is used.");
42-
TVM_ATTR_FIELD(descending).set_default(false).describe(
43-
"Whether to sort in descending order."
44-
"If it is not specified, it defaults to the ascending order.");
42+
TVM_ATTR_FIELD(descending)
43+
.set_default(false)
44+
.describe(
45+
"Whether to sort in descending order."
46+
"If it is not specified, it defaults to the ascending order.");
4547
}
4648
}; // struct SortAttrs
4749
} // namespace relax

src/relax/op/tensor/sort.h

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -26,6 +26,9 @@
2626

2727
#include <tvm/relax/attrs/sort.h>
2828

29+
#include <algorithm>
30+
#include <utility>
31+
2932
#include "../op_common.h"
3033

3134
namespace tvm {

0 commit comments

Comments
 (0)