From mboxrd@z Thu Jan 1 00:00:00 1970 Received: from NAM12-BN8-obe.outbound.protection.outlook.com (mail-bn8nam12on2070.outbound.protection.outlook.com [40.107.237.70]) (using TLSv1.2 with cipher ECDHE-RSA-AES256-GCM-SHA384 (256/256 bits)) (No client certificate requested) by smtp.subspace.kernel.org (Postfix) with ESMTPS id A4B2635949 for ; Sat, 25 Jan 2025 17:14:24 +0000 (UTC) Authentication-Results: smtp.subspace.kernel.org; arc=fail smtp.client-ip=40.107.237.70 ARC-Seal:i=2; a=rsa-sha256; d=subspace.kernel.org; s=arc-20240116; t=1737825266; cv=fail; b=WDGb60hbnmpMI61owcUMuIqQYAuFILznBGjAhIz/LemEqldS87YbzyngRi6JMAHRhtufX+hhFJTG2YVenD+5PTJMTcjWoi7CBMfJsyazZLrHEk9ee9UwPVViSmQ6sDRfdF0rhX50X/+K4PotnbTi1uXQKDbQB66TZ6jlZw8ck+Q= ARC-Message-Signature:i=2; a=rsa-sha256; d=subspace.kernel.org; s=arc-20240116; t=1737825266; c=relaxed/simple; bh=teAsk/8BSk7ZrwD8qYr3lCEDdOzGZHASaF/C75NlztE=; h=Date:From:To:Cc:Subject:Message-ID:References:Content-Type: Content-Disposition:In-Reply-To:MIME-Version; b=iaXGZE0Hju5/UEgYLdcoui9S2bflnyLDuBD8nRb1hrTDUGmGl5kHaYPH8HhgGEaQvENOV02WuNOxUPh15L8NujYybyj40Rz1s9ih0ErbKVjFcLrFBXouIo7Q9cvPGRKjq8o4OFjldOxRDeyMBNEq18dK7Rua30ZOAbiz+73XB5E= ARC-Authentication-Results:i=2; smtp.subspace.kernel.org; dmarc=pass (p=reject dis=none) header.from=nvidia.com; spf=fail smtp.mailfrom=nvidia.com; dkim=pass (2048-bit key) header.d=Nvidia.com header.i=@Nvidia.com header.b=gLxTYJAK; arc=fail smtp.client-ip=40.107.237.70 Authentication-Results: smtp.subspace.kernel.org; dmarc=pass (p=reject dis=none) header.from=nvidia.com Authentication-Results: smtp.subspace.kernel.org; spf=fail smtp.mailfrom=nvidia.com Authentication-Results: smtp.subspace.kernel.org; dkim=pass (2048-bit key) header.d=Nvidia.com header.i=@Nvidia.com header.b="gLxTYJAK" ARC-Seal: i=1; a=rsa-sha256; s=arcselector10001; d=microsoft.com; cv=none; b=FwV4zO9HBehOCpxkPGG1q8rhkSpg+TSHLJWtQ55STM/G8TRutex1HQ45o8j12sVdEsuiN2wAo4IVZ+//kAlPnUjQDwzmoV2dEUaiA/rHW8xzPE0sJeh7stpy8EDKxnKaM5X4/yWBem4XrPt8L+zzUocegu6bKLzS4OfkM597msNqurQRx3T3eJeUISXT5IcRtW4RndGtc/tgW8CYbZLxB4rEusLbCVlg6BBPOgmoqXuLh/qYRSlcRASuxwxbM5qgS+Cb461HoId3ehPdJgtrmWuCHcliqTaVe6zWqQdYxrDEiTqhPGzRGi8jIhjYS2VF8rSbmvexYefGZmbAXdweWg== ARC-Message-Signature: i=1; a=rsa-sha256; c=relaxed/relaxed; d=microsoft.com; s=arcselector10001; h=From:Date:Subject:Message-ID:Content-Type:MIME-Version:X-MS-Exchange-AntiSpam-MessageData-ChunkCount:X-MS-Exchange-AntiSpam-MessageData-0:X-MS-Exchange-AntiSpam-MessageData-1; bh=JkgsX46jVJLr6m2l8R+fLik6HPv/xgKBFrYRa8eRJ6c=; b=CWiNx0eLFHzREU1DIn/IlEbYrOUeHsgB1lCanwrxeWTrSyfU3epR2vGzWHhUjd9Eult5LeGEqjPOWC6iT+Bks7Y6FBoq/nufgTFfxQkV4es3sOVR8cm/AWFw9QQEnJwUCB7WFwgbmum9q+iqFUZ2nAujShWOF9cnAB48y+jJlzUlyGMNqd34lMlPdUG4LE4MZOGGUtUhaK5ab4N45uDpXVEqBX8HTv46hRXh7JbeQ46UhM0f+drX/N5yLG7aam2iLNEcfubJqa+d2SPmOWdtTe/KFkPcSpu8JFP2Gxw5Fpeh2O2vK2jugNkC76yFz559+oNpBhpPnVELuU04FrRYGQ== ARC-Authentication-Results: i=1; mx.microsoft.com 1; spf=pass smtp.mailfrom=nvidia.com; dmarc=pass action=none header.from=nvidia.com; dkim=pass header.d=nvidia.com; arc=none DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d=Nvidia.com; s=selector2; h=From:Date:Subject:Message-ID:Content-Type:MIME-Version:X-MS-Exchange-SenderADCheck; bh=JkgsX46jVJLr6m2l8R+fLik6HPv/xgKBFrYRa8eRJ6c=; b=gLxTYJAKxa7o8uEeXcb8RDy4wMamzDSxMLiezftrtEI5VPUbcAvHFZIIzsMRY/RJEgNSTalSdduK9EONE4ggWDCmYpVpp2VzIX9dqOJtKBp1GFkBGTyLW9TXqBEah1ooN/Eu2oJPtCeiswC0+ieW0yGuhzt5zLskSFgOBphYPUEf4OD0fxK5OIhzh54R/xSTYJVXvT3kRWgRNpKvP3bJcIP83CrMtuej/qDxcfgFFDjleArVYnw3KqStjq/zsUz9uKIzlcBE6v59nmmPydCbZacmkjIOBgYMvFrDFooxCfY58UNbc/jHH2wVrRko5Vg/fj5nUV7VqKMsmJO0x43JDQ== Authentication-Results: dkim=none (message not signed) header.d=none;dmarc=none action=none header.from=nvidia.com; Received: from CY5PR12MB6405.namprd12.prod.outlook.com (2603:10b6:930:3e::17) by DS0PR12MB6438.namprd12.prod.outlook.com (2603:10b6:8:ca::16) with Microsoft SMTP Server (version=TLS1_2, cipher=TLS_ECDHE_RSA_WITH_AES_256_GCM_SHA384) id 15.20.8377.17; Sat, 25 Jan 2025 17:14:21 +0000 Received: from CY5PR12MB6405.namprd12.prod.outlook.com ([fe80::2119:c96c:b455:53b5]) by CY5PR12MB6405.namprd12.prod.outlook.com ([fe80::2119:c96c:b455:53b5%4]) with mapi id 15.20.8377.009; Sat, 25 Jan 2025 17:14:21 +0000 Date: Sat, 25 Jan 2025 18:14:12 +0100 From: Andrea Righi To: Tejun Heo , David Vernet , Changwoo Min Cc: linux-kernel@vger.kernel.org Subject: [PATCH v2 sched_ext/for-6.14-fixes 1/2] tools/sched_ext: Add helper to check task migration state Message-ID: References: <20250125094138.207287-1-arighi@nvidia.com> <20250125094138.207287-2-arighi@nvidia.com> Content-Type: text/plain; charset=us-ascii Content-Disposition: inline In-Reply-To: <20250125094138.207287-2-arighi@nvidia.com> X-ClientProxiedBy: FR0P281CA0160.DEUP281.PROD.OUTLOOK.COM (2603:10a6:d10:b3::14) To CY5PR12MB6405.namprd12.prod.outlook.com (2603:10b6:930:3e::17) Precedence: bulk X-Mailing-List: linux-kernel@vger.kernel.org List-Id: List-Subscribe: List-Unsubscribe: MIME-Version: 1.0 X-MS-PublicTrafficType: Email X-MS-TrafficTypeDiagnostic: CY5PR12MB6405:EE_|DS0PR12MB6438:EE_ X-MS-Office365-Filtering-Correlation-Id: 6bf010cf-c8ad-4a76-b557-08dd3d63b59b X-MS-Exchange-SenderADCheck: 1 X-MS-Exchange-AntiSpam-Relay: 0 X-Microsoft-Antispam: BCL:0;ARA:13230040|366016|1800799024|376014; X-Microsoft-Antispam-Message-Info: =?us-ascii?Q?mwyl2+/y0NRxKvifNjqBIjGDnrU2woz4WMEFIqF7GdM6oOUoZ+VhjikhtPd6?= =?us-ascii?Q?GYzt5GZybPPivy6W/icK1gztYxqFl3takWCOXfEbjP3c+yQ65FBlNiPyghcE?= =?us-ascii?Q?tKixySbos22JWeevWUXM21N+AuG7WdPGGCsfyUEFEv/7eTNeV/lLm82QakCd?= =?us-ascii?Q?XwsFGauhJzFiTMGJpmni+MdATjElbrqPtnposgyRdfTunHZSsvQizbCNHRro?= =?us-ascii?Q?cjtotEOFOqvC5ca/TSR1bWCWL2Bd0ZMQiMeLJVdkmMfFY1sjZjVbJ7XPTiZ/?= =?us-ascii?Q?71Qn7NYjOyjhCzNSILfYaFu8XSlLKKS8BzskXNy5fQTBe27cGTL4ExlspE0J?= =?us-ascii?Q?OZjWbjJD3SvEFXgUj8dM4lGXEivJqR3jELvpqe5p6S6i2PuRgnE6PdhTvZUq?= =?us-ascii?Q?BK8fEQBlAtKhRcxgVnTn2jwzcwKHBx5sI36THKzQ1VHvc6qQEWSgNXClqrgR?= =?us-ascii?Q?59jJqvMRv14ELpO+gqtngGY3DZNRbvAo/K3fJnC6GMrTOpiUZ4k2Njm0WSFM?= =?us-ascii?Q?ADgdgB1JGiLFQbOH5m7SD4Br5QNLCYZMfof02LAmm5zALLUdX8jJ5g1NgUw/?= =?us-ascii?Q?Ukoxrm2i1voJof/14WEv5FCGob3VYtJNHA7+ytXz/rfCMuzQl3/DZqcuTD3M?= =?us-ascii?Q?AKjGVwfoqQ2Z3TbHMB5lUPoTUtsEYc8teG+uQ+BinecpFR9wynufPO9gfWzy?= =?us-ascii?Q?7uZWp1ga2wtKNWuoHKzEVBn/9GcZXQ3+3W6cY6J5BdvUuoZuFfSzFCVwfXi1?= =?us-ascii?Q?p8mKrgUK5v6zFYppzHL3KSiZ49NSZSyeEqR8AsgmnEyCcQ181BrD7ahgKLwS?= =?us-ascii?Q?gsUkpcv8Zj52EnHeUDXyD6GRhVybiksIyxu8H8+tTFxC/NIKgD/7rfKUs5Xq?= =?us-ascii?Q?j4FivR2mYEE/wnaaQktNdluBsl7QgiFPg+8HWdY/YFF0Rx41XxyR0Cyt9UGW?= =?us-ascii?Q?mNXZVNVQ7qxY9eJs+d/U+c3LEuZ8MI9eBStnvPGc4ejTfUJC5zKDerOJlMwn?= =?us-ascii?Q?3JWENWIESRyuU3imtB8a+rMaM021bVUNJ80EEI7OiCSN579a5t4yNGjThhIt?= =?us-ascii?Q?fwGob1R/hMpBw8X9OkHh5V7KkzE5znd5CHd/6hB3C7X5xhQ0aM2ChhgvrYCK?= =?us-ascii?Q?29oHiEEob1YBhdqhwSMZCTMDQ6PtdiPqGbla7Udic1EgcJK0xfF62hnQ41/+?= =?us-ascii?Q?lglKCbEQfZUhVPEL3+A9rvG4cJ6kCFDUGkmafbfgPxPYsapr8Dy4vhwRmrkp?= =?us-ascii?Q?DIZHmTNnTm99gBYEHB+rd1Op5g8sfjsK4hW87clFRCmkuVxknQtodmn6MAOy?= =?us-ascii?Q?9iZl8r7Bbd9+gidsoM+JTGlnbB4xWLt3ewYN6cpvRsUZvkqakNBE31SAhEWF?= =?us-ascii?Q?VwUZNeBkW24jU3Oem3eA8fqd63Mq?= X-Forefront-Antispam-Report: CIP:255.255.255.255;CTRY:;LANG:en;SCL:1;SRV:;IPV:NLI;SFV:NSPM;H:CY5PR12MB6405.namprd12.prod.outlook.com;PTR:;CAT:NONE;SFS:(13230040)(366016)(1800799024)(376014);DIR:OUT;SFP:1101; X-MS-Exchange-AntiSpam-MessageData-ChunkCount: 1 X-MS-Exchange-AntiSpam-MessageData-0: =?us-ascii?Q?hThm5vKA/oRwBIowkFea8CxqjI4RY0tD0Fg+jXOprw0049BjGfRe6ScdguQ3?= =?us-ascii?Q?4U0V4m44FdRjL7zgHzUWpGzB537zx0TTGjmknh5I76ilggwIVxkS2hkVVpP/?= =?us-ascii?Q?JFHicHg3mLqT+wWZ5GHe1qAZH7ta8haFUJg6tqvH/NSeW21y+/swEYbu/DpP?= =?us-ascii?Q?Jjfiulz9D5n7i9gWgWqVORYAc8902u+DhJ021TK6NwxVdVg78r80OCoRQxnj?= =?us-ascii?Q?AQdEZR6vdu6lljPu5pfq+cV+yYxYv+Bbtq7W4Tsk1bpKL7FiGDtqdpkOnjCJ?= =?us-ascii?Q?JNz4eOedXY2LBBjLuRYfybphhU5yTyIecI8siDxvhGcMyxwDUouJ6g7S6Dhw?= =?us-ascii?Q?lC/sYntz7eRWT+dwgyoJ2iUlnMRneKkXZ2HwfnkM0YjlQCDhTTBiyN8JK+wP?= =?us-ascii?Q?BWSfR6yYXLhFY/IvWY3l8QjS9fXAc5jtuyJTcK6IohUIIpaXwzG3Px3slDZo?= =?us-ascii?Q?puto9nHFprnTwpncucUV1JwotY3oMejanpmTSrbLRA0AFL2yKlFB1VOQD4hQ?= =?us-ascii?Q?J59kbJyDMVyb9SIvDEuArxpOHxj7AWS0vFKoNJyIeUeY+uYnTWpJzvhoiL9+?= =?us-ascii?Q?FfBQpm8KUg5t8zOCp7CqoJhz0maZPQjOJUbxGQAkzvaJAqH7gw7/5x/4kAuJ?= =?us-ascii?Q?X+YFAupN0ZWNSap/tvZycsRvcwm9KbrnUJ2SMhUJiV3THA8nv7qQyC8aXYd+?= =?us-ascii?Q?nnYuZFcy74RuqcOVImVyn9DV6ZXlwyaYq3RxmKC3jvejQhYw1vfsfXBqLIVb?= =?us-ascii?Q?4Dl67B0FAeqxfCyGoq7g47CxKdYQklpR88c8HAv0TWDlTTEKvojav8nTpHyJ?= =?us-ascii?Q?Ue/6kQ6ZVDuNOY66jopOVegwF2p4I0qTGWoaE5q8nTETZ4FytgzhhjqHSLei?= =?us-ascii?Q?ztRC5mcy/fVMNcGmXU0htLvpQEv4Nn+uFSd0qrgHoEdKUSWvglm4Y1mB023m?= =?us-ascii?Q?ZiIFILuHHkXyCTFPP3W9i1aB5mJ0T08fR9YeJS7EFx2byB+G0UaEh1l6vlbc?= =?us-ascii?Q?XunDlSrhbozgJ3BaOtoYhR9gBc/5qjdFhKb0wqhO074g1vthAnTKGWpavSkd?= =?us-ascii?Q?yarp/UBfnfklik3ZzMGUHupbFP5Q5C7OlGhI6gjxur4puOqH9FwUFy/vzWfU?= =?us-ascii?Q?iG7O9V/CrrzW4FXKTNhwcQ23lGSgSrjtz8QOmtYrG/jQPAja5phXpNMpTHYB?= =?us-ascii?Q?eyHpopKHexfp8YfgusBLq3RmRB4sxbVxxo8Ll0aZQ4M5jvEsZHULHBUITYzr?= =?us-ascii?Q?sF2w3IerdmMDgpOX8hBL9QzwHKTvjQmtoiEcvrBAnhtjueSjGrr1gKUtuNHS?= =?us-ascii?Q?IrSVLSWbuikLnO/IFFakaSvsmy8+8up1482EKUkyOkrTb/HH5gFJjN0RXPIS?= =?us-ascii?Q?u53DeTHoUbIzs1HxtOSf8ixlac0h82uI4McyudnbXsuXt1WrTP1Qmx2vpGK0?= =?us-ascii?Q?vaSEEpRuYqfl+U5vHR6y8UWcw9AcWWgwvYW2/bOH63xp5BjfGMd75z2W+Ktp?= =?us-ascii?Q?tQN+ojCZq17TBsK1pcMB89+qfj5hd04Eg1CrIa2FlhEMVcyQ/mFYhmJ/8CRp?= =?us-ascii?Q?ohi7s1Fr/ibPWxZVXarqv98BsdyCh12h/TbDYRg/?= X-OriginatorOrg: Nvidia.com X-MS-Exchange-CrossTenant-Network-Message-Id: 6bf010cf-c8ad-4a76-b557-08dd3d63b59b X-MS-Exchange-CrossTenant-AuthSource: CY5PR12MB6405.namprd12.prod.outlook.com X-MS-Exchange-CrossTenant-AuthAs: Internal X-MS-Exchange-CrossTenant-OriginalArrivalTime: 25 Jan 2025 17:14:21.5572 (UTC) X-MS-Exchange-CrossTenant-FromEntityHeader: Hosted X-MS-Exchange-CrossTenant-Id: 43083d15-7273-40c1-b7db-39efd9ccc17a X-MS-Exchange-CrossTenant-MailboxType: HOSTED X-MS-Exchange-CrossTenant-UserPrincipalName: flPk39Nd9PZseOaEBLP/sKKVahamEB3HykLlJhdCXaxExhVQXkOHrfgBwE+H5a+wueAUg0feN5ROknrIBmM6/w== X-MS-Exchange-Transport-CrossTenantHeadersStamped: DS0PR12MB6438 Introduce a new helper for BPF schedulers to determine whether a task can migrate or not (supporting both SMP and UP systems). Fixes: e9fe182772dc ("sched_ext: selftests/dsp_local_on: Fix sporadic failures") Signed-off-by: Andrea Righi --- tools/sched_ext/include/scx/common.bpf.h | 11 +++++++++++ 1 file changed, 11 insertions(+) ChangeLog v1 -> v2: - add missing inline diff --git a/tools/sched_ext/include/scx/common.bpf.h b/tools/sched_ext/include/scx/common.bpf.h index f3e15e9efa76..f254a39b86a5 100644 --- a/tools/sched_ext/include/scx/common.bpf.h +++ b/tools/sched_ext/include/scx/common.bpf.h @@ -404,6 +404,17 @@ static __always_inline const struct cpumask *cast_mask(struct bpf_cpumask *mask) return (const struct cpumask *)mask; } +/* + * Return true if task @p cannot migrate to a different CPU, false + * otherwise. + */ +static inline bool is_migration_disabled(const struct task_struct *p) +{ + if (bpf_core_field_exists(p->migration_disabled)) + return p->migration_disabled; + return false; +} + /* rcu */ void bpf_rcu_read_lock(void) __ksym; void bpf_rcu_read_unlock(void) __ksym; -- 2.48.1