Skip to content

Commit e3216a6

Browse files
LeiWang1999vinx13
authored andcommitted
lint fix
1 parent 76ceff2 commit e3216a6

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

src/tir/transforms/merge_shared_memory_allocations.cc

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,8 @@
2020
/*!
2121
* \file merge_shared_memory_allocations.cc
2222
* \brief Each GPU kernel is allowed to have only one dynamic or static shared memory allocation.
23-
* This pass merges multiple TIR-level dynamic or static shared memory allocations into one allocation.
23+
* This pass merges multiple TIR-level dynamic or static shared memory allocations into one
24+
* allocation.
2425
*/
2526
#include <tvm/runtime/registry.h>
2627
#include <tvm/tir/expr.h>
@@ -650,8 +651,7 @@ namespace transform {
650651

651652
Pass MergeSharedMemoryAllocations() {
652653
auto pass_func = [](PrimFunc f, IRModule m, PassContext ctx) {
653-
bool merge_static_smem =
654-
ctx->GetConfig<Bool>("tir.merge_static_smem", Bool(false)).value();
654+
bool merge_static_smem = ctx->GetConfig<Bool>("tir.merge_static_smem", Bool(false)).value();
655655
auto* n = f.CopyOnWrite();
656656
n->body = MergeSharedMemoryAllocations(std::move(n->body), merge_static_smem);
657657
return f;

0 commit comments

Comments
 (0)