1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
|
/*
* Copyright (c) 2006-2024, RT-Thread Development Team
*
* SPDX-License-Identifier: Apache-2.0
*
* Change Logs:
* Date Author Notes
* 2024-5-1 GuEe-GUI first version
*/
#include <rtthread.h>
#include <rtdevice.h>
#define DBG_TAG "soc.k1x-tcm"
#define DBG_LVL DBG_INFO
#include <rtdbg.h>
#include <csr.h>
#include <spacemit.h>
static struct rt_memheap *tcm_heap[RT_CPUS_NR];
void *tcm_alloc_cpu(int cpuid, rt_size_t size)
{
if (cpuid >= 0 && cpuid < RT_ARRAY_SIZE(tcm_heap) && tcm_heap[cpuid])
{
return rt_memheap_alloc(tcm_heap[cpuid], size);
}
return RT_NULL;
}
void tcm_free_cpu(int cpuid, void *ptr)
{
if (cpuid >= 0 && cpuid < RT_ARRAY_SIZE(tcm_heap) && tcm_heap[cpuid])
{
rt_memheap_free(ptr);
}
}
static rt_err_t spacemit_tcm_probe(struct rt_platform_device *pdev)
{
struct rt_ofw_node *np = pdev->parent.ofw_node, *tcm_np;
csr_write(CSR_TCMCFG, TCM_EN);
rt_ofw_foreach_child_node(np, tcm_np)
{
int cpuid = 0;
void *tcm_base;
rt_uint64_t addr, size;
const char *string, *node_name;
if (rt_ofw_get_address(tcm_np, 0, &addr, &size))
{
continue;
}
string = node_name = rt_fdt_node_name(tcm_np->full_name);
string += sizeof("core") - 1;
while (*string >= '0' && *string <= '9')
{
cpuid *= 10;
cpuid += *string - '0';
++string;
}
if (cpuid < 0 || cpuid >= RT_ARRAY_SIZE(tcm_heap))
{
LOG_W("skip TCM node %s for CPU %d", node_name, cpuid);
continue;
}
tcm_heap[cpuid] = rt_calloc(1, sizeof(struct rt_memheap));
if (!tcm_heap[cpuid])
{
return -RT_ENOMEM;
}
if (!(tcm_base = rt_ioremap_cached((void *)addr, size)))
{
rt_free(tcm_heap[cpuid]);
tcm_heap[cpuid] = RT_NULL;
return -RT_EIO;
}
if (rt_memheap_init(tcm_heap[cpuid], node_name, tcm_base, size))
{
rt_free(tcm_heap[cpuid]);
tcm_heap[cpuid] = RT_NULL;
}
}
return RT_EOK;
}
static rt_err_t spacemit_tcm_shutdown(struct rt_platform_device *pdev)
{
csr_clear(CSR_TCMCFG, TCM_EN);
return RT_EOK;
}
static const struct rt_ofw_node_id spacemit_tcm_ofw_ids[] =
{
{ .compatible = "spacemit,k1-x-tcm" },
{ .compatible = "spacemit,k1-pro-tcm" },
{ /* sentinel */ }
};
static struct rt_platform_driver spacemit_tcm_driver =
{
.name = "spacemit-tcm",
.ids = spacemit_tcm_ofw_ids,
.probe = spacemit_tcm_probe,
.shutdown = spacemit_tcm_shutdown,
};
static int spacemit_tcm_drv_register(void)
{
rt_platform_driver_register(&spacemit_tcm_driver);
return 0;
}
INIT_SUBSYS_EXPORT(spacemit_tcm_drv_register);
|
