From mboxrd@z Thu Jan 1 00:00:00 1970 Return-Path: X-Spam-Checker-Version: SpamAssassin 3.4.0 (2014-02-07) on aws-us-west-2-korg-lkml-1.web.codeaurora.org Received: from kanga.kvack.org (kanga.kvack.org [205.233.56.17]) by smtp.lore.kernel.org (Postfix) with ESMTP id C55DEE77188 for ; Tue, 14 Jan 2025 04:09:28 +0000 (UTC) Received: by kanga.kvack.org (Postfix) id 5A814280002; Mon, 13 Jan 2025 23:09:28 -0500 (EST) Received: by kanga.kvack.org (Postfix, from userid 40) id 55734280001; Mon, 13 Jan 2025 23:09:28 -0500 (EST) X-Delivered-To: int-list-linux-mm@kvack.org Received: by kanga.kvack.org (Postfix, from userid 63042) id 3F90C280002; Mon, 13 Jan 2025 23:09:28 -0500 (EST) X-Delivered-To: linux-mm@kvack.org Received: from relay.hostedemail.com (smtprelay0011.hostedemail.com [216.40.44.11]) by kanga.kvack.org (Postfix) with ESMTP id 1D314280001 for ; Mon, 13 Jan 2025 23:09:28 -0500 (EST) Received: from smtpin03.hostedemail.com (a10.router.float.18 [10.200.18.1]) by unirelay09.hostedemail.com (Postfix) with ESMTP id 8D805805EB for ; Tue, 14 Jan 2025 04:09:27 +0000 (UTC) X-FDA: 83004727974.03.8CD4E61 Received: from mail-pl1-f172.google.com (mail-pl1-f172.google.com [209.85.214.172]) by imf29.hostedemail.com (Postfix) with ESMTP id 97B9212000D for ; Tue, 14 Jan 2025 04:09:25 +0000 (UTC) Authentication-Results: imf29.hostedemail.com; dkim=pass header.d=gmail.com header.s=20230601 header.b=gqBiKAks; spf=pass (imf29.hostedemail.com: domain of 21cnbao@gmail.com designates 209.85.214.172 as permitted sender) smtp.mailfrom=21cnbao@gmail.com; dmarc=pass (policy=none) header.from=gmail.com ARC-Message-Signature: i=1; a=rsa-sha256; c=relaxed/relaxed; d=hostedemail.com; s=arc-20220608; t=1736827765; h=from:from:sender:reply-to:subject:subject:date:date: message-id:message-id:to:to:cc:cc:mime-version:mime-version: content-type:content-type: content-transfer-encoding:content-transfer-encoding: in-reply-to:in-reply-to:references:references:dkim-signature; bh=n9OhR8H2gjpx6QaBe0RO8fceTVFYek4y7WXPW0XqYgE=; b=T/d/kPi5cmMyYaJ1fO1ZNIeVlSVKjRQjFTIup0OWJKImQWIb7w6CPJbXgqiyRCZEW0fO8Q 3WoON9FKWeqlTfcseg0KLKNFo6z1Hj6SpZlUlSm2XOBXkNkMq5jIUgrOZ+qrey8UvKj+NG BhkT960mgKnTtZTLZa8ygm35d41jPrg= ARC-Seal: i=1; s=arc-20220608; d=hostedemail.com; t=1736827765; a=rsa-sha256; cv=none; b=VPnBuJBnb/pTtL48suNrigNKPDDNpTOgp5crmhCBIGEODBB1WQNfJYN50xdMrQBDtrNe28 mKmCFx5X5MS1tUHS2GvPq3dFjOMu0QFaXFIgmmebN9lRdwJs2Z+X3D0Z+KK5UiAZozGH6j w9Mx2PnW5V6F/8Q7HF4lQGjhCnt0LS0= ARC-Authentication-Results: i=1; imf29.hostedemail.com; dkim=pass header.d=gmail.com header.s=20230601 header.b=gqBiKAks; spf=pass (imf29.hostedemail.com: domain of 21cnbao@gmail.com designates 209.85.214.172 as permitted sender) smtp.mailfrom=21cnbao@gmail.com; dmarc=pass (policy=none) header.from=gmail.com Received: by mail-pl1-f172.google.com with SMTP id d9443c01a7336-2164b1f05caso87709315ad.3 for ; Mon, 13 Jan 2025 20:09:25 -0800 (PST) DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d=gmail.com; s=20230601; t=1736827764; x=1737432564; darn=kvack.org; h=content-transfer-encoding:mime-version:references:in-reply-to :message-id:date:subject:cc:to:from:from:to:cc:subject:date :message-id:reply-to; bh=n9OhR8H2gjpx6QaBe0RO8fceTVFYek4y7WXPW0XqYgE=; b=gqBiKAksfOzr4gAvHe18MZpsJZksR8dnS1TdeFlSbT8BQNUqEz4uafWnBNqjSDRq/1 xGAgvenYCTK0emq5GdQt1akz/QDpb39BTx/RZ4FbhnTqfhWDj7RYyq3nQ7wb1gtn7cPw yJ9DZ5/pR1kX+BdInQLERTqm1LykjIorQaNYdpPaXYcpYjaw5QbBvUqiiXepXhbHAira QMBswiLIpdgpe8i94XV13I729GrhM8upczGOIaZzqk1+uRlXdCo5OwPTOnwEDjCm19mh YpW7l1vil+cmWbSFumiodQRfnqhmsjbaNCnGGM3PEUv6y10jW/wvq+XplB6mWa/k5S+K ejoQ== X-Google-DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d=1e100.net; s=20230601; t=1736827764; x=1737432564; h=content-transfer-encoding:mime-version:references:in-reply-to :message-id:date:subject:cc:to:from:x-gm-message-state:from:to:cc :subject:date:message-id:reply-to; bh=n9OhR8H2gjpx6QaBe0RO8fceTVFYek4y7WXPW0XqYgE=; b=uUYU99Vkzu0xIcjqK5jbWvzPKGodxvXMr0XVzXJcILvxdhLimSpzFTRbsjX6zdn5DT 8kc5LhOL3jtfkkGESMm6leUsyYvTjCNZh7ScecY8GyhMsf1auJ+aB75r4yf+UbWZ2yXU PyqebvatOqNxkr4KjdrYZDEndc+L5VOhuJVYfZvjR/rbfvrDtx5Ojgo15A1ISa9Yx6bx l/6dyy6szF1PEgkDZZauwvRsqZql7fvP8kGNl8jJEuicniMYYoPEs3Kc56HPCZorKNKt mm42MrSHaErphQbNc6qQBfsYYJkgfDzqXK9uGwAm1irtxs3zR7tUyzc7PrRuMFCHfiY9 ozyg== X-Forwarded-Encrypted: i=1; AJvYcCUqjxgHQfuLS0ihqBnnPvQtqT3e50FfXEpe0tCxgGXBWuYj5bprW20X2GoDhedeawIuwttl+J9sRQ==@kvack.org X-Gm-Message-State: AOJu0YyFM0IS7/ZobAM2KDr5jBIlWU0aWIOinmOgzqczEFwErGIBbR6O A7dJPIt0N+3GPh3a1zoJpMuIltW+DTnUgvLlR6M3LyTOZGYv/mWn X-Gm-Gg: ASbGncu4rHHo0Q5rVlQBsLtvhAuksKPwsUFHKeRAOxqdn4hjtrWapAHl5DRMQnVy2bJ muc6BRr5VuW5Qc1xasAzWlvjzGxFaB48wwEhgryfovXzfxvg/raeJO1KK7ZsP9gQx/K1O/USU1E yGxQR6lM01z7gd5D/DVupl1ljmZA9U5CQJb3x2yphGhjorXLO3vfW40PrSM91rj9MpkEdtholqd Y7trHD/2e3w1rrrBqgnXqr9R4j4/Quvw+iae49k32I2S99/voTNAEA3/+VdhKGeMCzqrthyESX4 OMeUGYB5 X-Google-Smtp-Source: AGHT+IHT2li0BzNMESB2ucdlAXFPz0+v3u/41Z6CwFvGFPW3mNOsWkDv1pbcf1Z6H258l6sGfOmnJg== X-Received: by 2002:a17:902:ecc5:b0:216:57a6:2887 with SMTP id d9443c01a7336-21a83f5db98mr392066215ad.21.1736827764068; Mon, 13 Jan 2025 20:09:24 -0800 (PST) Received: from Barrys-MBP.hub ([2407:7000:af65:8200:39b5:3f0b:acf3:9158]) by smtp.gmail.com with ESMTPSA id 98e67ed59e1d1-2f54a2872a3sm10803658a91.16.2025.01.13.20.09.17 (version=TLS1_3 cipher=TLS_CHACHA20_POLY1305_SHA256 bits=256/256); Mon, 13 Jan 2025 20:09:23 -0800 (PST) From: Barry Song <21cnbao@gmail.com> To: baolin.wang@linux.alibaba.com Cc: 21cnbao@gmail.com, akpm@linux-foundation.org, chrisl@kernel.org, david@redhat.com, ioworker0@gmail.com, kasong@tencent.com, linux-arm-kernel@lists.infradead.org, linux-kernel@vger.kernel.org, linux-mm@kvack.org, linux-riscv@lists.infradead.org, lorenzo.stoakes@oracle.com, ryan.roberts@arm.com, v-songbaohua@oppo.com, x86@kernel.org, ying.huang@intel.com, zhengtangquan@oppo.com Subject: Re: [PATCH v2 4/4] mm: Avoid splitting pmd for lazyfree pmd-mapped THP in try_to_unmap Date: Tue, 14 Jan 2025 17:09:14 +1300 Message-Id: <20250114040914.9986-1-21cnbao@gmail.com> X-Mailer: git-send-email 2.39.3 (Apple Git-146) In-Reply-To: References: MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit X-Rspamd-Queue-Id: 97B9212000D X-Rspam-User: X-Rspamd-Server: rspam07 X-Stat-Signature: bi9ri8roga4reddqyrdyts57swky54t3 X-HE-Tag: 1736827765-443106 X-HE-Meta: U2FsdGVkX19+CToarznKvti3d6F3klIKHYN0jCcLPFmwmCQqPqHYBn8hYODfz7jwbEIRo9fPEESXx0yrNZYbRtHNzSc/XVLi3Fn+xXViRFq+Svv0TY3s7QtbaRzVC71fAGytISUUc7LDW5v+r6z2rNDFoX1ywtrcu/oZoKSVeTnhujsWrlrDmpc1UDz9vAmDyj9l331D93fOIhYcVnSjolOWer3re2sazE9DWeHJJ3mAquWccoYcKALf9uqCqazRB/HqZvubCnW1n5juk6a0isROI5FYupwSHqNOiBK1vA8xvGAjYAS1fTKb3TXszTpIPc95ijDI/NBJC73daZEU2oIyhGAdAb9ayrY2ZpUGxYqbQFfdObV08r2RyqWpYN2meIGdBHnmKBPbRf0glolRH19bT6Yu9VbsUhFgkHh/7xjS1hYppdaQaYhwldP5hE+uozAeieqabVu9Ya6na9lt6LmSFGLxkcVF+NYc7hJ1vbTZwDyXtskiXg6H98j7Cki3zh1Zht/2u2URiCOk5sGwX0ZOz/obc881qVBfEbKEHW5cPXRkHNB8f0LkLZBWboPWXw865hGe5v02ZDSvbqwjIGvZuqvxII8VqTlTtMQeMsA/296/LhCFTi52iSfiu4XZJ5i0GoTWcbymhooTHhyjlR+hYHf1STFHHeTwkrNw8rwAfm0O9LyOyuu9XKTJxNGuqaDotMO0U+7z0EF9Rx+qlG8mfTs504f6KIs1GKP1Y1wYqYEpWx/OP6i+zUntSwkzbbH6qO55p0Ije4rn3AKalHwwUSCBWw1iMWgODI9mtW7NFLgFB0cf47HxOvw9Qk6Z73BTKupyR4q3xor5bctjJ9t/VLvYepYDbW+jliWVfOhwkhT579Wobd/zIllOLhdpi0Aihs+6ykuAcrMqj2CUl1gwg28Rp0xixrorYEaqKM3/KXo1DjcrDe1Yzl8j7uM4y8XrsKc3Hs29/wC9zeQ P4fQdgg+ 3vo32keiDgD9M/5D+bpg72mFG8CjWQTJMrtZ4Bj7Un08a4ABBivDIYO7d6laCdMdDC2QJNKQY6xkm2qh4kVLM8QJ0NEstXfZQlBpBqSsUxDmrRBRx/CHLuqMVD+HOoOtIuY2ZbDlCaWOEqtULX+jFrZRMyHKvnTsN85T/X8pYmZux3k9WvBCuIazQtWhrZiBWk1429q5QiTnfGk3aXVUXRxZIe3TGzTiz5Xm427ta/GRqCxoBOE7XgiWPm2XGR0zyFLOFFhJi9YmDv8QVej+zz0xRBB5eHwzVcnbADrK0xwh6P6gLMklVLQeYbMcyPNmtco8K8JjxGi33TuwN6fxtjx57HBZDv2xmo66kgvVfz8Y7B9PPHIC9QJZFrE4tdfNLYQsjz1EIcKgpylpkZk3uWYGePOq2kzOlH3iK6+/Qi3sumUH01VRcixWo8zHiZhPM7rcGItbFYqRcZ3NFlHlpwfB352Ia4RMaKetTmu9582QTe2smPZiCxy2ipV6yxUpL6X3gZfuBE/eed+0= X-Bogosity: Ham, tests=bogofilter, spamicity=0.000000, version=1.2.4 Sender: owner-linux-mm@kvack.org Precedence: bulk X-Loop: owner-majordomo@kvack.org List-ID: List-Subscribe: List-Unsubscribe: On Tue, Jan 14, 2025 at 4:40 PM Baolin Wang wrote: > > > > On 2025/1/13 11:39, Barry Song wrote: > > From: Barry Song > > > > The try_to_unmap_one() function currently handles PMD-mapped THPs > > inefficiently. It first splits the PMD into PTEs, copies the dirty > > state from the PMD to the PTEs, iterates over the PTEs to locate > > the dirty state, and then marks the THP as swap-backed. This process > > involves unnecessary PMD splitting and redundant iteration. Instead, > > this functionality can be efficiently managed in > > __discard_anon_folio_pmd_locked(), avoiding the extra steps and > > improving performance. > > > > The following microbenchmark redirties folios after invoking MADV_FREE, > > then measures the time taken to perform memory reclamation (actually > > set those folios swapbacked again) on the redirtied folios. > > > > #include > > #include > > #include > > #include > > > > #define SIZE 128*1024*1024 // 128 MB > > > > int main(int argc, char *argv[]) > > { > > while(1) { > > volatile int *p = mmap(0, SIZE, PROT_READ | PROT_WRITE, > > MAP_PRIVATE | MAP_ANONYMOUS, -1, 0); > > > > memset((void *)p, 1, SIZE); > > madvise((void *)p, SIZE, MADV_FREE); > > /* redirty after MADV_FREE */ > > memset((void *)p, 1, SIZE); > > > > clock_t start_time = clock(); > > madvise((void *)p, SIZE, MADV_PAGEOUT); > > clock_t end_time = clock(); > > > > double elapsed_time = (double)(end_time - start_time) / CLOCKS_PER_SEC; > > printf("Time taken by reclamation: %f seconds\n", elapsed_time); > > > > munmap((void *)p, SIZE); > > } > > return 0; > > } > > > > Testing results are as below, > > w/o patch: > > ~ # ./a.out > > Time taken by reclamation: 0.007300 seconds > > Time taken by reclamation: 0.007226 seconds > > Time taken by reclamation: 0.007295 seconds > > Time taken by reclamation: 0.007731 seconds > > Time taken by reclamation: 0.007134 seconds > > Time taken by reclamation: 0.007285 seconds > > Time taken by reclamation: 0.007720 seconds > > Time taken by reclamation: 0.007128 seconds > > Time taken by reclamation: 0.007710 seconds > > Time taken by reclamation: 0.007712 seconds > > Time taken by reclamation: 0.007236 seconds > > Time taken by reclamation: 0.007690 seconds > > Time taken by reclamation: 0.007174 seconds > > Time taken by reclamation: 0.007670 seconds > > Time taken by reclamation: 0.007169 seconds > > Time taken by reclamation: 0.007305 seconds > > Time taken by reclamation: 0.007432 seconds > > Time taken by reclamation: 0.007158 seconds > > Time taken by reclamation: 0.007133 seconds > > … > > > > w/ patch > > > > ~ # ./a.out > > Time taken by reclamation: 0.002124 seconds > > Time taken by reclamation: 0.002116 seconds > > Time taken by reclamation: 0.002150 seconds > > Time taken by reclamation: 0.002261 seconds > > Time taken by reclamation: 0.002137 seconds > > Time taken by reclamation: 0.002173 seconds > > Time taken by reclamation: 0.002063 seconds > > Time taken by reclamation: 0.002088 seconds > > Time taken by reclamation: 0.002169 seconds > > Time taken by reclamation: 0.002124 seconds > > Time taken by reclamation: 0.002111 seconds > > Time taken by reclamation: 0.002224 seconds > > Time taken by reclamation: 0.002297 seconds > > Time taken by reclamation: 0.002260 seconds > > Time taken by reclamation: 0.002246 seconds > > Time taken by reclamation: 0.002272 seconds > > Time taken by reclamation: 0.002277 seconds > > Time taken by reclamation: 0.002462 seconds > > … > > > > This patch significantly speeds up try_to_unmap_one() by allowing it > > to skip redirtied THPs without splitting the PMD. > > > > Suggested-by: Baolin Wang > > Suggested-by: Lance Yang > > Signed-off-by: Barry Song > > --- > > mm/huge_memory.c | 17 ++++++++++++++--- > > mm/rmap.c | 11 ++++++++++- > > 2 files changed, 24 insertions(+), 4 deletions(-) > > > > diff --git a/mm/huge_memory.c b/mm/huge_memory.c > > index 3d3ebdc002d5..aea49f7125f1 100644 > > --- a/mm/huge_memory.c > > +++ b/mm/huge_memory.c > > @@ -3070,8 +3070,12 @@ static bool __discard_anon_folio_pmd_locked(struct vm_area_struct *vma, > > int ref_count, map_count; > > pmd_t orig_pmd = *pmdp; > > > > - if (folio_test_dirty(folio) || pmd_dirty(orig_pmd)) > > + if (pmd_dirty(orig_pmd)) > > + folio_set_dirty(folio); > > + if (folio_test_dirty(folio) && !(vma->vm_flags & VM_DROPPABLE)) { > > + folio_set_swapbacked(folio); > > return false; > > + } > > > > orig_pmd = pmdp_huge_clear_flush(vma, addr, pmdp); > > > > @@ -3098,8 +3102,15 @@ static bool __discard_anon_folio_pmd_locked(struct vm_area_struct *vma, > > * > > * The only folio refs must be one from isolation plus the rmap(s). > > */ > > - if (folio_test_dirty(folio) || pmd_dirty(orig_pmd) || > > - ref_count != map_count + 1) { > > + if (pmd_dirty(orig_pmd)) > > + folio_set_dirty(folio); > > + if (folio_test_dirty(folio) && !(vma->vm_flags & VM_DROPPABLE)) { > > + folio_set_swapbacked(folio); > > + set_pmd_at(mm, addr, pmdp, orig_pmd); > > + return false; > > + } > > + > > + if (ref_count != map_count + 1) { > > set_pmd_at(mm, addr, pmdp, orig_pmd); > > return false; > > } > > diff --git a/mm/rmap.c b/mm/rmap.c > > index 3ef659310797..02c4e4b2cd7b 100644 > > --- a/mm/rmap.c > > +++ b/mm/rmap.c > > @@ -1671,7 +1671,7 @@ static bool try_to_unmap_one(struct folio *folio, struct vm_area_struct *vma, > > DEFINE_FOLIO_VMA_WALK(pvmw, folio, vma, address, 0); > > pte_t pteval; > > struct page *subpage; > > - bool anon_exclusive, ret = true; > > + bool anon_exclusive, lazyfree, ret = true; > > struct mmu_notifier_range range; > > enum ttu_flags flags = (enum ttu_flags)(long)arg; > > int nr_pages = 1; > > @@ -1724,9 +1724,18 @@ static bool try_to_unmap_one(struct folio *folio, struct vm_area_struct *vma, > > } > > > > if (!pvmw.pte) { > > + lazyfree = folio_test_anon(folio) && !folio_test_swapbacked(folio); > > You've checked lazyfree here, so can we remove the duplicate check in > unmap_huge_pmd_locked()? Then the code should be: > > if (lazyfree && unmap_huge_pmd_locked(...)) > goto walk_done; right. it seems unmap_huge_pmd_locked() only handles lazyfree pmd-mapped thp. so i guess the code could be: diff --git a/mm/huge_memory.c b/mm/huge_memory.c index aea49f7125f1..c4c3a7896de4 100644 --- a/mm/huge_memory.c +++ b/mm/huge_memory.c @@ -3131,11 +3131,10 @@ bool unmap_huge_pmd_locked(struct vm_area_struct *vma, unsigned long addr, VM_WARN_ON_FOLIO(!folio_test_pmd_mappable(folio), folio); VM_WARN_ON_FOLIO(!folio_test_locked(folio), folio); VM_WARN_ON_ONCE(!IS_ALIGNED(addr, HPAGE_PMD_SIZE)); + VM_WARN_ON_FOLIO(!folio_test_anon(folio), folio); + VM_WARN_ON_FOLIO(folio_test_swapbacked(folio), folio); - if (folio_test_anon(folio) && !folio_test_swapbacked(folio)) - return __discard_anon_folio_pmd_locked(vma, addr, pmdp, folio); - - return false; + return __discard_anon_folio_pmd_locked(vma, addr, pmdp, folio); } static void remap_page(struct folio *folio, unsigned long nr, int flags) diff --git a/mm/rmap.c b/mm/rmap.c index 02c4e4b2cd7b..72907eb1b8fe 100644 --- a/mm/rmap.c +++ b/mm/rmap.c @@ -1671,7 +1671,7 @@ static bool try_to_unmap_one(struct folio *folio, struct vm_area_struct *vma, DEFINE_FOLIO_VMA_WALK(pvmw, folio, vma, address, 0); pte_t pteval; struct page *subpage; - bool anon_exclusive, lazyfree, ret = true; + bool anon_exclusive, ret = true; struct mmu_notifier_range range; enum ttu_flags flags = (enum ttu_flags)(long)arg; int nr_pages = 1; @@ -1724,18 +1724,16 @@ static bool try_to_unmap_one(struct folio *folio, struct vm_area_struct *vma, } if (!pvmw.pte) { - lazyfree = folio_test_anon(folio) && !folio_test_swapbacked(folio); - - if (unmap_huge_pmd_locked(vma, pvmw.address, pvmw.pmd, - folio)) - goto walk_done; - /* - * unmap_huge_pmd_locked has either already marked - * the folio as swap-backed or decided to retain it - * due to GUP or speculative references. - */ - if (lazyfree) + if (folio_test_anon(folio) && !folio_test_swapbacked(folio)) { + if (unmap_huge_pmd_locked(vma, pvmw.address, pvmw.pmd, folio)) + goto walk_done; + /* + * unmap_huge_pmd_locked has either already marked + * the folio as swap-backed or decided to retain it + * due to GUP or speculative references. + */ goto walk_abort; + } if (flags & TTU_SPLIT_HUGE_PMD) { /* > > > if (unmap_huge_pmd_locked(vma, pvmw.address, pvmw.pmd, > > folio)) > > goto walk_done; > > + /* > > + * unmap_huge_pmd_locked has either already marked > > + * the folio as swap-backed or decided to retain it > > + * due to GUP or speculative references. > > + */ > > + if (lazyfree) > > + goto walk_abort; > > > > if (flags & TTU_SPLIT_HUGE_PMD) { > > /*