clk-imx8mm: Move CLK_ENET_AXI to the non-SPL section
[platform/kernel/u-boot.git] / drivers / clk / imx / clk-imx8mm.c
1 // SPDX-License-Identifier: GPL-2.0
2 /*
3  * Copyright 2019 NXP
4  * Peng Fan <peng.fan@nxp.com>
5  */
6
7 #include <common.h>
8 #include <clk.h>
9 #include <clk-uclass.h>
10 #include <dm.h>
11 #include <log.h>
12 #include <asm/arch/clock.h>
13 #include <asm/arch/imx-regs.h>
14 #include <dt-bindings/clock/imx8mm-clock.h>
15
16 #include "clk.h"
17
18 static const char *pll_ref_sels[] = { "clock-osc-24m", "dummy", "dummy", "dummy", };
19 static const char *dram_pll_bypass_sels[] = {"dram_pll", "dram_pll_ref_sel", };
20 static const char *arm_pll_bypass_sels[] = {"arm_pll", "arm_pll_ref_sel", };
21 static const char *sys_pll1_bypass_sels[] = {"sys_pll1", "sys_pll1_ref_sel", };
22 static const char *sys_pll2_bypass_sels[] = {"sys_pll2", "sys_pll2_ref_sel", };
23 static const char *sys_pll3_bypass_sels[] = {"sys_pll3", "sys_pll3_ref_sel", };
24
25 static const char *imx8mm_a53_sels[] = {"clock-osc-24m", "arm_pll_out", "sys_pll2_500m", "sys_pll2_1000m",
26                                         "sys_pll1_800m", "sys_pll1_400m", "audio_pll1_out", "sys_pll3_out", };
27
28 static const char *imx8mm_ahb_sels[] = {"clock-osc-24m", "sys_pll1_133m", "sys_pll1_800m", "sys_pll1_400m",
29                                         "sys_pll2_125m", "sys_pll3_out", "audio_pll1_out", "video_pll1_out", };
30
31 #ifndef CONFIG_SPL_BUILD
32 static const char *imx8mm_enet_axi_sels[] = {"clock-osc-24m", "sys_pll1_266m", "sys_pll1_800m", "sys_pll2_250m",
33                                              "sys_pll2_200m", "audio_pll1_out", "video_pll1_out", "sys_pll3_out", };
34
35 static const char *imx8mm_enet_ref_sels[] = {"clock-osc-24m", "sys_pll2_125m", "sys_pll2_50m", "sys_pll2_100m",
36                                              "sys_pll1_160m", "audio_pll1_out", "video_pll1_out", "clk_ext4", };
37
38 static const char *imx8mm_enet_timer_sels[] = {"clock-osc-24m", "sys_pll2_100m", "audio_pll1_out", "clk_ext1", "clk_ext2",
39                                                "clk_ext3", "clk_ext4", "video_pll1_out", };
40
41 static const char *imx8mm_enet_phy_sels[] = {"clock-osc-24m", "sys_pll2_50m", "sys_pll2_125m", "sys_pll2_200m",
42                                              "sys_pll2_500m", "video_pll1_out", "audio_pll2_out", };
43 #endif
44
45 static const char *imx8mm_nand_usdhc_sels[] = {"clock-osc-24m", "sys_pll1_266m", "sys_pll1_800m", "sys_pll2_200m",
46                                                "sys_pll1_133m", "sys_pll3_out", "sys_pll2_250m", "audio_pll1_out", };
47
48 static const char *imx8mm_usb_bus_sels[] = {"clock-osc-24m", "sys_pll2_500m", "sys_pll1_800m", "sys_pll2_100m",
49                                             "sys_pll2_200m", "clk_ext2", "clk_ext4", "audio_pll2_out", };
50
51 static const char *imx8mm_usdhc1_sels[] = {"clock-osc-24m", "sys_pll1_400m", "sys_pll1_800m", "sys_pll2_500m",
52                                            "sys_pll3_out", "sys_pll1_266m", "audio_pll2_out", "sys_pll1_100m", };
53
54 static const char *imx8mm_usdhc2_sels[] = {"clock-osc-24m", "sys_pll1_400m", "sys_pll1_800m", "sys_pll2_500m",
55                                            "sys_pll3_out", "sys_pll1_266m", "audio_pll2_out", "sys_pll1_100m", };
56
57 static const char *imx8mm_i2c1_sels[] = {"clock-osc-24m", "sys_pll1_160m", "sys_pll2_50m", "sys_pll3_out", "audio_pll1_out",
58                                          "video_pll1_out", "audio_pll2_out", "sys_pll1_133m", };
59
60 static const char *imx8mm_i2c2_sels[] = {"clock-osc-24m", "sys_pll1_160m", "sys_pll2_50m", "sys_pll3_out", "audio_pll1_out",
61                                          "video_pll1_out", "audio_pll2_out", "sys_pll1_133m", };
62
63 static const char *imx8mm_i2c3_sels[] = {"clock-osc-24m", "sys_pll1_160m", "sys_pll2_50m", "sys_pll3_out", "audio_pll1_out",
64                                          "video_pll1_out", "audio_pll2_out", "sys_pll1_133m", };
65
66 static const char *imx8mm_i2c4_sels[] = {"clock-osc-24m", "sys_pll1_160m", "sys_pll2_50m", "sys_pll3_out", "audio_pll1_out",
67                                          "video_pll1_out", "audio_pll2_out", "sys_pll1_133m", };
68
69 #ifndef CONFIG_SPL_BUILD
70 static const char *imx8mm_pwm1_sels[] = {"clock-osc-24m", "sys_pll2_100m", "sys_pll1_160m", "sys_pll1_40m",
71                                          "sys_pll3_out", "clk_ext1", "sys_pll1_80m", "video_pll1_out", };
72
73 static const char *imx8mm_pwm2_sels[] = {"clock-osc-24m", "sys_pll2_100m", "sys_pll1_160m", "sys_pll1_40m",
74                                          "sys_pll3_out", "clk_ext1", "sys_pll1_80m", "video_pll1_out", };
75
76 static const char *imx8mm_pwm3_sels[] = {"clock-osc-24m", "sys_pll2_100m", "sys_pll1_160m", "sys_pll1_40m",
77                                          "sys_pll3_out", "clk_ext2", "sys_pll1_80m", "video_pll1_out", };
78
79 static const char *imx8mm_pwm4_sels[] = {"clock-osc-24m", "sys_pll2_100m", "sys_pll1_160m", "sys_pll1_40m",
80                                          "sys_pll3_out", "clk_ext2", "sys_pll1_80m", "video_pll1_out", };
81 #endif
82
83 static const char *imx8mm_wdog_sels[] = {"clock-osc-24m", "sys_pll1_133m", "sys_pll1_160m", "vpu_pll_out",
84                                          "sys_pll2_125m", "sys_pll3_out", "sys_pll1_80m", "sys_pll2_166m", };
85
86 static const char *imx8mm_usdhc3_sels[] = {"clock-osc-24m", "sys_pll1_400m", "sys_pll1_800m", "sys_pll2_500m",
87                                            "sys_pll3_out", "sys_pll1_266m", "audio_pll2_clk", "sys_pll1_100m", };
88
89 static const char *imx8mm_qspi_sels[] = {"clock-osc-24m", "sys_pll1_400m", "sys_pll2_333m", "sys_pll2_500m",
90                                            "audio_pll2_out", "sys_pll1_266m", "sys_pll3_out", "sys_pll1_100m", };
91
92 static const char *imx8mm_usb_core_sels[] = {"clock-osc-24m", "sys_pll1_100m", "sys_pll1_40m", "sys_pll2_100m",
93                                              "sys_pll2_200m", "clk_ext2", "clk_ext3", "audio_pll2_out", };
94
95 static const char *imx8mm_usb_phy_sels[] = {"clock-osc-24m", "sys_pll1_100m", "sys_pll1_40m", "sys_pll2_100m",
96                                              "sys_pll2_200m", "clk_ext2", "clk_ext3", "audio_pll2_out", };
97
98 static const char *imx8mm_ecspi1_sels[] = {"clock-osc-24m", "sys_pll2_200m", "sys_pll1_40m", "sys_pll1_160m",
99                                            "sys_pll1_800m", "sys_pll3_out", "sys_pll2_250m", "audio_pll2_out", };
100
101 static const char *imx8mm_ecspi2_sels[] = {"clock-osc-24m", "sys_pll2_200m", "sys_pll1_40m", "sys_pll1_160m",
102                                            "sys_pll1_800m", "sys_pll3_out", "sys_pll2_250m", "audio_pll2_out", };
103
104 static const char *imx8mm_ecspi3_sels[] = {"clock-osc-24m", "sys_pll2_200m", "sys_pll1_40m", "sys_pll1_160m",
105                                            "sys_pll1_800m", "sys_pll3_out", "sys_pll2_250m", "audio_pll2_out", };
106
107 static int imx8mm_clk_probe(struct udevice *dev)
108 {
109         void __iomem *base;
110
111         base = (void *)ANATOP_BASE_ADDR;
112
113         clk_dm(IMX8MM_DRAM_PLL_REF_SEL,
114                imx_clk_mux("dram_pll_ref_sel", base + 0x50, 0, 2,
115                            pll_ref_sels, ARRAY_SIZE(pll_ref_sels)));
116         clk_dm(IMX8MM_ARM_PLL_REF_SEL,
117                imx_clk_mux("arm_pll_ref_sel", base + 0x84, 0, 2,
118                            pll_ref_sels, ARRAY_SIZE(pll_ref_sels)));
119         clk_dm(IMX8MM_SYS_PLL1_REF_SEL,
120                imx_clk_mux("sys_pll1_ref_sel", base + 0x94, 0, 2,
121                            pll_ref_sels, ARRAY_SIZE(pll_ref_sels)));
122         clk_dm(IMX8MM_SYS_PLL2_REF_SEL,
123                imx_clk_mux("sys_pll2_ref_sel", base + 0x104, 0, 2,
124                            pll_ref_sels, ARRAY_SIZE(pll_ref_sels)));
125         clk_dm(IMX8MM_SYS_PLL3_REF_SEL,
126                imx_clk_mux("sys_pll3_ref_sel", base + 0x114, 0, 2,
127                            pll_ref_sels, ARRAY_SIZE(pll_ref_sels)));
128
129         clk_dm(IMX8MM_DRAM_PLL,
130                imx_clk_pll14xx("dram_pll", "dram_pll_ref_sel",
131                                base + 0x50, &imx_1443x_dram_pll));
132         clk_dm(IMX8MM_ARM_PLL,
133                imx_clk_pll14xx("arm_pll", "arm_pll_ref_sel",
134                                base + 0x84, &imx_1416x_pll));
135         clk_dm(IMX8MM_SYS_PLL1,
136                imx_clk_pll14xx("sys_pll1", "sys_pll1_ref_sel",
137                                base + 0x94, &imx_1416x_pll));
138         clk_dm(IMX8MM_SYS_PLL2,
139                imx_clk_pll14xx("sys_pll2", "sys_pll2_ref_sel",
140                                base + 0x104, &imx_1416x_pll));
141         clk_dm(IMX8MM_SYS_PLL3,
142                imx_clk_pll14xx("sys_pll3", "sys_pll3_ref_sel",
143                                base + 0x114, &imx_1416x_pll));
144
145         /* PLL bypass out */
146         clk_dm(IMX8MM_DRAM_PLL_BYPASS,
147                imx_clk_mux_flags("dram_pll_bypass", base + 0x50, 4, 1,
148                                  dram_pll_bypass_sels,
149                                  ARRAY_SIZE(dram_pll_bypass_sels),
150                                  CLK_SET_RATE_PARENT));
151         clk_dm(IMX8MM_ARM_PLL_BYPASS,
152                imx_clk_mux_flags("arm_pll_bypass", base + 0x84, 4, 1,
153                                  arm_pll_bypass_sels,
154                                  ARRAY_SIZE(arm_pll_bypass_sels),
155                                  CLK_SET_RATE_PARENT));
156         clk_dm(IMX8MM_SYS_PLL1_BYPASS,
157                imx_clk_mux_flags("sys_pll1_bypass", base + 0x94, 4, 1,
158                                  sys_pll1_bypass_sels,
159                                  ARRAY_SIZE(sys_pll1_bypass_sels),
160                                  CLK_SET_RATE_PARENT));
161         clk_dm(IMX8MM_SYS_PLL2_BYPASS,
162                imx_clk_mux_flags("sys_pll2_bypass", base + 0x104, 4, 1,
163                                  sys_pll2_bypass_sels,
164                                  ARRAY_SIZE(sys_pll2_bypass_sels),
165                                  CLK_SET_RATE_PARENT));
166         clk_dm(IMX8MM_SYS_PLL3_BYPASS,
167                imx_clk_mux_flags("sys_pll3_bypass", base + 0x114, 4, 1,
168                                  sys_pll3_bypass_sels,
169                                  ARRAY_SIZE(sys_pll3_bypass_sels),
170                                  CLK_SET_RATE_PARENT));
171
172         /* PLL out gate */
173         clk_dm(IMX8MM_DRAM_PLL_OUT,
174                imx_clk_gate("dram_pll_out", "dram_pll_bypass",
175                             base + 0x50, 13));
176         clk_dm(IMX8MM_ARM_PLL_OUT,
177                imx_clk_gate("arm_pll_out", "arm_pll_bypass",
178                             base + 0x84, 11));
179         clk_dm(IMX8MM_SYS_PLL1_OUT,
180                imx_clk_gate("sys_pll1_out", "sys_pll1_bypass",
181                             base + 0x94, 11));
182         clk_dm(IMX8MM_SYS_PLL2_OUT,
183                imx_clk_gate("sys_pll2_out", "sys_pll2_bypass",
184                             base + 0x104, 11));
185         clk_dm(IMX8MM_SYS_PLL3_OUT,
186                imx_clk_gate("sys_pll3_out", "sys_pll3_bypass",
187                             base + 0x114, 11));
188
189         /* SYS PLL fixed output */
190         clk_dm(IMX8MM_SYS_PLL1_40M,
191                imx_clk_fixed_factor("sys_pll1_40m", "sys_pll1_out", 1, 20));
192         clk_dm(IMX8MM_SYS_PLL1_80M,
193                imx_clk_fixed_factor("sys_pll1_80m", "sys_pll1_out", 1, 10));
194         clk_dm(IMX8MM_SYS_PLL1_100M,
195                imx_clk_fixed_factor("sys_pll1_100m", "sys_pll1_out", 1, 8));
196         clk_dm(IMX8MM_SYS_PLL1_133M,
197                imx_clk_fixed_factor("sys_pll1_133m", "sys_pll1_out", 1, 6));
198         clk_dm(IMX8MM_SYS_PLL1_160M,
199                imx_clk_fixed_factor("sys_pll1_160m", "sys_pll1_out", 1, 5));
200         clk_dm(IMX8MM_SYS_PLL1_200M,
201                imx_clk_fixed_factor("sys_pll1_200m", "sys_pll1_out", 1, 4));
202         clk_dm(IMX8MM_SYS_PLL1_266M,
203                imx_clk_fixed_factor("sys_pll1_266m", "sys_pll1_out", 1, 3));
204         clk_dm(IMX8MM_SYS_PLL1_400M,
205                imx_clk_fixed_factor("sys_pll1_400m", "sys_pll1_out", 1, 2));
206         clk_dm(IMX8MM_SYS_PLL1_800M,
207                imx_clk_fixed_factor("sys_pll1_800m", "sys_pll1_out", 1, 1));
208
209         clk_dm(IMX8MM_SYS_PLL2_50M,
210                imx_clk_fixed_factor("sys_pll2_50m", "sys_pll2_out", 1, 20));
211         clk_dm(IMX8MM_SYS_PLL2_100M,
212                imx_clk_fixed_factor("sys_pll2_100m", "sys_pll2_out", 1, 10));
213         clk_dm(IMX8MM_SYS_PLL2_125M,
214                imx_clk_fixed_factor("sys_pll2_125m", "sys_pll2_out", 1, 8));
215         clk_dm(IMX8MM_SYS_PLL2_166M,
216                imx_clk_fixed_factor("sys_pll2_166m", "sys_pll2_out", 1, 6));
217         clk_dm(IMX8MM_SYS_PLL2_200M,
218                imx_clk_fixed_factor("sys_pll2_200m", "sys_pll2_out", 1, 5));
219         clk_dm(IMX8MM_SYS_PLL2_250M,
220                imx_clk_fixed_factor("sys_pll2_250m", "sys_pll2_out", 1, 4));
221         clk_dm(IMX8MM_SYS_PLL2_333M,
222                imx_clk_fixed_factor("sys_pll2_333m", "sys_pll2_out", 1, 3));
223         clk_dm(IMX8MM_SYS_PLL2_500M,
224                imx_clk_fixed_factor("sys_pll2_500m", "sys_pll2_out", 1, 2));
225         clk_dm(IMX8MM_SYS_PLL2_1000M,
226                imx_clk_fixed_factor("sys_pll2_1000m", "sys_pll2_out", 1, 1));
227
228         base = dev_read_addr_ptr(dev);
229         if (!base)
230                 return -EINVAL;
231
232         clk_dm(IMX8MM_CLK_A53_SRC,
233                imx_clk_mux2("arm_a53_src", base + 0x8000, 24, 3,
234                             imx8mm_a53_sels, ARRAY_SIZE(imx8mm_a53_sels)));
235         clk_dm(IMX8MM_CLK_A53_CG,
236                imx_clk_gate3("arm_a53_cg", "arm_a53_src", base + 0x8000, 28));
237         clk_dm(IMX8MM_CLK_A53_DIV,
238                imx_clk_divider2("arm_a53_div", "arm_a53_cg",
239                                 base + 0x8000, 0, 3));
240
241         clk_dm(IMX8MM_CLK_AHB,
242                imx8m_clk_composite_critical("ahb", imx8mm_ahb_sels,
243                                             base + 0x9000));
244         clk_dm(IMX8MM_CLK_IPG_ROOT,
245                imx_clk_divider2("ipg_root", "ahb", base + 0x9080, 0, 1));
246
247         clk_dm(IMX8MM_CLK_NAND_USDHC_BUS,
248                imx8m_clk_composite_critical("nand_usdhc_bus",
249                                             imx8mm_nand_usdhc_sels,
250                                             base + 0x8900));
251         clk_dm(IMX8MM_CLK_USB_BUS,
252                 imx8m_clk_composite("usb_bus", imx8mm_usb_bus_sels, base + 0x8b80));
253
254         /* IP */
255         clk_dm(IMX8MM_CLK_USDHC1,
256                imx8m_clk_composite("usdhc1", imx8mm_usdhc1_sels,
257                                    base + 0xac00));
258         clk_dm(IMX8MM_CLK_USDHC2,
259                imx8m_clk_composite("usdhc2", imx8mm_usdhc2_sels,
260                                    base + 0xac80));
261         clk_dm(IMX8MM_CLK_I2C1,
262                imx8m_clk_composite("i2c1", imx8mm_i2c1_sels, base + 0xad00));
263         clk_dm(IMX8MM_CLK_I2C2,
264                imx8m_clk_composite("i2c2", imx8mm_i2c2_sels, base + 0xad80));
265         clk_dm(IMX8MM_CLK_I2C3,
266                imx8m_clk_composite("i2c3", imx8mm_i2c3_sels, base + 0xae00));
267         clk_dm(IMX8MM_CLK_I2C4,
268                imx8m_clk_composite("i2c4", imx8mm_i2c4_sels, base + 0xae80));
269         clk_dm(IMX8MM_CLK_WDOG,
270                imx8m_clk_composite("wdog", imx8mm_wdog_sels, base + 0xb900));
271         clk_dm(IMX8MM_CLK_USDHC3,
272                imx8m_clk_composite("usdhc3", imx8mm_usdhc3_sels,
273                                    base + 0xbc80));
274         clk_dm(IMX8MM_CLK_QSPI,
275                imx8m_clk_composite("qspi", imx8mm_qspi_sels, base + 0xab80));
276         clk_dm(IMX8MM_CLK_USB_CORE_REF,
277                 imx8m_clk_composite("usb_core_ref", imx8mm_usb_core_sels, base + 0xb100));
278         clk_dm(IMX8MM_CLK_USB_PHY_REF,
279                 imx8m_clk_composite("usb_phy_ref", imx8mm_usb_phy_sels, base + 0xb180));
280         clk_dm(IMX8MM_CLK_ECSPI1,
281                imx8m_clk_composite("ecspi1", imx8mm_ecspi1_sels, base + 0xb280));
282         clk_dm(IMX8MM_CLK_ECSPI2,
283                imx8m_clk_composite("ecspi2", imx8mm_ecspi2_sels, base + 0xb300));
284         clk_dm(IMX8MM_CLK_ECSPI3,
285                imx8m_clk_composite("ecspi3", imx8mm_ecspi3_sels, base + 0xc180));
286
287         clk_dm(IMX8MM_CLK_ECSPI1_ROOT,
288                imx_clk_gate4("ecspi1_root_clk", "ecspi1", base + 0x4070, 0));
289         clk_dm(IMX8MM_CLK_ECSPI2_ROOT,
290                imx_clk_gate4("ecspi2_root_clk", "ecspi2", base + 0x4080, 0));
291         clk_dm(IMX8MM_CLK_ECSPI3_ROOT,
292                imx_clk_gate4("ecspi3_root_clk", "ecspi3", base + 0x4090, 0));
293         clk_dm(IMX8MM_CLK_I2C1_ROOT,
294                imx_clk_gate4("i2c1_root_clk", "i2c1", base + 0x4170, 0));
295         clk_dm(IMX8MM_CLK_I2C2_ROOT,
296                imx_clk_gate4("i2c2_root_clk", "i2c2", base + 0x4180, 0));
297         clk_dm(IMX8MM_CLK_I2C3_ROOT,
298                imx_clk_gate4("i2c3_root_clk", "i2c3", base + 0x4190, 0));
299         clk_dm(IMX8MM_CLK_I2C4_ROOT,
300                imx_clk_gate4("i2c4_root_clk", "i2c4", base + 0x41a0, 0));
301         clk_dm(IMX8MM_CLK_OCOTP_ROOT,
302                imx_clk_gate4("ocotp_root_clk", "ipg_root", base + 0x4220, 0));
303         clk_dm(IMX8MM_CLK_USDHC1_ROOT,
304                imx_clk_gate4("usdhc1_root_clk", "usdhc1", base + 0x4510, 0));
305         clk_dm(IMX8MM_CLK_USDHC2_ROOT,
306                imx_clk_gate4("usdhc2_root_clk", "usdhc2", base + 0x4520, 0));
307         clk_dm(IMX8MM_CLK_WDOG1_ROOT,
308                imx_clk_gate4("wdog1_root_clk", "wdog", base + 0x4530, 0));
309         clk_dm(IMX8MM_CLK_WDOG2_ROOT,
310                imx_clk_gate4("wdog2_root_clk", "wdog", base + 0x4540, 0));
311         clk_dm(IMX8MM_CLK_WDOG3_ROOT,
312                imx_clk_gate4("wdog3_root_clk", "wdog", base + 0x4550, 0));
313         clk_dm(IMX8MM_CLK_USDHC3_ROOT,
314                imx_clk_gate4("usdhc3_root_clk", "usdhc3", base + 0x45e0, 0));
315         clk_dm(IMX8MM_CLK_QSPI_ROOT,
316                imx_clk_gate4("qspi_root_clk", "qspi", base + 0x42f0, 0));
317         clk_dm(IMX8MM_CLK_USB1_CTRL_ROOT,
318                 imx_clk_gate4("usb1_ctrl_root_clk", "usb_bus", base + 0x44d0, 0));
319
320         /* clks not needed in SPL stage */
321 #ifndef CONFIG_SPL_BUILD
322         clk_dm(IMX8MM_CLK_ENET_AXI,
323                imx8m_clk_composite("enet_axi", imx8mm_enet_axi_sels,
324                                    base + 0x8880));
325         clk_dm(IMX8MM_CLK_ENET_REF,
326                imx8m_clk_composite("enet_ref", imx8mm_enet_ref_sels,
327                base + 0xa980));
328         clk_dm(IMX8MM_CLK_ENET_TIMER,
329                imx8m_clk_composite("enet_timer", imx8mm_enet_timer_sels,
330                base + 0xaa00));
331         clk_dm(IMX8MM_CLK_ENET_PHY_REF,
332                imx8m_clk_composite("enet_phy", imx8mm_enet_phy_sels,
333                base + 0xaa80));
334         clk_dm(IMX8MM_CLK_ENET1_ROOT,
335                imx_clk_gate4("enet1_root_clk", "enet_axi",
336                base + 0x40a0, 0));
337         clk_dm(IMX8MM_CLK_PWM1,
338                imx8m_clk_composite("pwm1", imx8mm_pwm1_sels, base + 0xb380));
339         clk_dm(IMX8MM_CLK_PWM2,
340                imx8m_clk_composite("pwm2", imx8mm_pwm2_sels, base + 0xb400));
341         clk_dm(IMX8MM_CLK_PWM3,
342                imx8m_clk_composite("pwm3", imx8mm_pwm3_sels, base + 0xb480));
343         clk_dm(IMX8MM_CLK_PWM4,
344                imx8m_clk_composite("pwm4", imx8mm_pwm4_sels, base + 0xb500));
345         clk_dm(IMX8MM_CLK_PWM1_ROOT,
346                imx_clk_gate4("pwm1_root_clk", "pwm1", base + 0x4280, 0));
347         clk_dm(IMX8MM_CLK_PWM2_ROOT,
348                imx_clk_gate4("pwm2_root_clk", "pwm2", base + 0x4290, 0));
349         clk_dm(IMX8MM_CLK_PWM3_ROOT,
350                imx_clk_gate4("pwm3_root_clk", "pwm3", base + 0x42a0, 0));
351         clk_dm(IMX8MM_CLK_PWM4_ROOT,
352                imx_clk_gate4("pwm4_root_clk", "pwm4", base + 0x42b0, 0));
353 #endif
354
355         return 0;
356 }
357
358 static const struct udevice_id imx8mm_clk_ids[] = {
359         { .compatible = "fsl,imx8mm-ccm" },
360         { },
361 };
362
363 U_BOOT_DRIVER(imx8mm_clk) = {
364         .name = "clk_imx8mm",
365         .id = UCLASS_CLK,
366         .of_match = imx8mm_clk_ids,
367         .ops = &ccf_clk_ops,
368         .probe = imx8mm_clk_probe,
369         .flags = DM_FLAG_PRE_RELOC,
370 };