mirror of
https://git.kernel.org/pub/scm/linux/kernel/git/chenhuacai/linux-loongson
synced 2025-08-27 06:50:37 +00:00
b168ed458d
This code is based on the RDMA and misc cgroup initially, but now uses page_counter. It uses the same min/low/max semantics as the memory cgroup as a result. There's a small mismatch as TTM uses u64, and page_counter long pages. In practice it's not a problem. 32-bits systems don't really come with >=4GB cards and as long as we're consistently wrong with units, it's fine. The device page size may not be in the same units as kernel page size, and each region might also have a different page size (VRAM vs GART for example). The interface is simple: - Call dmem_cgroup_register_region() - Use dmem_cgroup_try_charge to check if you can allocate a chunk of memory, use dmem_cgroup__uncharge when freeing it. This may return an error code, or -EAGAIN when the cgroup limit is reached. In that case a reference to the limiting pool is returned. - The limiting cs can be used as compare function for dmem_cgroup_state_evict_valuable. - After having evicted enough, drop reference to limiting cs with dmem_cgroup_pool_state_put. This API allows you to limit device resources with cgroups. You can see the supported cards in /sys/fs/cgroup/dmem.capacity You need to echo +dmem to cgroup.subtree_control, and then you can partition device memory. Co-developed-by: Friedrich Vock <friedrich.vock@gmx.de> Signed-off-by: Friedrich Vock <friedrich.vock@gmx.de> Co-developed-by: Maxime Ripard <mripard@kernel.org> Signed-off-by: Maarten Lankhorst <dev@lankhorst.se> Acked-by: Tejun Heo <tj@kernel.org> Link: https://lore.kernel.org/r/20241204143112.1250983-1-dev@lankhorst.se Signed-off-by: Maxime Ripard <mripard@kernel.org>
67 lines
1.9 KiB
C
67 lines
1.9 KiB
C
/* SPDX-License-Identifier: MIT */
|
|
/*
|
|
* Copyright © 2023-2024 Intel Corporation
|
|
*/
|
|
|
|
#ifndef _CGROUP_DMEM_H
|
|
#define _CGROUP_DMEM_H
|
|
|
|
#include <linux/types.h>
|
|
#include <linux/llist.h>
|
|
|
|
struct dmem_cgroup_pool_state;
|
|
|
|
/* Opaque definition of a cgroup region, used internally */
|
|
struct dmem_cgroup_region;
|
|
|
|
#if IS_ENABLED(CONFIG_CGROUP_DMEM)
|
|
struct dmem_cgroup_region *dmem_cgroup_register_region(u64 size, const char *name_fmt, ...) __printf(2,3);
|
|
void dmem_cgroup_unregister_region(struct dmem_cgroup_region *region);
|
|
int dmem_cgroup_try_charge(struct dmem_cgroup_region *region, u64 size,
|
|
struct dmem_cgroup_pool_state **ret_pool,
|
|
struct dmem_cgroup_pool_state **ret_limit_pool);
|
|
void dmem_cgroup_uncharge(struct dmem_cgroup_pool_state *pool, u64 size);
|
|
bool dmem_cgroup_state_evict_valuable(struct dmem_cgroup_pool_state *limit_pool,
|
|
struct dmem_cgroup_pool_state *test_pool,
|
|
bool ignore_low, bool *ret_hit_low);
|
|
|
|
void dmem_cgroup_pool_state_put(struct dmem_cgroup_pool_state *pool);
|
|
#else
|
|
static inline __printf(2,3) struct dmem_cgroup_region *
|
|
dmem_cgroup_register_region(u64 size, const char *name_fmt, ...)
|
|
{
|
|
return NULL;
|
|
}
|
|
|
|
static inline void dmem_cgroup_unregister_region(struct dmem_cgroup_region *region)
|
|
{ }
|
|
|
|
static inline int dmem_cgroup_try_charge(struct dmem_cgroup_region *region, u64 size,
|
|
struct dmem_cgroup_pool_state **ret_pool,
|
|
struct dmem_cgroup_pool_state **ret_limit_pool)
|
|
{
|
|
*ret_pool = NULL;
|
|
|
|
if (ret_limit_pool)
|
|
*ret_limit_pool = NULL;
|
|
|
|
return 0;
|
|
}
|
|
|
|
static inline void dmem_cgroup_uncharge(struct dmem_cgroup_pool_state *pool, u64 size)
|
|
{ }
|
|
|
|
static inline
|
|
bool dmem_cgroup_state_evict_valuable(struct dmem_cgroup_pool_state *limit_pool,
|
|
struct dmem_cgroup_pool_state *test_pool,
|
|
bool ignore_low, bool *ret_hit_low)
|
|
{
|
|
return true;
|
|
}
|
|
|
|
static inline void dmem_cgroup_pool_state_put(struct dmem_cgroup_pool_state *pool)
|
|
{ }
|
|
|
|
#endif
|
|
#endif /* _CGROUP_DMEM_H */
|