Received-SPF: pass (google.com: domain of linux-kernel-owner@vger.kernel.org designates 2620:137:e000::1:20 as permitted sender) client-ip=2620:137:e000::1:20;
From:   "Liam R. Howlett" <Liam.Howlett@oracle.com>
To:     Andrew Morton <akpm@linux-foundation.org>
Cc:     maple-tree@lists.infradead.org, linux-mm@kvack.org,
        linux-kernel@vger.kernel.org,
        Peng Zhang <zhangpeng.00@bytedance.com>,
        "Liam R. Howlett" <Liam.Howlett@oracle.com>
Subject: [PATCH v2 01/16] maple_tree: Add benchmarking for mas_for_each
Date:   Mon, 12 Jun 2023 16:39:38 -0400
Message-Id: <20230612203953.2093911-2-Liam.Howlett@oracle.com>
In-Reply-To: <20230612203953.2093911-1-Liam.Howlett@oracle.com>
References: <20230612203953.2093911-1-Liam.Howlett@oracle.com>
Content-Transfer-Encoding: 8bit
Content-Type: text/plain
MIME-Version: 1.0
X-MS-Exchange-AntiSpam-MessageData-ChunkCount: 1
X-MS-Exchange-AntiSpam-MessageData-0: =?us-ascii?Q?PK24/H/bH4sp6g0lnYi8T04v+p9CYODDF4XNdnG2XAzUhWrQnaRXfEvbWvvw?=
 =?us-ascii?Q?Vvx7E0bkH9J3rCqb0C7exbW9DD2EMygv3HjppwM1y34FAeBMioMVl9wkjM0s?=
 =?us-ascii?Q?CAkaNx9wrNxlyqcPddgqrZs5d8Oo3ma13RB5KFCkGYaeWn/YcAzQhbpY3NrQ?=
 =?us-ascii?Q?fYaYwVqJAI4jcCXHgQhg5AhYKeNDbeDIHA86rGcmzY6Oi1yMTJogGqGxQlIj?=
 =?us-ascii?Q?jbVMVYqe1QZWctJfVJvXqfSlakSSjOj2kRK8kUXtbpH0VT6B42QpvOOv0Fbw?=
 =?us-ascii?Q?Zt0Smb+L3KBh7FBU8/zl4vfGB6t3HGDXgslKqZNQHx+XMZQbyx/sPYeerIj5?=
 =?us-ascii?Q?jKUzkwCbmd5S5YpoKyUpKxDkjFOxVX9elc8Le+crGzHLIa0fEEcJraOMeIed?=
 =?us-ascii?Q?G6OVvL836HzCiw9l+v56R61uR98F2TJ2rhZSyeExcuz36NBPTysVS5ncLw+Q?=
 =?us-ascii?Q?zgsF2PhxRNrwLt74WjFk7HXwUomJXOPyl2M1iXMad7ellvSlQxuJBMRvujyZ?=
 =?us-ascii?Q?aiY3w37GMVMcCbLkhIhTxmfFng8NAueIQmP/3MzF4lVnzuLRN6EVxj7HSdh9?=
 =?us-ascii?Q?wpwnzlkdMLTGvW2E3gwc32S7JVTP/2IjWcmcZPBbqjSNrX3XkU1a6v/WB/jo?=
 =?us-ascii?Q?/T3gagjrltwIY+XFKXipmhSyQYpycDQ+uwyjw7Xnlw2CF5WGD/z7RY8yt2WT?=
 =?us-ascii?Q?v4b6SoiWgGyzBnlObSywy2X6x+yMWZAxjj5mELlrIUzNj1KAwXXuEVs/h1b6?=
 =?us-ascii?Q?Y0xSx04CcU6wx0eCD2hJVOwioysaffGzMDdQ3fZJMDGZXDNrCmLrdGmzZG7h?=
 =?us-ascii?Q?WzdFJvUY9q6FCmEH0MjhzGCrMinwhdwT8XJ72482qioe15oWkW4lwTmSXV6Z?=
 =?us-ascii?Q?yVg/xb2d2oLwpmXf9CWqJllmsm0SE2pMhYJ3ZUJbbxo0fvhH3jxDzsFD5JX7?=
 =?us-ascii?Q?lTKy4xOmYumQwclzIYo5tsJxc9QJAMEBAOOyvirT0D/yW58pLyNUQ6nSuyCJ?=
 =?us-ascii?Q?C1nzMaJNdY2ZyBelzhJiB8RJoi8AtNm4qu2SKp8TuGSA6kPUvBoYncUall3q?=
 =?us-ascii?Q?CjA9g8BjMxvJUozcwt+uos7wm2JPvzjo2Jfg4wFRO51xJ0Xh399O0BDrKcku?=
 =?us-ascii?Q?XIrOL1qxDoJo1hPbygQRuTC3nV9xdFyCqUp9DgT/s+vNjPzS7A9jk2Q7RIqq?=
 =?us-ascii?Q?SFoYn0vZjxMEOtMY2Wt1BlLkcBuaNj8Jbls2QAb0z/7o90qxYalsmrqU7PF+?=
 =?us-ascii?Q?hrgAkYeGlcHTvuVhbI1Yzv8WcB2PIy5bAZySkloq1eBE1fYgZKco7jTB+hbo?=
 =?us-ascii?Q?vlZBde7tWs4gPVDmD4UXAG50UCRVt+6i8COdnmPNIOVVW8rjXUZrVbrjplm2?=
 =?us-ascii?Q?Ls88qbEj8lIHhRZCm1wqjEhjT8QrNZQcKYUq8dklBh/Hbdb2qTorBYWICB7N?=
 =?us-ascii?Q?pwe0Tnt+ronR3hqkT8rABLxU3fyBaH0ei2ZkCKZeHU9E6vwJoNC8EdQLUVNB?=
 =?us-ascii?Q?C3FweTTgwSC6GdTdDKl7neA0C5ouPMe9KiA7SX9/kT0Ph4LTjOK79x7Mfvwl?=
 =?us-ascii?Q?mbetrUOY4cgbIPeoZ6DBNqSA+nM74oZqElWWFapJ6BRHjlKkifiyEmrtYAH8?=
 =?us-ascii?Q?tQ=3D=3D?=
X-MS-Exchange-AntiSpam-ExternalHop-MessageData-ChunkCount: 1
X-MS-Exchange-AntiSpam-ExternalHop-MessageData-0: =?us-ascii?Q?K3jREBT0wMYXRrQH6DzUPIZJu1ACqaHkoRAYjmoEpR4GFkUGek/bcbVdF4q+?=
 =?us-ascii?Q?g1sbVpnBm5wa9v7MZVG6cKkZlYg+Ok6fjhG+bzqpXQnEcV0VRlsYuj0KtPQy?=
 =?us-ascii?Q?I0/io8/s37dBtbValX/4BSz40Il+hFOKGjgsun7PbQF1o2OjHimndmMZ7sCS?=
 =?us-ascii?Q?O1p2ytBAevDdAi8TzF0Bd2enu4SzmlrlUPalTjm+lSgSD3Nhd4+1S48acIVQ?=
 =?us-ascii?Q?3faAqLxzesVpP87UX729jPVAZPbbfGSzUk6MwCnJpnOFtEhBvQN8fQLDZdpX?=
 =?us-ascii?Q?ObFvOLSkbKZQPBwSp0sADYSvVZKh6Wly0qY9zjxfSj9cl7s2Sb08AHW1OtpA?=
 =?us-ascii?Q?WrxMAha1pHfF63gCVtrnHHY9B/h4B83Opl5e0e/DMFPInEr/+mtLVWhlZ3BK?=
 =?us-ascii?Q?pQM/K/95Gx+VLdVL5gTmF31PPjUBcabEHAEJQslXJ049nFiUvkTxCe9/8aSP?=
 =?us-ascii?Q?qxM6lMHVHvvQGCXm+x10E4yiADUCrkIfAlvYVDoIXj6eu3GpO6j3RvKLtarn?=
 =?us-ascii?Q?FykcUGxZSLN1tjj2BOdpDDctjMwN1pFleEVndZmUZaKYdumBcvGMfXWpUwbo?=
 =?us-ascii?Q?9yGdnRUIQ7qPwK2/EjG7WMZpQhTDQOwywTmTa8uUBffo5s6LmDaNYRNw1Q7x?=
 =?us-ascii?Q?sUWbeKBF9M/8feeWPr3WmRdz1I9/++y9eMdhr9+u1dMi2YiX4c+9v9pbU3sI?=
 =?us-ascii?Q?yxzmOM80gzcDl41Sd3cD8sVGML5bIemJtT7lBgTQ/9VTtVplqUf53ajbf+pz?=
 =?us-ascii?Q?N31dEvtO1jbU85TeY9U3e1nKRouH6ztHaYivY9uAOpiLcKeuntJSWd0TxoPj?=
 =?us-ascii?Q?/2bm6GgRgIpp8lWQQ9DhJCAvxqmtBQGRln0Vf6NDIk9BoiBCuMWC7+E8qkS1?=
 =?us-ascii?Q?nJb20RdCqNoWY/MGMDwTy4vIPP7HoJj9DG0rZbfr50u5sHKIihxYDvbdpjS/?=
X-OriginatorOrg: oracle.com
X-MS-Exchange-CrossTenant-Network-Message-Id: ccbe0029-e4e2-47eb-a0b5-08db6b8531e4
X-MS-Exchange-CrossTenant-AuthSource: SN6PR10MB3022.namprd10.prod.outlook.com
X-MS-Exchange-CrossTenant-AuthAs: Internal
X-MS-Exchange-CrossTenant-OriginalArrivalTime: 12 Jun 2023 20:40:01.6074
 (UTC)
X-MS-Exchange-CrossTenant-FromEntityHeader: Hosted
X-MS-Exchange-CrossTenant-Id: 4e2c6054-71cb-48f1-bd6c-3a9705aca71b
X-MS-Exchange-CrossTenant-MailboxType: HOSTED
X-MS-Exchange-CrossTenant-UserPrincipalName: rcgMFxXNmcwNsU0/9zCTs/vzfM2zVBGf5qnHZO2izEGjlxBUnrt5TxjlRA88+ggyMKf5wnEUQOFxIc5f9g5Stw==
X-MS-Exchange-Transport-CrossTenantHeadersStamped: SA0PR10MB6425
X-Proofpoint-Virus-Version: vendor=baseguard
 engine=ICAP:2.0.254,Aquarius:18.0.957,Hydra:6.0.573,FMLib:17.11.176.26
 definitions=2023-06-12_15,2023-06-12_02,2023-05-22_02
X-Proofpoint-Spam-Details: rule=notspam policy=default score=0 malwarescore=0 suspectscore=0
 mlxscore=0 adultscore=0 spamscore=0 phishscore=0 bulkscore=0
 mlxlogscore=999 classifier=spam adjust=0 reason=mlx scancount=1
 engine=8.12.0-2305260000 definitions=main-2306120176
X-Proofpoint-GUID: bfNJLouZY3-BwEbTzzdisdQZgRFu3f59
X-Proofpoint-ORIG-GUID: bfNJLouZY3-BwEbTzzdisdQZgRFu3f59
X-Spam-Status: No, score=-2.8 required=5.0 tests=BAYES_00,DKIM_SIGNED,
        DKIM_VALID,DKIM_VALID_AU,DKIM_VALID_EF,RCVD_IN_DNSWL_LOW,
        RCVD_IN_MSPIKE_H5,RCVD_IN_MSPIKE_WL,SPF_HELO_NONE,SPF_NONE,
        T_SCC_BODY_TEXT_LINE autolearn=ham autolearn_force=no version=3.4.6
X-Spam-Checker-Version: SpamAssassin 3.4.6 (2021-04-09) on
        lindbergh.monkeyblade.net
Precedence: bulk
List-ID: <linux-kernel.vger.kernel.org>
X-Mailing-List: linux-kernel@vger.kernel.org

Add a way to test the speed of mas_for_each() to the testing code.

Signed-off-by: Liam R. Howlett <Liam.Howlett@oracle.com>
---
 lib/test_maple_tree.c | 38 ++++++++++++++++++++++++++++++++++++++
 1 file changed, 38 insertions(+)

diff --git a/lib/test_maple_tree.c b/lib/test_maple_tree.c
index 9939be34e516..3dbf99c3f2b1 100644
--- a/lib/test_maple_tree.c
+++ b/lib/test_maple_tree.c
@@ -44,6 +44,7 @@ atomic_t maple_tree_tests_passed;
 /* #define BENCH_WALK */
 /* #define BENCH_MT_FOR_EACH */
 /* #define BENCH_FORK */
+/* #define BENCH_MAS_FOR_EACH */
 
 #ifdef __KERNEL__
 #define mt_set_non_kernel(x)		do {} while (0)
@@ -1705,6 +1706,36 @@ static noinline void __init bench_mt_for_each(struct maple_tree *mt)
 }
 #endif
 
+#if defined(BENCH_MAS_FOR_EACH)
+static noinline void __init bench_mas_for_each(struct maple_tree *mt)
+{
+	int i, count = 1000000;
+	unsigned long max = 2500;
+	void *entry;
+	MA_STATE(mas, mt, 0, 0);
+
+	for (i = 0; i < max; i += 5) {
+		int gap = 4;
+		if (i % 30 == 0)
+			gap = 3;
+		mtree_store_range(mt, i, i + gap, xa_mk_value(i), GFP_KERNEL);
+	}
+
+	rcu_read_lock();
+	for (i = 0; i < count; i++) {
+		unsigned long j = 0;
+
+		mas_for_each(&mas, entry, max) {
+			MT_BUG_ON(mt, entry != xa_mk_value(j));
+			j += 5;
+		}
+		mas_set(&mas, 0);
+	}
+	rcu_read_unlock();
+
+}
+#endif
+
 /* check_forking - simulate the kernel forking sequence with the tree. */
 static noinline void __init check_forking(struct maple_tree *mt)
 {
@@ -3430,6 +3461,13 @@ static int __init maple_tree_seed(void)
 	mtree_destroy(&tree);
 	goto skip;
 #endif
+#if defined(BENCH_MAS_FOR_EACH)
+#define BENCH
+	mt_init_flags(&tree, MT_FLAGS_ALLOC_RANGE);
+	bench_mas_for_each(&tree);
+	mtree_destroy(&tree);
+	goto skip;
+#endif
 
 	mt_init_flags(&tree, MT_FLAGS_ALLOC_RANGE);
 	check_iteration(&tree);
-- 
2.39.2