shinys000114/linux
1
0
Fork 0
mirror of https://github.com/hardkernel/linux.git synced 2026-06-06 19:08:57 +09:00
Code Issues Packages Projects Releases Wiki Activity

driver: rknpu: Add multiple power domain support

Browse Source 
Signed-off-by: Finley Xiao <finley.xiao@rock-chips.com>
Change-Id: I7d2db5b7e269fb62b2c17e6ec1d400b282684ff9
This commit is contained in:
Finley Xiao
2021-12-10 20:45:30 +08:00
committed by Tao Huang
parent a454f56b96
commit 3785bf3a30
2 changed files with 73 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
drivers/rknpu/include/rknpu_drv.h
View File

@@ -83,6 +83,10 @@ struct rknpu_device {
unsigned long current_volt;
int bypass_irq_handler;
int bypass_soft_reset;
struct device *genpd_dev_npu0;
struct device *genpd_dev_npu1;
struct device *genpd_dev_npu2;
bool multiple_domains;
};
#endif /* __LINUX_RKNPU_DRV_H_ */

69
drivers/rknpu/rknpu_drv.c
View File

@@ -27,6 +27,7 @@
#include <linux/wait.h>
#include <linux/sched.h>
#include <linux/clk.h>
#include <linux/pm_domain.h>
#include <linux/pm_runtime.h>
#include <linux/devfreq_cooling.h>
@@ -364,6 +365,41 @@ static int rknpu_power_on(struct rknpu_device *rknpu_dev)
return ret;
}
if (rknpu_dev->multiple_domains) {
if (rknpu_dev->genpd_dev_npu0) {
ret = pm_runtime_resume_and_get(
rknpu_dev->genpd_dev_npu0);
if (ret < 0) {
LOG_DEV_ERROR(
dev,
"failed to get pm runtime for npu0, ret = %d\n",
ret);
return ret;
}
}
if (rknpu_dev->genpd_dev_npu1) {
ret = pm_runtime_resume_and_get(
rknpu_dev->genpd_dev_npu1);
if (ret < 0) {
LOG_DEV_ERROR(
dev,
"failed to get pm runtime for npu1, ret = %d\n",
ret);
return ret;
}
}
if (rknpu_dev->genpd_dev_npu2) {
ret = pm_runtime_resume_and_get(
rknpu_dev->genpd_dev_npu2);
if (ret < 0) {
LOG_DEV_ERROR(
dev,
"failed to get pm runtime for npu2, ret = %d\n",
ret);
return ret;
}
}
}
ret = pm_runtime_get_sync(dev);
if (ret < 0) {
LOG_DEV_ERROR(dev,
@@ -381,6 +417,15 @@ static int rknpu_power_off(struct rknpu_device *rknpu_dev)
pm_runtime_put_sync(dev);
if (rknpu_dev->multiple_domains) {
if (rknpu_dev->genpd_dev_npu2)
pm_runtime_put_sync(rknpu_dev->genpd_dev_npu2);
if (rknpu_dev->genpd_dev_npu1)
pm_runtime_put_sync(rknpu_dev->genpd_dev_npu1);
if (rknpu_dev->genpd_dev_npu0)
pm_runtime_put_sync(rknpu_dev->genpd_dev_npu0);
}
clk_bulk_disable_unprepare(rknpu_dev->num_clks, rknpu_dev->clks);
regulator_disable(rknpu_dev->vdd);
@@ -638,6 +683,7 @@ static int rknpu_probe(struct platform_device *pdev)
struct resource *res = NULL;
struct rknpu_device *rknpu_dev = NULL;
struct device *dev = &pdev->dev;
struct device *virt_dev = NULL;
const struct of_device_id *match = NULL;
const struct rknpu_config *config = NULL;
int ret = -EINVAL;
@@ -770,6 +816,20 @@ static int rknpu_probe(struct platform_device *pdev)
pm_runtime_enable(dev);
if (of_count_phandle_with_args(dev->of_node, "power-domains",
"#power-domain-cells") > 1) {
virt_dev = dev_pm_domain_attach_by_name(dev, "npu0");
if (!IS_ERR(virt_dev))
rknpu_dev->genpd_dev_npu0 = virt_dev;
virt_dev = dev_pm_domain_attach_by_name(dev, "npu1");
if (!IS_ERR(virt_dev))
rknpu_dev->genpd_dev_npu1 = virt_dev;
virt_dev = dev_pm_domain_attach_by_name(dev, "npu2");
if (!IS_ERR(virt_dev))
rknpu_dev->genpd_dev_npu2 = virt_dev;
rknpu_dev->multiple_domains = true;
}
ret = rknpu_power_on(rknpu_dev);
if (ret)
goto err_free_fence_context;
@@ -797,6 +857,15 @@ static int rknpu_remove(struct platform_device *pdev)
rknpu_power_off(rknpu_dev);
if (rknpu_dev->multiple_domains) {
if (rknpu_dev->genpd_dev_npu0)
dev_pm_domain_detach(rknpu_dev->genpd_dev_npu0, true);
if (rknpu_dev->genpd_dev_npu1)
dev_pm_domain_detach(rknpu_dev->genpd_dev_npu1, true);
if (rknpu_dev->genpd_dev_npu2)
dev_pm_domain_detach(rknpu_dev->genpd_dev_npu2, true);
}
pm_runtime_disable(&pdev->dev);
return 0;