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,
        "Liam R. Howlett" <Liam.Howlett@oracle.com>
Subject: [PATCH 01/14] maple_tree: Add benchmarking for mas_for_each
Date:   Wed, 31 May 2023 22:15:52 -0400
Message-Id: <20230601021605.2823123-2-Liam.Howlett@oracle.com>
In-Reply-To: <20230601021605.2823123-1-Liam.Howlett@oracle.com>
References: <20230601021605.2823123-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?BHSEWj66j8mV+uDESMp/8VFkbYO+MV44bevas5Ha0mu9/iZqS2jHCJyEw7pN?=
 =?us-ascii?Q?G539ZHooJIvfwF7Jma+1GMhAytMZMiiUV3H/WRVuGZmOKM9g35d/iRzTiKK5?=
 =?us-ascii?Q?K4lK2RjR3CB6pDEiCVHTLPfBWvXee1aTRszuuFQBQlkZOdXjRgBAlF1aVODk?=
 =?us-ascii?Q?sv7xCycTkQ7EOSS6jeWSVa7gTC/aCEXwI6OSm86Zj7vl5GRLlclKiWIBZ7GS?=
 =?us-ascii?Q?CsPDs3xF33sQSsB9cA3NrBY0i0h23/mWrWIK95tfuVMx3hh1CPSLGS4TxeLB?=
 =?us-ascii?Q?bQsBf2u8GDcPVv6kOPXWf2jN/TM/ngI4i+rjDXLtgYxMkE8K7czhJCLR4grh?=
 =?us-ascii?Q?RFW+xnicc7jHrfiofWhAxUXcv/aFhB+G2XBL6tNRDMc5XcvLJlQJUzQJqDn0?=
 =?us-ascii?Q?UEOgoaGRhVrI3XG7jTP5HJlhSQ8cNIl9Pg/IbMyCfkGoKiccsGLQ0YzwolMG?=
 =?us-ascii?Q?xUpyAOKXAUCRUDjOKd46bfjShxH/1MrsxAE34YtxleiF8hloUnMa2Gl3JIYw?=
 =?us-ascii?Q?+oOO4Hzn0IMBaOyTtIqzb1zwlm2ZsOIUzGS+eIJuyLfgby6/cjXJM++0ks0h?=
 =?us-ascii?Q?n0jf9+dwxz4Poh/lXjFaiYeXKJb5CJTrV4qraEDCRwU/Aca0EUkiThh9kHvh?=
 =?us-ascii?Q?o8nl0c5QN3g2LMR+nxfYOJb1d+hejgAGbpGN4inpcjOJeilKTK+Y0Yc7XX1G?=
 =?us-ascii?Q?nd1FbrWzMsGg2hGXiyP0JP6+i2fyK/cD0ywzkgq15/rOfSgZ5k6/sCkvreS/?=
 =?us-ascii?Q?4BexgmN0dQAqV3hsJvd8Ex5bCK6pMyu0MXOnABFkpt0Hk6T2eqSoF24I3aG5?=
 =?us-ascii?Q?dCKA/9uT/bHkeOR+cQFBR2BXB8+XQX51wnGaMKePfsyY72pQ3etiTqV2Kk3M?=
 =?us-ascii?Q?1I/glPTnFPA+szqAUpAHFXfmBusOKKeTif3SMx8KlhpAjyZ/GAwwMOYDZGxW?=
 =?us-ascii?Q?YtkMRkZdemQEuxtEVTlYFXdwtbcZG/fQb3VJPg8LrfqrV5BvM8192+3yv7tH?=
 =?us-ascii?Q?BiPX9WbOSnJrpEpX0InyWwdUa6dcr9qAp0uC44sLRDBaFXlATkrsZkfxtTZj?=
 =?us-ascii?Q?Do1z65yavOj/HTP0b+HaKVRTQvvEWUZDYyDasAMg3kVIN0H5RqktpicrnONP?=
 =?us-ascii?Q?3Y0iBrCpF84+7D+xwYetfwpC11NllG2fz6o597B0KwCnhPorH5lUal/N7Oc+?=
 =?us-ascii?Q?bqzaDaf93csHpk1KOJU3KoPdeX5Xwb2xp7g17IKehJAvA3OAdaOpI2gjgn6D?=
 =?us-ascii?Q?4HvzH5JHdndHhsHPC32gccaRE8j3jBgdes8wjW9aDeS7VwVyLytpNCqsXwBZ?=
 =?us-ascii?Q?miioaGBA/7QcdsPeFcmr4qs5A1TIPMco6ym1bk60n20C8Yesgb3OyHdpSDaz?=
 =?us-ascii?Q?4860dRkk6P6peRZ+iMf7pvU7pjB4rnRadAkoTCOuOZwlPxeK1iqmtw2dRDzw?=
 =?us-ascii?Q?g25rx5nBMZ3PXxLCCzYtA5rfTpxX+j5oL4pZpXME6s+UXBWu3uO+67VAJlRw?=
 =?us-ascii?Q?Z42/FaFFITdq8undHQH8n+P6Aq0YZVumMVOcGi2KYbg1F+T2QA/45Mr15aXe?=
 =?us-ascii?Q?0xOV0DY+Wpjd0FsDyvWUeSbXYd0UOFbc/DncMoegcNOwXJ3oYuM2ARRb7Hk9?=
 =?us-ascii?Q?4Q=3D=3D?=
X-MS-Exchange-AntiSpam-ExternalHop-MessageData-ChunkCount: 1
X-MS-Exchange-AntiSpam-ExternalHop-MessageData-0: QeAagsZEiM0IDX1uQVxoNzeYmhgpxKG2STo4+DCvagDC4B8rlUuefPfKlzhdlohAiKcrfImbk9iMWz+19zZ5iLswIYpYV3YMSNsu9W8pymO1IishavyZ2DFJRvznNErk98k20UN1ndrM1A8Y57QmrOsXykEspXiGQG/TFz2cORe0vV9lHKseil03ueN2Jctbc0OyDVCjVDCsPmjnwS6y6v4HDZAQnmoR3YN5QgTmuGVmgwWAHajreopHp7RxsStgDyG6SvTH6Sh85ETUmS1H5eNLfzF1pZhyE8VIxADpSCSD/tOM7w8MikJuAoL7WmTK+61SAuIo1ksiPx3WNrhpbnuAec4NZ4BGNkLmE/T/QpXQs+B3GYaciihr0V/FmkqeYaTra7dPO2UkRRW6mUDRpIjTljSJqOrDLD6BhFGo/0bvRdwR+mvjrh7/sFUQy+6CK/XbODREC0wokXViO1FKBkVff/mLufyHAlJr9co1zNv9eLjOy5EVNgCDp7CLYdCH/K8KNek2GrDZkFLoKCQXKvQkMol+tG6fOHZSln1sjcJjovzOGvpCVKl+5jrhVMEL7toNmSyhAFlo15uhaUKhWfgA4r4FzZSu20i5gY+PHvSBEYHLPgjm6w8C6l7vIe3hssPTIj0Ha8RxmbUB8ZVaKAx301/kbmN8jxPVEKaXOPgsD6XZ3EEgVNUpXtAzqoSdzsaUVrAIkQFcpa7MGjGyYpPF2sYDSOmoQoltVYjU5kYhT/+SCQqPRQZVI1wCQLJRTzlzZU1F/sUA1zedqShiaq9H67Akj1YNNoNwKLPE+CzyNfDHKp+eaQ1uFqJUw7kCsxNpv8qtfjz0X9WUO5ynz8DmtKBmDjyLqG7SE2b1rZjomMtJF1ab+H9M6KeYcG8IHRFxzE9M7AJv1RyNTjsTcg==
X-OriginatorOrg: oracle.com
X-MS-Exchange-CrossTenant-Network-Message-Id: 21388493-825e-42de-318a-08db62462faa
X-MS-Exchange-CrossTenant-AuthSource: SN6PR10MB3022.namprd10.prod.outlook.com
X-MS-Exchange-CrossTenant-AuthAs: Internal
X-MS-Exchange-CrossTenant-OriginalArrivalTime: 01 Jun 2023 02:16:19.1550
 (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: DzrGHif99CZjSZUb65ZQ6wrh3HUuA9OtTd/84ewcUJqg+EKyNtpwMp5Mm7m9012+ot8sqoG23iYuBOvjEovKeg==
X-MS-Exchange-Transport-CrossTenantHeadersStamped: SJ0PR10MB4687
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-05-31_18,2023-05-31_03,2023-05-22_02
X-Proofpoint-Spam-Details: rule=notspam policy=default score=0 suspectscore=0 adultscore=0 spamscore=0
 phishscore=0 malwarescore=0 mlxlogscore=999 mlxscore=0 bulkscore=0
 classifier=spam adjust=0 reason=mlx scancount=1 engine=8.12.0-2304280000
 definitions=main-2306010016
X-Proofpoint-GUID: -Uc8XSWebcAtuRfKkCAt5pLCsuA9bRo1
X-Proofpoint-ORIG-GUID: -Uc8XSWebcAtuRfKkCAt5pLCsuA9bRo1
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