Received-SPF: pass (google.com: domain of linux-kernel+bounces-139227-linux.lists.archive=gmail.com@vger.kernel.org designates 139.178.88.99 as permitted sender) client-ip=139.178.88.99;
Message-ID: <801a0a7015137a7ef573e2cdd69dca019f328da7.camel@linux.intel.com>
Subject: Re: [PATCH v2] mm: swap: prejudgement swap_has_cache to avoid page
 allocation
From: Tim Chen <tim.c.chen@linux.intel.com>
To: Zhaoyu Liu <liuzhaoyu.zackary@bytedance.com>, akpm@linux-foundation.org,
  ryncsn@gmail.com, nphamcs@gmail.com
Cc: ying.huang@intel.com, songmuchun@bytedance.com, david@redhat.com, 
	chrisl@kernel.org, guo.ziliang@zte.com.cn, yosryahmed@google.com, 
	linux-kernel@vger.kernel.org, linux-mm@kvack.org
Date: Wed, 10 Apr 2024 11:55:43 -0700
In-Reply-To: <20240408121439.GA252652@bytedance>
References: <20240408121439.GA252652@bytedance>
Content-Type: text/plain; charset="UTF-8"
Content-Transfer-Encoding: quoted-printable
User-Agent: Evolution 3.44.4 (3.44.4-2.fc36) 
Precedence: bulk
MIME-Version: 1.0

On Mon, 2024-04-08 at 20:14 +0800, Zhaoyu Liu wrote:
> Based on qemu arm64 - latest kernel + 100M memory + 1024M swapfile.
> Create 1G anon mmap and set it to shared, and has two processes
> randomly access the shared memory. When they are racing on swap cache,
> on average, each "alloc_pages_mpol + swapcache_prepare + folio_put"
> took about 1475 us.
>=20
> So skip page allocation if SWAP_HAS_CACHE was set, just
> schedule_timeout_uninterruptible and continue to acquire page
> via filemap_get_folio() from swap cache, to speedup
> __read_swap_cache_async.
>=20
> Signed-off-by: Zhaoyu Liu <liuzhaoyu.zackary@bytedance.com>
> ---
> Changes in v2:
>   - Fix the patch format and rebase to latest linux-next.
> ---
>  include/linux/swap.h |  6 ++++++
>  mm/swap_state.c      | 10 ++++++++++
>  mm/swapfile.c        | 15 +++++++++++++++
>  3 files changed, 31 insertions(+)
>=20
> diff --git a/include/linux/swap.h b/include/linux/swap.h
> index 11c53692f65f..a374070e05a7 100644
> --- a/include/linux/swap.h
> +++ b/include/linux/swap.h
> @@ -492,6 +492,7 @@ extern sector_t swapdev_block(int, pgoff_t);
>  extern int __swap_count(swp_entry_t entry);
>  extern int swap_swapcount(struct swap_info_struct *si, swp_entry_t entry=
);
>  extern int swp_swapcount(swp_entry_t entry);
> +extern bool swap_has_cache(struct swap_info_struct *si, swp_entry_t entr=
y);
>  struct swap_info_struct *swp_swap_info(swp_entry_t entry);
>  struct backing_dev_info;
>  extern int init_swap_address_space(unsigned int type, unsigned long nr_p=
ages);
> @@ -583,6 +584,11 @@ static inline int swp_swapcount(swp_entry_t entry)
>  	return 0;
>  }
> =20
> +static inline bool swap_has_cache(struct swap_info_struct *si, swp_entry=
_t entry)
> +{
> +	return false;
> +}
> +
>  static inline swp_entry_t folio_alloc_swap(struct folio *folio)
>  {
>  	swp_entry_t entry;
> diff --git a/mm/swap_state.c b/mm/swap_state.c
> index 642c30d8376c..f117fbf18b59 100644
> --- a/mm/swap_state.c
> +++ b/mm/swap_state.c
> @@ -462,6 +462,15 @@ struct folio *__read_swap_cache_async(swp_entry_t en=
try, gfp_t gfp_mask,
>  		if (!swap_swapcount(si, entry) && swap_slot_cache_enabled)
>  			goto fail_put_swap;
> =20
> +		/*
> +		 * Skipping page allocation if SWAP_HAS_CACHE was set,
> +		 * just schedule_timeout_uninterruptible and continue to
> +		 * acquire page via filemap_get_folio() from swap cache,
> +		 * to speedup __read_swap_cache_async.
> +		 */
> +		if (swap_has_cache(si, entry))
> +			goto skip_alloc;
> +

I think most of the cases where a page already exists will be caught by
filemap_get_folio().  The cases caught by this extra check should be when w=
e have races
between page cache update and the read async, which may not be that
often.  So please verify with benchmark that this extra check with its
own overhead would buy us anything.

Tim

>  		/*
>  		 * Get a new folio to read into from swap.  Allocate it now,
>  		 * before marking swap_map SWAP_HAS_CACHE, when -EEXIST will
> @@ -483,6 +492,7 @@ struct folio *__read_swap_cache_async(swp_entry_t ent=
ry, gfp_t gfp_mask,
>  		if (err !=3D -EEXIST)
>  			goto fail_put_swap;
> =20
> +skip_alloc:
>  		/*
>  		 * Protect against a recursive call to __read_swap_cache_async()
>  		 * on the same entry waiting forever here because SWAP_HAS_CACHE
> diff --git a/mm/swapfile.c b/mm/swapfile.c
> index 3ee8957a46e6..b016ebc43b0d 100644
> --- a/mm/swapfile.c
> +++ b/mm/swapfile.c
> @@ -1511,6 +1511,21 @@ int swp_swapcount(swp_entry_t entry)
>  	return count;
>  }
> =20
> +/*
> + * Verify that a swap entry has been tagged with SWAP_HAS_CACHE
> + */
> +bool swap_has_cache(struct swap_info_struct *si, swp_entry_t entry)
> +{
> +	pgoff_t offset =3D swp_offset(entry);
> +	struct swap_cluster_info *ci;
> +	bool has_cache;
> +
> +	ci =3D lock_cluster_or_swap_info(si, offset);
> +	has_cache =3D !!(si->swap_map[offset] & SWAP_HAS_CACHE);
> +	unlock_cluster_or_swap_info(si, ci);
> +	return has_cache;
> +}
> +
>  static bool swap_page_trans_huge_swapped(struct swap_info_struct *si,
>  					 swp_entry_t entry,
>  					 unsigned int nr_pages)