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 C66EBC47DD9 for ; Wed, 27 Mar 2024 04:16:58 +0000 (UTC) Received: by kanga.kvack.org (Postfix) id 33E346B0083; Wed, 27 Mar 2024 00:16:56 -0400 (EDT) Received: by kanga.kvack.org (Postfix, from userid 40) id 2C7D06B0085; Wed, 27 Mar 2024 00:16:56 -0400 (EDT) X-Delivered-To: int-list-linux-mm@kvack.org Received: by kanga.kvack.org (Postfix, from userid 63042) id 0F31B6B0087; Wed, 27 Mar 2024 00:16:55 -0400 (EDT) X-Delivered-To: linux-mm@kvack.org Received: from relay.hostedemail.com (smtprelay0014.hostedemail.com [216.40.44.14]) by kanga.kvack.org (Postfix) with ESMTP id E814E6B0083 for ; Wed, 27 Mar 2024 00:16:55 -0400 (EDT) Received: from smtpin03.hostedemail.com (a10.router.float.18 [10.200.18.1]) by unirelay08.hostedemail.com (Postfix) with ESMTP id A27E51401C3 for ; Wed, 27 Mar 2024 04:16:55 +0000 (UTC) X-FDA: 81941508390.03.5A6835F Received: from mail-qt1-f174.google.com (mail-qt1-f174.google.com [209.85.160.174]) by imf26.hostedemail.com (Postfix) with ESMTP id C94C0140009 for ; Wed, 27 Mar 2024 04:16:53 +0000 (UTC) Authentication-Results: imf26.hostedemail.com; dkim=pass header.d=bytedance.com header.s=google header.b=f8lU3ISe; dmarc=pass (policy=quarantine) header.from=bytedance.com; spf=pass (imf26.hostedemail.com: domain of horenchuang@bytedance.com designates 209.85.160.174 as permitted sender) smtp.mailfrom=horenchuang@bytedance.com ARC-Message-Signature: i=1; a=rsa-sha256; c=relaxed/relaxed; d=hostedemail.com; s=arc-20220608; t=1711513013; 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-transfer-encoding:content-transfer-encoding: in-reply-to:in-reply-to:references:references:dkim-signature; bh=QB9vPuIKSQgIpOAN1iO1OyFVlWoszt4mBas5cObYBt0=; b=HUUlUq/GA0bKxZx9PnWN/CONhfy20kih463OsBADXYUd2m9NPkT38x8/fNMlJZ866XxErc 8bTDH6juXZzeD5JKC/GSqrc6JhqwqvGSBbkACqCX1hj6hi1IJpyvJ+z1ExTq7ICqEH+1WZ Aoo4HL9Z6hp4ewpm1WHzTarPsg4f2cE= ARC-Authentication-Results: i=1; imf26.hostedemail.com; dkim=pass header.d=bytedance.com header.s=google header.b=f8lU3ISe; dmarc=pass (policy=quarantine) header.from=bytedance.com; spf=pass (imf26.hostedemail.com: domain of horenchuang@bytedance.com designates 209.85.160.174 as permitted sender) smtp.mailfrom=horenchuang@bytedance.com ARC-Seal: i=1; s=arc-20220608; d=hostedemail.com; t=1711513013; a=rsa-sha256; cv=none; b=gYvICrHvsE8lfz5GN6QUlKleLrUhwRaeBy2bnaukaFmqkAJcYeIdcnOtgfqKGdsz4x6FTa GQqfBA3jt8hGMYA+RbUP2F3b1Dc1Aw93/GGsdisosE6EPyZrKWTQZ7kf2qJ4Xt8IVWKP2G 6eXcwbmuIniRPiQ2tZKknXE62vaDJnQ= Received: by mail-qt1-f174.google.com with SMTP id d75a77b69052e-430bf84977dso42400081cf.1 for ; Tue, 26 Mar 2024 21:16:53 -0700 (PDT) DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d=bytedance.com; s=google; t=1711513013; x=1712117813; 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=QB9vPuIKSQgIpOAN1iO1OyFVlWoszt4mBas5cObYBt0=; b=f8lU3ISe4S+e3Zlw26CSg/xEloUzmWxaksPsnfSbwg6BD7A3hLnD+/5CQpIx0cGcj6 ykavMgQ6TDoZ5dhhwK3KyTCCspGuv9mWI8GX/5uAM5ZL20/Q5i+TChI+xSePvgd8eqce MKqjbNDQ3gnNrM7kpihJLkYuxeG0IGAwW5Ioe3COuuUJJmOsa9DkczgQ1t+Hadpx9Nqk uCfKLoD8ffUA8pPikjYwsaUR9biOX7ICmMePBj3L0jtwEdmyn7pbIs4NSl8QnpXZj6sz tEVMGEuPQtyzYgAAyiHaLHjYBTJN2VTRHmNFO7O6xiO9G0ktWCWTIqyq0ugMFvim4TDT MWnA== X-Google-DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d=1e100.net; s=20230601; t=1711513013; x=1712117813; 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=QB9vPuIKSQgIpOAN1iO1OyFVlWoszt4mBas5cObYBt0=; b=Ly+1syl/hHRQVTcPrVevPx22GtzOxuUBdpvoGEIjLcWB7siwr8y8eI6196I28mo9yA p9DBL191HS4axU6r7wrl7zWqcuZlEZi79DM8YgQZ+1XuXFJtIqQ0rP9gJjj0VSuLEpP5 9gGQLITpCHR1UGb6mvujb+LrQvLy7+Jab08o2mqG3nP2FHEc8IrxxOMZiSaOS0GWma2H iDI138fo7h/Kik8qtwvvd6GKYdDFfaPmU86pqwH4O0yjYKzb1XMzcRouBU9HA2JUQNQL jM8F8o1UGv6Aq55MozWUD5NnlyqT88tu92MDmebVGRBtrvtmp3R3Lr77aQSZd2jnsey7 H1YQ== X-Forwarded-Encrypted: i=1; AJvYcCVMmNJjiuGHKcPVUN2Km+TemhiIMVHKjMPvUqUakP/Eab5hGQ4Y3qQH5cDMCyqCtzHwbEo/iExlC/gY9Nmmy32RyaE= X-Gm-Message-State: AOJu0YxjsIFBhYfMoYPesbxYhRLtL9vx7lHn7xkZdIkqAJEI4YQ2K74z SfJIpL6Q9z4+VCoom9ZIeq5RnOUJUEsctOXheFE7KhkvPSAdPh2dBaepNSBVvOw= X-Google-Smtp-Source: AGHT+IGro00ZV3fYTe7TE2hRIr5DsQebt1LEKkxDnTXu/2fbTJUhVHnBTE3BDQ8/xUarXeNtlbUQ4Q== X-Received: by 2002:ac8:7fc3:0:b0:430:bd04:ad6 with SMTP id b3-20020ac87fc3000000b00430bd040ad6mr3474694qtk.34.1711513012904; Tue, 26 Mar 2024 21:16:52 -0700 (PDT) Received: from n231-228-171.byted.org ([147.160.184.93]) by smtp.gmail.com with ESMTPSA id hb11-20020a05622a2b4b00b0043123c8b6a6sm4370696qtb.4.2024.03.26.21.16.52 (version=TLS1_3 cipher=TLS_AES_256_GCM_SHA384 bits=256/256); Tue, 26 Mar 2024 21:16:52 -0700 (PDT) From: "Ho-Ren (Jack) Chuang" To: "Huang, Ying" , "Gregory Price" , aneesh.kumar@linux.ibm.com, mhocko@suse.com, tj@kernel.org, john@jagalactic.com, "Eishan Mirakhur" , "Vinicius Tavares Petrucci" , "Ravis OpenSrc" , "Alistair Popple" , "Srinivasulu Thanneeru" , Dan Williams , Vishal Verma , Dave Jiang , Andrew Morton , nvdimm@lists.linux.dev, linux-cxl@vger.kernel.org, linux-kernel@vger.kernel.org, linux-mm@kvack.org Cc: "Ho-Ren (Jack) Chuang" , "Ho-Ren (Jack) Chuang" , "Ho-Ren (Jack) Chuang" , qemu-devel@nongnu.org, Hao Xiang Subject: [PATCH v5 2/2] memory tier: create CPUless memory tiers after obtaining HMAT info Date: Wed, 27 Mar 2024 04:16:46 +0000 Message-Id: <20240327041646.3258110-3-horenchuang@bytedance.com> X-Mailer: git-send-email 2.20.1 In-Reply-To: <20240327041646.3258110-1-horenchuang@bytedance.com> References: <20240327041646.3258110-1-horenchuang@bytedance.com> MIME-Version: 1.0 Content-Transfer-Encoding: 8bit X-Rspamd-Server: rspam09 X-Rspamd-Queue-Id: C94C0140009 X-Stat-Signature: xgk68jbkhp64h6a6aupb6yzp5zxjoh7t X-Rspam-User: X-HE-Tag: 1711513013-787814 X-HE-Meta: U2FsdGVkX1/CKtrvpyVd2rk1FDS2viu83bc1xmagtzbjKRKRQterbXA473QJQu410VPM1MFaF3B2g12CWR0IMuQ1jBfV5Sbh/TIERMMrjfsiwB1UenD6rBhHCl8ZQkMmOWd0Tnqmk/uUUP0oAKsVbVeYh4nKYKY5hdeZcmjlEPMEsfligC0pDogCeoOjL6gLCYiwvhnza+RDy/PN3OKfu1uG/+qo5vT++OVrmNZ9Wj3oiqP+Zs6VyBh4UYdos6gooonEEZU2FQcA+6H4hjiLcs0jT9WCA8fqxyff7XcvlXCyrd2pCjrWf3YkKGCWsYSG+V0ZLvCKN/AADLTwGEoD55w0KfIfLW+5AzINnbYnegbTXWHe+Na5vG5OrfYa6IwTtMY90RdLJGZ2Ih8iIsmcHzxYomoFfZFuY5lFMdGbHar8BlbtzNSeKQExjukrnGFbHTLeYe+lSsB2b7U6sNQznDd4gsjkhG/a6ghqdp8wZuev6DMdehHAPWP0myMijTshsW0rzFUuBzAXIlbocvW3c+rCTiyNyVD650JPGwFRqZ/iULHFvER58yCXFqyJj5vD4mVmYv8JXnAqndLPNgWIyw5C6pRtVq7KBs/EJfQTo7X3Nrdmo4A9oVx8F6Mc6P4L3H7mBAvSgXWi9lNafs0zXnExitSgWPm/R7sM6ra+Zc5KLqtuHXFHErhrUQ1dIGVFTjWeHl4/26uwi9iJ9T506ztBTQ4hnOavUeHMC0iZ4WFm5sIC4U6+MWVbVYL7qU7/OEH4pltJ3uYc7z9XxXvPcLpNHc3muaHSisAcC6BzhPd+pJM4nsy16bilwKT6OLEacDNTiw77Wf1vCdQ5G4YLrUmue7c6a3fLeD4tXGHuxcv08atbNHKJuAh7IqEb6Zm8JgDTQkSSI98moXu3VwuB2faShGE8DDs4D4vnLcea5aD7N5uf084ZVpcQeIYkrt2Lw2NZMQghM2xgCfpMDKr 1obinuxu YSzulG9WW9pMWc9jGvmGxg0bb9niiiPTDQdMgDfbdML96mRMAJwQo8pWc5/2jSQF9naPslK+VQLKb2BEzMgFTY5Fx3nkroo7SqufETVNvlKWC02lFJuc7koN9sT9yyJ3nF7I9aWM34X2cksGW7qxdXJW7LsZA/5wn3RbrtWd7obWuhwScxvz/Q45Cal/txCEc3whacJoawVJ3SmTfuooLZoLCdLcXQpNadkRCiX1Nyqk8J74eCySXuRd+M8K4igi5W0wz0sXV8fNSIuAljIUHk3kEQwAoG63ELdnuzV6tWSamNi46J9WcxbISDq/sxQxfd/xSRHFATSH2IRQsNXg6N4cWyjc4B/2mxAE+SzFfF+0B/A2adPT7tOWSE0CLW+SCV+J48JxYYtXL47pmuLR1mT3tm1hQ/SZp9sfMOOjlsby6vfSUUu8fiVkSKEogYqtZoanwRdfskhVK95KVaNSlLr6CL3sMGF2PXSte2LiRI5e2xs8= 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: The current implementation treats emulated memory devices, such as CXL1.1 type3 memory, as normal DRAM when they are emulated as normal memory (E820_TYPE_RAM). However, these emulated devices have different characteristics than traditional DRAM, making it important to distinguish them. Thus, we modify the tiered memory initialization process to introduce a delay specifically for CPUless NUMA nodes. This delay ensures that the memory tier initialization for these nodes is deferred until HMAT information is obtained during the boot process. Finally, demotion tables are recalculated at the end. * late_initcall(memory_tier_late_init); Some device drivers may have initialized memory tiers between `memory_tier_init()` and `memory_tier_late_init()`, potentially bringing online memory nodes and configuring memory tiers. They should be excluded in the late init. * Handle cases where there is no HMAT when creating memory tiers There is a scenario where a CPUless node does not provide HMAT information. If no HMAT is specified, it falls back to using the default DRAM tier. * Introduce another new lock `default_dram_perf_lock` for adist calculation In the current implementation, iterating through CPUlist nodes requires holding the `memory_tier_lock`. However, `mt_calc_adistance()` will end up trying to acquire the same lock, leading to a potential deadlock. Therefore, we propose introducing a standalone `default_dram_perf_lock` to protect `default_dram_perf_*`. This approach not only avoids deadlock but also prevents holding a large lock simultaneously. Besides, this patch slightly shortens the time holding the lock by putting the lock closer to what it protects as well. * Upgrade `set_node_memory_tier` to support additional cases, including default DRAM, late CPUless, and hot-plugged initializations. To cover hot-plugged memory nodes, `mt_calc_adistance()` and `mt_find_alloc_memory_type()` are moved into `set_node_memory_tier()` to handle cases where memtype is not initialized and where HMAT information is available. * Introduce `default_memory_types` for those memory types that are not initialized by device drivers. Because late initialized memory and default DRAM memory need to be managed, a default memory type is created for storing all memory types that are not initialized by device drivers and as a fallback. * Fix a deadlock bug in `mt_perf_to_adistance` Because an error path was not handled properly in `mt_perf_to_adistance`, unlock before returning the error. Signed-off-by: Ho-Ren (Jack) Chuang Signed-off-by: Hao Xiang --- mm/memory-tiers.c | 85 +++++++++++++++++++++++++++++++++++++++-------- 1 file changed, 72 insertions(+), 13 deletions(-) diff --git a/mm/memory-tiers.c b/mm/memory-tiers.c index 974af10cfdd8..610db9581ba4 100644 --- a/mm/memory-tiers.c +++ b/mm/memory-tiers.c @@ -36,6 +36,11 @@ struct node_memory_type_map { static DEFINE_MUTEX(memory_tier_lock); static LIST_HEAD(memory_tiers); +/* + * The list is used to store all memory types that are not created + * by a device driver. + */ +static LIST_HEAD(default_memory_types); static struct node_memory_type_map node_memory_types[MAX_NUMNODES]; struct memory_dev_type *default_dram_type; @@ -108,6 +113,8 @@ static struct demotion_nodes *node_demotion __read_mostly; static BLOCKING_NOTIFIER_HEAD(mt_adistance_algorithms); +/* The lock is used to protect `default_dram_perf*` info and nid. */ +static DEFINE_MUTEX(default_dram_perf_lock); static bool default_dram_perf_error; static struct access_coordinate default_dram_perf; static int default_dram_perf_ref_nid = NUMA_NO_NODE; @@ -505,7 +512,8 @@ static inline void __init_node_memory_type(int node, struct memory_dev_type *mem static struct memory_tier *set_node_memory_tier(int node) { struct memory_tier *memtier; - struct memory_dev_type *memtype; + struct memory_dev_type *mtype = default_dram_type; + int adist = MEMTIER_ADISTANCE_DRAM; pg_data_t *pgdat = NODE_DATA(node); @@ -514,11 +522,20 @@ static struct memory_tier *set_node_memory_tier(int node) if (!node_state(node, N_MEMORY)) return ERR_PTR(-EINVAL); - __init_node_memory_type(node, default_dram_type); + mt_calc_adistance(node, &adist); + if (node_memory_types[node].memtype == NULL) { + mtype = mt_find_alloc_memory_type(adist, &default_memory_types); + if (IS_ERR(mtype)) { + mtype = default_dram_type; + pr_info("Failed to allocate a memory type. Fall back.\n"); + } + } - memtype = node_memory_types[node].memtype; - node_set(node, memtype->nodes); - memtier = find_create_memory_tier(memtype); + __init_node_memory_type(node, mtype); + + mtype = node_memory_types[node].memtype; + node_set(node, mtype->nodes); + memtier = find_create_memory_tier(mtype); if (!IS_ERR(memtier)) rcu_assign_pointer(pgdat->memtier, memtier); return memtier; @@ -655,6 +672,34 @@ void mt_put_memory_types(struct list_head *memory_types) } EXPORT_SYMBOL_GPL(mt_put_memory_types); +/* + * This is invoked via `late_initcall()` to initialize memory tiers for + * CPU-less memory nodes after driver initialization, which is + * expected to provide `adistance` algorithms. + */ +static int __init memory_tier_late_init(void) +{ + int nid; + + mutex_lock(&memory_tier_lock); + for_each_node_state(nid, N_MEMORY) + if (!node_state(nid, N_CPU) && + node_memory_types[nid].memtype == NULL) + /* + * Some device drivers may have initialized memory tiers + * between `memory_tier_init()` and `memory_tier_late_init()`, + * potentially bringing online memory nodes and + * configuring memory tiers. Exclude them here. + */ + set_node_memory_tier(nid); + + establish_demotion_targets(); + mutex_unlock(&memory_tier_lock); + + return 0; +} +late_initcall(memory_tier_late_init); + static void dump_hmem_attrs(struct access_coordinate *coord, const char *prefix) { pr_info( @@ -668,7 +713,6 @@ int mt_set_default_dram_perf(int nid, struct access_coordinate *perf, { int rc = 0; - mutex_lock(&memory_tier_lock); if (default_dram_perf_error) { rc = -EIO; goto out; @@ -680,6 +724,7 @@ int mt_set_default_dram_perf(int nid, struct access_coordinate *perf, goto out; } + mutex_lock(&default_dram_perf_lock); if (default_dram_perf_ref_nid == NUMA_NO_NODE) { default_dram_perf = *perf; default_dram_perf_ref_nid = nid; @@ -716,23 +761,26 @@ int mt_set_default_dram_perf(int nid, struct access_coordinate *perf, } out: - mutex_unlock(&memory_tier_lock); + mutex_unlock(&default_dram_perf_lock); return rc; } int mt_perf_to_adistance(struct access_coordinate *perf, int *adist) { + int rc = 0; + if (default_dram_perf_error) return -EIO; - if (default_dram_perf_ref_nid == NUMA_NO_NODE) - return -ENOENT; - if (perf->read_latency + perf->write_latency == 0 || perf->read_bandwidth + perf->write_bandwidth == 0) return -EINVAL; - mutex_lock(&memory_tier_lock); + mutex_lock(&default_dram_perf_lock); + if (default_dram_perf_ref_nid == NUMA_NO_NODE) { + rc = -ENOENT; + goto out; + } /* * The abstract distance of a memory node is in direct proportion to * its memory latency (read + write) and inversely proportional to its @@ -745,8 +793,10 @@ int mt_perf_to_adistance(struct access_coordinate *perf, int *adist) (default_dram_perf.read_latency + default_dram_perf.write_latency) * (default_dram_perf.read_bandwidth + default_dram_perf.write_bandwidth) / (perf->read_bandwidth + perf->write_bandwidth); - mutex_unlock(&memory_tier_lock); + mutex_unlock(&default_dram_perf_lock); +out: + mutex_unlock(&default_dram_perf_lock); return 0; } EXPORT_SYMBOL_GPL(mt_perf_to_adistance); @@ -858,7 +908,8 @@ static int __init memory_tier_init(void) * For now we can have 4 faster memory tiers with smaller adistance * than default DRAM tier. */ - default_dram_type = alloc_memory_type(MEMTIER_ADISTANCE_DRAM); + default_dram_type = mt_find_alloc_memory_type(MEMTIER_ADISTANCE_DRAM, + &default_memory_types); if (IS_ERR(default_dram_type)) panic("%s() failed to allocate default DRAM tier\n", __func__); @@ -868,6 +919,14 @@ static int __init memory_tier_init(void) * types assigned. */ for_each_node_state(node, N_MEMORY) { + if (!node_state(node, N_CPU)) + /* + * Defer memory tier initialization on CPUless numa nodes. + * These will be initialized after firmware and devices are + * initialized. + */ + continue; + memtier = set_node_memory_tier(node); if (IS_ERR(memtier)) /* -- Ho-Ren (Jack) Chuang