162306a36Sopenharmony_ci// SPDX-License-Identifier: GPL-2.0-only
262306a36Sopenharmony_ci/* Copyright(c) 2023 Huawei. All rights reserved. */
362306a36Sopenharmony_ci
462306a36Sopenharmony_ci#include <linux/device.h>
562306a36Sopenharmony_ci#include <linux/slab.h>
662306a36Sopenharmony_ci#include <linux/idr.h>
762306a36Sopenharmony_ci#include <cxlmem.h>
862306a36Sopenharmony_ci#include <pmu.h>
962306a36Sopenharmony_ci#include <cxl.h>
1062306a36Sopenharmony_ci#include "core.h"
1162306a36Sopenharmony_ci
1262306a36Sopenharmony_cistatic void cxl_pmu_release(struct device *dev)
1362306a36Sopenharmony_ci{
1462306a36Sopenharmony_ci	struct cxl_pmu *pmu = to_cxl_pmu(dev);
1562306a36Sopenharmony_ci
1662306a36Sopenharmony_ci	kfree(pmu);
1762306a36Sopenharmony_ci}
1862306a36Sopenharmony_ci
1962306a36Sopenharmony_ciconst struct device_type cxl_pmu_type = {
2062306a36Sopenharmony_ci	.name = "cxl_pmu",
2162306a36Sopenharmony_ci	.release = cxl_pmu_release,
2262306a36Sopenharmony_ci};
2362306a36Sopenharmony_ci
2462306a36Sopenharmony_cistatic void remove_dev(void *dev)
2562306a36Sopenharmony_ci{
2662306a36Sopenharmony_ci	device_unregister(dev);
2762306a36Sopenharmony_ci}
2862306a36Sopenharmony_ci
2962306a36Sopenharmony_ciint devm_cxl_pmu_add(struct device *parent, struct cxl_pmu_regs *regs,
3062306a36Sopenharmony_ci		     int assoc_id, int index, enum cxl_pmu_type type)
3162306a36Sopenharmony_ci{
3262306a36Sopenharmony_ci	struct cxl_pmu *pmu;
3362306a36Sopenharmony_ci	struct device *dev;
3462306a36Sopenharmony_ci	int rc;
3562306a36Sopenharmony_ci
3662306a36Sopenharmony_ci	pmu = kzalloc(sizeof(*pmu), GFP_KERNEL);
3762306a36Sopenharmony_ci	if (!pmu)
3862306a36Sopenharmony_ci		return -ENOMEM;
3962306a36Sopenharmony_ci
4062306a36Sopenharmony_ci	pmu->assoc_id = assoc_id;
4162306a36Sopenharmony_ci	pmu->index = index;
4262306a36Sopenharmony_ci	pmu->type = type;
4362306a36Sopenharmony_ci	pmu->base = regs->pmu;
4462306a36Sopenharmony_ci	dev = &pmu->dev;
4562306a36Sopenharmony_ci	device_initialize(dev);
4662306a36Sopenharmony_ci	device_set_pm_not_required(dev);
4762306a36Sopenharmony_ci	dev->parent = parent;
4862306a36Sopenharmony_ci	dev->bus = &cxl_bus_type;
4962306a36Sopenharmony_ci	dev->type = &cxl_pmu_type;
5062306a36Sopenharmony_ci	switch (pmu->type) {
5162306a36Sopenharmony_ci	case CXL_PMU_MEMDEV:
5262306a36Sopenharmony_ci		rc = dev_set_name(dev, "pmu_mem%d.%d", assoc_id, index);
5362306a36Sopenharmony_ci		break;
5462306a36Sopenharmony_ci	}
5562306a36Sopenharmony_ci	if (rc)
5662306a36Sopenharmony_ci		goto err;
5762306a36Sopenharmony_ci
5862306a36Sopenharmony_ci	rc = device_add(dev);
5962306a36Sopenharmony_ci	if (rc)
6062306a36Sopenharmony_ci		goto err;
6162306a36Sopenharmony_ci
6262306a36Sopenharmony_ci	return devm_add_action_or_reset(parent, remove_dev, dev);
6362306a36Sopenharmony_ci
6462306a36Sopenharmony_cierr:
6562306a36Sopenharmony_ci	put_device(&pmu->dev);
6662306a36Sopenharmony_ci	return rc;
6762306a36Sopenharmony_ci}
6862306a36Sopenharmony_ciEXPORT_SYMBOL_NS_GPL(devm_cxl_pmu_add, CXL);
69