File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change 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
651652Pass 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;
You can’t perform that action at this time.
0 commit comments