162306a36Sopenharmony_ci% SPDX-License-Identifier: GPL-2.0
262306a36Sopenharmony_ci%
362306a36Sopenharmony_ci% run as: octave-cli memcg_protection.m
462306a36Sopenharmony_ci%
562306a36Sopenharmony_ci% This script simulates reclaim protection behavior on a single level of memcg
662306a36Sopenharmony_ci% hierarchy to illustrate how overcommitted protection spreads among siblings
762306a36Sopenharmony_ci% (as it depends also on their current consumption).
862306a36Sopenharmony_ci%
962306a36Sopenharmony_ci% Simulation assumes siblings consumed the initial amount of memory (w/out
1062306a36Sopenharmony_ci% reclaim) and then the reclaim starts, all memory is reclaimable, i.e. treated
1162306a36Sopenharmony_ci% same. It simulates only non-low reclaim and assumes all memory.min = 0.
1262306a36Sopenharmony_ci%
1362306a36Sopenharmony_ci% Input configurations
1462306a36Sopenharmony_ci% --------------------
1562306a36Sopenharmony_ci% E number	parent effective protection
1662306a36Sopenharmony_ci% n vector	nominal protection of siblings set at the given level (memory.low)
1762306a36Sopenharmony_ci% c vector	current consumption -,,- (memory.current)
1862306a36Sopenharmony_ci
1962306a36Sopenharmony_ci% example from testcase (values in GB)
2062306a36Sopenharmony_ciE = 50 / 1024;
2162306a36Sopenharmony_cin = [75 25 0 500 ] / 1024;
2262306a36Sopenharmony_cic = [50 50 50 0] / 1024;
2362306a36Sopenharmony_ci
2462306a36Sopenharmony_ci% Reclaim parameters
2562306a36Sopenharmony_ci% ------------------
2662306a36Sopenharmony_ci
2762306a36Sopenharmony_ci% Minimal reclaim amount (GB)
2862306a36Sopenharmony_cicluster = 32*4 / 2**20;
2962306a36Sopenharmony_ci
3062306a36Sopenharmony_ci% Reclaim coefficient (think as 0.5^sc->priority)
3162306a36Sopenharmony_cialpha = .1
3262306a36Sopenharmony_ci
3362306a36Sopenharmony_ci% Simulation parameters
3462306a36Sopenharmony_ci% ---------------------
3562306a36Sopenharmony_ciepsilon = 1e-7;
3662306a36Sopenharmony_citimeout = 1000;
3762306a36Sopenharmony_ci
3862306a36Sopenharmony_ci% Simulation loop
3962306a36Sopenharmony_ci% ---------------
4062306a36Sopenharmony_ci
4162306a36Sopenharmony_cich = [];
4262306a36Sopenharmony_cieh = [];
4362306a36Sopenharmony_cirh = [];
4462306a36Sopenharmony_ci
4562306a36Sopenharmony_cifor t = 1:timeout
4662306a36Sopenharmony_ci        % low_usage
4762306a36Sopenharmony_ci        u = min(c, n);
4862306a36Sopenharmony_ci        siblings = sum(u);
4962306a36Sopenharmony_ci
5062306a36Sopenharmony_ci        % effective_protection()
5162306a36Sopenharmony_ci        protected = min(n, c);                % start with nominal
5262306a36Sopenharmony_ci        e = protected * min(1, E / siblings); % normalize overcommit
5362306a36Sopenharmony_ci
5462306a36Sopenharmony_ci        % recursive protection
5562306a36Sopenharmony_ci        unclaimed = max(0, E - siblings);
5662306a36Sopenharmony_ci        parent_overuse = sum(c) - siblings;
5762306a36Sopenharmony_ci        if (unclaimed > 0 && parent_overuse > 0)
5862306a36Sopenharmony_ci                overuse = max(0, c - protected);
5962306a36Sopenharmony_ci                e += unclaimed * (overuse / parent_overuse);
6062306a36Sopenharmony_ci        endif
6162306a36Sopenharmony_ci
6262306a36Sopenharmony_ci        % get_scan_count()
6362306a36Sopenharmony_ci        r = alpha * c;             % assume all memory is in a single LRU list
6462306a36Sopenharmony_ci
6562306a36Sopenharmony_ci        % commit 1bc63fb1272b ("mm, memcg: make scan aggression always exclude protection")
6662306a36Sopenharmony_ci        sz = max(e, c);
6762306a36Sopenharmony_ci        r .*= (1 - (e+epsilon) ./ (sz+epsilon));
6862306a36Sopenharmony_ci
6962306a36Sopenharmony_ci        % uncomment to debug prints
7062306a36Sopenharmony_ci        % e, c, r
7162306a36Sopenharmony_ci
7262306a36Sopenharmony_ci        % nothing to reclaim, reached equilibrium
7362306a36Sopenharmony_ci        if max(r) < epsilon
7462306a36Sopenharmony_ci                break;
7562306a36Sopenharmony_ci        endif
7662306a36Sopenharmony_ci
7762306a36Sopenharmony_ci        % SWAP_CLUSTER_MAX roundup
7862306a36Sopenharmony_ci        r = max(r, (r > epsilon) .* cluster);
7962306a36Sopenharmony_ci        % XXX here I do parallel reclaim of all siblings
8062306a36Sopenharmony_ci        % in reality reclaim is serialized and each sibling recalculates own residual
8162306a36Sopenharmony_ci        c = max(c - r, 0);
8262306a36Sopenharmony_ci
8362306a36Sopenharmony_ci        ch = [ch ; c];
8462306a36Sopenharmony_ci        eh = [eh ; e];
8562306a36Sopenharmony_ci        rh = [rh ; r];
8662306a36Sopenharmony_ciendfor
8762306a36Sopenharmony_ci
8862306a36Sopenharmony_cit
8962306a36Sopenharmony_cic, e
90