Skip to content

Commit 951f2de

Browse files
LeiWang1999LeiWang1999
committed
lint fix
1 parent 3c175e4 commit 951f2de

35 files changed

+93
-74
lines changed

‎src/ir.cc‎

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -233,7 +233,8 @@ class KernelLaunchFrame : public TIRFrame {
233233
KernelLaunchFrame KernelLaunch(const Array<PrimExpr> &grid_size,
234234
const Optional<Array<PrimExpr>> &block_size_opt,
235235
const Map<String, ffi::Any> &attrs) {
236-
ObjectPtr<KernelLaunchFrameNode> n = tvm::ffi::make_object<KernelLaunchFrameNode>();
236+
ObjectPtr<KernelLaunchFrameNode> n =
237+
tvm::ffi::make_object<KernelLaunchFrameNode>();
237238

238239
// If the kernel is a CPU kernel, we don't need to launch any threads.
239240
bool is_cpu_kernel_frame =
@@ -348,7 +349,8 @@ class WarpSpecializeFrame : public TIRFrame {
348349
WarpSpecializeFrame WarpSpecialize(const Array<IntImm> &warp_group_ids,
349350
const PrimExpr &thread_idx,
350351
int warp_group_size = 128) {
351-
ObjectPtr<WarpSpecializeFrameNode> n = tvm::ffi::make_object<WarpSpecializeFrameNode>();
352+
ObjectPtr<WarpSpecializeFrameNode> n =
353+
tvm::ffi::make_object<WarpSpecializeFrameNode>();
352354
PrimExpr condition;
353355
std::vector<int> warp_groups;
354356
warp_groups.reserve(warp_group_ids.size());

‎src/layout/layout.cc‎

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -334,8 +334,8 @@ Fragment::Fragment(Array<IterVar> forward_var, Array<PrimExpr> forward_index,
334334
forward_index.Map([&](const PrimExpr &e) { return Substitute(e, vmap); });
335335
forward_thread = Substitute(forward_thread, vmap);
336336

337-
auto n = tvm::ffi::make_object<FragmentNode>(input_size, forward_index, forward_thread,
338-
replicate_size);
337+
auto n = tvm::ffi::make_object<FragmentNode>(input_size, forward_index,
338+
forward_thread, replicate_size);
339339
data_ = std::move(n);
340340
}
341341

@@ -346,8 +346,8 @@ Fragment::Fragment(Array<PrimExpr> input_size, Array<PrimExpr> forward_index,
346346
forward_thread = Substitute(
347347
forward_thread, {{replicate_var.value(), ReplicationPlaceholder()}});
348348
}
349-
auto n = tvm::ffi::make_object<FragmentNode>(input_size, forward_index, forward_thread,
350-
replicate_size);
349+
auto n = tvm::ffi::make_object<FragmentNode>(input_size, forward_index,
350+
forward_thread, replicate_size);
351351
data_ = std::move(n);
352352
}
353353

‎src/layout/layout.h‎

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -49,7 +49,8 @@ class LayoutNode : public Object {
4949

5050
static void RegisterReflection();
5151
TVM_FFI_DECLARE_OBJECT_INFO("tl.Layout", LayoutNode, Object);
52-
static constexpr TVMFFISEqHashKind _type_s_eq_hash_kind = kTVMFFISEqHashKindTreeNode;
52+
static constexpr TVMFFISEqHashKind _type_s_eq_hash_kind =
53+
kTVMFFISEqHashKindTreeNode;
5354

5455
protected:
5556
virtual Map<Var, Range> getVarMap() const;
@@ -111,7 +112,8 @@ class FragmentNode : public LayoutNode {
111112
static void RegisterReflection();
112113

113114
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("tl.Fragment", FragmentNode, LayoutNode);
114-
static constexpr TVMFFISEqHashKind _type_s_eq_hash_kind = kTVMFFISEqHashKindTreeNode;
115+
static constexpr TVMFFISEqHashKind _type_s_eq_hash_kind =
116+
kTVMFFISEqHashKindTreeNode;
115117

116118
protected:
117119
Map<Var, Range> getVarMap() const final;

‎src/layout/swizzle.cc‎

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -87,14 +87,16 @@ SwizzledLayout::SwizzledLayout(Array<IterVar> forward_var,
8787
forward_index =
8888
forward_index.Map([&](const PrimExpr &e) { return Substitute(e, vmap); });
8989

90-
auto n = tvm::ffi::make_object<SwizzledLayoutNode>(input_size, forward_index, pattern);
90+
auto n = tvm::ffi::make_object<SwizzledLayoutNode>(input_size, forward_index,
91+
pattern);
9192
data_ = std::move(n);
9293
}
9394

9495
SwizzledLayout::SwizzledLayout(Array<PrimExpr> input_size,
9596
Array<PrimExpr> forward_index,
9697
SwizzlePattern pattern) {
97-
auto n = tvm::ffi::make_object<SwizzledLayoutNode>(input_size, forward_index, pattern);
98+
auto n = tvm::ffi::make_object<SwizzledLayoutNode>(input_size, forward_index,
99+
pattern);
98100
data_ = std::move(n);
99101
}
100102

‎src/op/atomic_add.h‎

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -63,7 +63,8 @@ class AtomicAddNode : public TileOperatorNode {
6363
/// Wrapper class for atomic addition operations
6464
class AtomicAdd : public TileOperator {
6565
public:
66-
TVM_FFI_DEFINE_OBJECT_REF_METHODS_NULLABLE(AtomicAdd, TileOperator, AtomicAddNode);
66+
TVM_FFI_DEFINE_OBJECT_REF_METHODS_NULLABLE(AtomicAdd, TileOperator,
67+
AtomicAddNode);
6768
TVM_DLL AtomicAdd(Array<PrimExpr> args, BufferMap vmap);
6869
static const Op &Get();
6970
};

‎src/op/copy.cc‎

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1723,7 +1723,8 @@ Array<PrimExpr> TMADesc::EncodeCallArgs() const {
17231723
* @param vmap Mapping from original buffer variables to actual Buffer objects.
17241724
*/
17251725
Conv2DIm2ColOp::Conv2DIm2ColOp(Array<PrimExpr> args, BufferMap vmap) {
1726-
ObjectPtr<Conv2DIm2ColOpNode> node = tvm::ffi::make_object<Conv2DIm2ColOpNode>();
1726+
ObjectPtr<Conv2DIm2ColOpNode> node =
1727+
tvm::ffi::make_object<Conv2DIm2ColOpNode>();
17271728
node->src = vmap[GetVarFromAccessPtr(args[0])];
17281729
node->dst = vmap[GetVarFromAccessPtr(args[1])];
17291730
node->nhw_step = args[2];

‎src/op/finalize_reducer.h‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -47,7 +47,7 @@ class FinalizeReducerOpNode : public TileOperatorNode {
4747
class FinalizeReducerOp : public TileOperator {
4848
public:
4949
TVM_FFI_DEFINE_OBJECT_REF_METHODS_NULLABLE(FinalizeReducerOp, TileOperator,
50-
FinalizeReducerOpNode);
50+
FinalizeReducerOpNode);
5151
TVM_DLL FinalizeReducerOp(Array<PrimExpr> args, BufferMap vmap);
5252
static const Op &Get();
5353
};

‎src/op/gemm.h‎

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -30,8 +30,7 @@ class GemmWarpPolicyNode : public Object {
3030
mutable int n_warp{0};
3131
int policy_type;
3232

33-
TVM_FFI_DECLARE_OBJECT_INFO("tl.GemmWarpPolicy", GemmWarpPolicyNode,
34-
Object);
33+
TVM_FFI_DECLARE_OBJECT_INFO("tl.GemmWarpPolicy", GemmWarpPolicyNode, Object);
3534

3635
static void RegisterReflection() {
3736
namespace refl = tvm::ffi::reflection;
@@ -59,7 +58,8 @@ class GemmWarpPolicyNode : public Object {
5958

6059
class GemmWarpPolicy : public ObjectRef {
6160
public:
62-
TVM_FFI_DEFINE_OBJECT_REF_METHODS_NULLABLE(GemmWarpPolicy, ObjectRef, GemmWarpPolicyNode);
61+
TVM_FFI_DEFINE_OBJECT_REF_METHODS_NULLABLE(GemmWarpPolicy, ObjectRef,
62+
GemmWarpPolicyNode);
6363

6464
explicit GemmWarpPolicy(GemmWarpPolicyType policy_type) {
6565
auto node = tvm::ffi::make_object<GemmWarpPolicyNode>();

‎src/op/gemm_py.cc‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -228,8 +228,8 @@ Stmt GemmPyNode::Lower(const LowerArgs &T, arith::Analyzer *analyzer) const {
228228

229229
if (const auto f = ffi::Function::GetGlobal("tl.gemm_py.lower")) {
230230
auto prim_func =
231-
Downcast<PrimFunc>((*f)(tvm::ffi::GetRef<GemmPy>(this), T.layout_map, T.target,
232-
T.thread_bounds, T.thread_var));
231+
Downcast<PrimFunc>((*f)(tvm::ffi::GetRef<GemmPy>(this), T.layout_map,
232+
T.target, T.thread_bounds, T.thread_var));
233233
ICHECK(prim_func->attrs.defined());
234234
auto global_symbol =
235235
prim_func->attrs.GetAttr<tvm::ffi::String>("global_symbol");

‎src/op/gemm_sp.h‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,7 @@ class GemmSPWarpPolicyNode : public GemmWarpPolicyNode {
2828
class GemmSPWarpPolicy : public ObjectRef {
2929
public:
3030
TVM_FFI_DEFINE_OBJECT_REF_METHODS_NULLABLE(GemmSPWarpPolicy, ObjectRef,
31-
GemmSPWarpPolicyNode);
31+
GemmSPWarpPolicyNode);
3232

3333
explicit GemmSPWarpPolicy(GemmWarpPolicyType policy_type) {
3434
auto node = tvm::ffi::make_object<GemmSPWarpPolicyNode>();

0 commit comments

Comments
 (0)