LXR linux/drivers/infiniband/core/cgroup.c

   1// SPDX-License-Identifier: GPL-2.0-only
   2/*
   3 * Copyright (C) 2016 Parav Pandit <pandit.parav@gmail.com>
   4 */
   5
   6#include "core_priv.h"
   7
   8/**
   9 * ib_device_register_rdmacg - register with rdma cgroup.
  10 * @device: device to register to participate in resource
  11 *          accounting by rdma cgroup.
  12 *
  13 * Register with the rdma cgroup. Should be called before
  14 * exposing rdma device to user space applications to avoid
  15 * resource accounting leak.
  16 */
  17void ib_device_register_rdmacg(struct ib_device *device)
  18{
  19        device->cg_device.name = device->name;
  20        rdmacg_register_device(&device->cg_device);
  21}
  22
  23/**
  24 * ib_device_unregister_rdmacg - unregister with rdma cgroup.
  25 * @device: device to unregister.
  26 *
  27 * Unregister with the rdma cgroup. Should be called after
  28 * all the resources are deallocated, and after a stage when any
  29 * other resource allocation by user application cannot be done
  30 * for this device to avoid any leak in accounting.
  31 */
  32void ib_device_unregister_rdmacg(struct ib_device *device)
  33{
  34        rdmacg_unregister_device(&device->cg_device);
  35}
  36
  37int ib_rdmacg_try_charge(struct ib_rdmacg_object *cg_obj,
  38                         struct ib_device *device,
  39                         enum rdmacg_resource_type resource_index)
  40{
  41        return rdmacg_try_charge(&cg_obj->cg, &device->cg_device,
  42                                 resource_index);
  43}
  44EXPORT_SYMBOL(ib_rdmacg_try_charge);
  45
  46void ib_rdmacg_uncharge(struct ib_rdmacg_object *cg_obj,
  47                        struct ib_device *device,
  48                        enum rdmacg_resource_type resource_index)
  49{
  50        rdmacg_uncharge(cg_obj->cg, &device->cg_device,
  51                        resource_index);
  52}
  53EXPORT_SYMBOL(ib_rdmacg_uncharge);
  54