xref: /openbmc/linux/sound/soc/tegra/tegra210_amx.c (revision 1dd0dd0b)
1 // SPDX-License-Identifier: GPL-2.0-only
2 //
3 // tegra210_amx.c - Tegra210 AMX driver
4 //
5 // Copyright (c) 2021 NVIDIA CORPORATION.  All rights reserved.
6 
7 #include <linux/clk.h>
8 #include <linux/device.h>
9 #include <linux/io.h>
10 #include <linux/module.h>
11 #include <linux/of.h>
12 #include <linux/of_device.h>
13 #include <linux/platform_device.h>
14 #include <linux/pm_runtime.h>
15 #include <linux/regmap.h>
16 #include <sound/core.h>
17 #include <sound/pcm.h>
18 #include <sound/pcm_params.h>
19 #include <sound/soc.h>
20 
21 #include "tegra210_amx.h"
22 #include "tegra_cif.h"
23 
24 /*
25  * The counter is in terms of AHUB clock cycles. If a frame is not
26  * received within these clock cycles, the AMX input channel gets
27  * automatically disabled. For now the counter is calculated as a
28  * function of sample rate (8 kHz) and AHUB clock (49.152 MHz).
29  * If later an accurate number is needed, the counter needs to be
30  * calculated at runtime.
31  *
32  *     count = ahub_clk / sample_rate
33  */
34 #define TEGRA194_MAX_FRAME_IDLE_COUNT	0x1800
35 
36 #define AMX_CH_REG(id, reg) ((reg) + ((id) * TEGRA210_AMX_AUDIOCIF_CH_STRIDE))
37 
38 static const struct reg_default tegra210_amx_reg_defaults[] = {
39 	{ TEGRA210_AMX_RX_INT_MASK, 0x0000000f},
40 	{ TEGRA210_AMX_RX1_CIF_CTRL, 0x00007000},
41 	{ TEGRA210_AMX_RX2_CIF_CTRL, 0x00007000},
42 	{ TEGRA210_AMX_RX3_CIF_CTRL, 0x00007000},
43 	{ TEGRA210_AMX_RX4_CIF_CTRL, 0x00007000},
44 	{ TEGRA210_AMX_TX_INT_MASK, 0x00000001},
45 	{ TEGRA210_AMX_TX_CIF_CTRL, 0x00007000},
46 	{ TEGRA210_AMX_CG, 0x1},
47 	{ TEGRA210_AMX_CFG_RAM_CTRL, 0x00004000},
48 };
49 
50 static void tegra210_amx_write_map_ram(struct tegra210_amx *amx)
51 {
52 	int i;
53 
54 	regmap_write(amx->regmap, TEGRA210_AMX_CFG_RAM_CTRL,
55 		     TEGRA210_AMX_CFG_RAM_CTRL_SEQ_ACCESS_EN |
56 		     TEGRA210_AMX_CFG_RAM_CTRL_ADDR_INIT_EN |
57 		     TEGRA210_AMX_CFG_RAM_CTRL_RW_WRITE);
58 
59 	for (i = 0; i < TEGRA210_AMX_RAM_DEPTH; i++)
60 		regmap_write(amx->regmap, TEGRA210_AMX_CFG_RAM_DATA,
61 			     amx->map[i]);
62 
63 	regmap_write(amx->regmap, TEGRA210_AMX_OUT_BYTE_EN0, amx->byte_mask[0]);
64 	regmap_write(amx->regmap, TEGRA210_AMX_OUT_BYTE_EN1, amx->byte_mask[1]);
65 }
66 
67 static int tegra210_amx_startup(struct snd_pcm_substream *substream,
68 				struct snd_soc_dai *dai)
69 {
70 	struct tegra210_amx *amx = snd_soc_dai_get_drvdata(dai);
71 	unsigned int val;
72 	int err;
73 
74 	/* Ensure if AMX is disabled */
75 	err = regmap_read_poll_timeout(amx->regmap, TEGRA210_AMX_STATUS, val,
76 				       !(val & 0x1), 10, 10000);
77 	if (err < 0) {
78 		dev_err(dai->dev, "failed to stop AMX, err = %d\n", err);
79 		return err;
80 	}
81 
82 	/*
83 	 * Soft Reset: Below performs module soft reset which clears
84 	 * all FSM logic, flushes flow control of FIFO and resets the
85 	 * state register. It also brings module back to disabled
86 	 * state (without flushing the data in the pipe).
87 	 */
88 	regmap_update_bits(amx->regmap, TEGRA210_AMX_SOFT_RESET,
89 			   TEGRA210_AMX_SOFT_RESET_SOFT_RESET_MASK,
90 			   TEGRA210_AMX_SOFT_RESET_SOFT_EN);
91 
92 	err = regmap_read_poll_timeout(amx->regmap, TEGRA210_AMX_SOFT_RESET,
93 				       val, !(val & 0x1), 10, 10000);
94 	if (err < 0) {
95 		dev_err(dai->dev, "failed to reset AMX, err = %d\n", err);
96 		return err;
97 	}
98 
99 	return 0;
100 }
101 
102 static int __maybe_unused tegra210_amx_runtime_suspend(struct device *dev)
103 {
104 	struct tegra210_amx *amx = dev_get_drvdata(dev);
105 
106 	regcache_cache_only(amx->regmap, true);
107 	regcache_mark_dirty(amx->regmap);
108 
109 	return 0;
110 }
111 
112 static int __maybe_unused tegra210_amx_runtime_resume(struct device *dev)
113 {
114 	struct tegra210_amx *amx = dev_get_drvdata(dev);
115 
116 	regcache_cache_only(amx->regmap, false);
117 	regcache_sync(amx->regmap);
118 
119 	regmap_update_bits(amx->regmap,
120 		TEGRA210_AMX_CTRL,
121 		TEGRA210_AMX_CTRL_RX_DEP_MASK,
122 		TEGRA210_AMX_WAIT_ON_ANY << TEGRA210_AMX_CTRL_RX_DEP_SHIFT);
123 
124 	tegra210_amx_write_map_ram(amx);
125 
126 	return 0;
127 }
128 
129 static int tegra210_amx_set_audio_cif(struct snd_soc_dai *dai,
130 				      struct snd_pcm_hw_params *params,
131 				      unsigned int reg)
132 {
133 	struct tegra210_amx *amx = snd_soc_dai_get_drvdata(dai);
134 	int channels, audio_bits;
135 	struct tegra_cif_conf cif_conf;
136 
137 	memset(&cif_conf, 0, sizeof(struct tegra_cif_conf));
138 
139 	channels = params_channels(params);
140 
141 	switch (params_format(params)) {
142 	case SNDRV_PCM_FORMAT_S8:
143 		audio_bits = TEGRA_ACIF_BITS_8;
144 		break;
145 	case SNDRV_PCM_FORMAT_S16_LE:
146 		audio_bits = TEGRA_ACIF_BITS_16;
147 		break;
148 	case SNDRV_PCM_FORMAT_S32_LE:
149 		audio_bits = TEGRA_ACIF_BITS_32;
150 		break;
151 	default:
152 		return -EINVAL;
153 	}
154 
155 	cif_conf.audio_ch = channels;
156 	cif_conf.client_ch = channels;
157 	cif_conf.audio_bits = audio_bits;
158 	cif_conf.client_bits = audio_bits;
159 
160 	tegra_set_cif(amx->regmap, reg, &cif_conf);
161 
162 	return 0;
163 }
164 
165 static int tegra210_amx_in_hw_params(struct snd_pcm_substream *substream,
166 				     struct snd_pcm_hw_params *params,
167 				     struct snd_soc_dai *dai)
168 {
169 	struct tegra210_amx *amx = snd_soc_dai_get_drvdata(dai);
170 
171 	if (amx->soc_data->auto_disable) {
172 		regmap_write(amx->regmap,
173 			     AMX_CH_REG(dai->id, TEGRA194_AMX_RX1_FRAME_PERIOD),
174 			     TEGRA194_MAX_FRAME_IDLE_COUNT);
175 		regmap_write(amx->regmap, TEGRA210_AMX_CYA, 1);
176 	}
177 
178 	return tegra210_amx_set_audio_cif(dai, params,
179 			AMX_CH_REG(dai->id, TEGRA210_AMX_RX1_CIF_CTRL));
180 }
181 
182 static int tegra210_amx_out_hw_params(struct snd_pcm_substream *substream,
183 				      struct snd_pcm_hw_params *params,
184 				      struct snd_soc_dai *dai)
185 {
186 	return tegra210_amx_set_audio_cif(dai, params,
187 					  TEGRA210_AMX_TX_CIF_CTRL);
188 }
189 
190 static int tegra210_amx_get_byte_map(struct snd_kcontrol *kcontrol,
191 				     struct snd_ctl_elem_value *ucontrol)
192 {
193 	struct snd_soc_component *cmpnt = snd_soc_kcontrol_component(kcontrol);
194 	struct soc_mixer_control *mc =
195 		(struct soc_mixer_control *)kcontrol->private_value;
196 	struct tegra210_amx *amx = snd_soc_component_get_drvdata(cmpnt);
197 	unsigned char *bytes_map = (unsigned char *)&amx->map;
198 	int reg = mc->reg;
199 	int enabled;
200 
201 	if (reg > 31)
202 		enabled = amx->byte_mask[1] & (1 << (reg - 32));
203 	else
204 		enabled = amx->byte_mask[0] & (1 << reg);
205 
206 	if (enabled)
207 		ucontrol->value.integer.value[0] = bytes_map[reg];
208 	else
209 		ucontrol->value.integer.value[0] = 0;
210 
211 	return 0;
212 }
213 
214 static int tegra210_amx_put_byte_map(struct snd_kcontrol *kcontrol,
215 				     struct snd_ctl_elem_value *ucontrol)
216 {
217 	struct soc_mixer_control *mc =
218 		(struct soc_mixer_control *)kcontrol->private_value;
219 	struct snd_soc_component *cmpnt = snd_soc_kcontrol_component(kcontrol);
220 	struct tegra210_amx *amx = snd_soc_component_get_drvdata(cmpnt);
221 	unsigned char *bytes_map = (unsigned char *)&amx->map;
222 	int reg = mc->reg;
223 	int value = ucontrol->value.integer.value[0];
224 
225 	if (value == bytes_map[reg])
226 		return 0;
227 
228 	if (value >= 0 && value <= 255) {
229 		/* Update byte map and enable slot */
230 		bytes_map[reg] = value;
231 		if (reg > 31)
232 			amx->byte_mask[1] |= (1 << (reg - 32));
233 		else
234 			amx->byte_mask[0] |= (1 << reg);
235 	} else {
236 		/* Reset byte map and disable slot */
237 		bytes_map[reg] = 0;
238 		if (reg > 31)
239 			amx->byte_mask[1] &= ~(1 << (reg - 32));
240 		else
241 			amx->byte_mask[0] &= ~(1 << reg);
242 	}
243 
244 	return 1;
245 }
246 
247 static const struct snd_soc_dai_ops tegra210_amx_out_dai_ops = {
248 	.hw_params	= tegra210_amx_out_hw_params,
249 	.startup	= tegra210_amx_startup,
250 };
251 
252 static const struct snd_soc_dai_ops tegra210_amx_in_dai_ops = {
253 	.hw_params	= tegra210_amx_in_hw_params,
254 };
255 
256 #define IN_DAI(id)						\
257 	{							\
258 		.name = "AMX-RX-CIF" #id,			\
259 		.playback = {					\
260 			.stream_name = "RX" #id "-CIF-Playback",\
261 			.channels_min = 1,			\
262 			.channels_max = 16,			\
263 			.rates = SNDRV_PCM_RATE_8000_192000,	\
264 			.formats = SNDRV_PCM_FMTBIT_S8 |	\
265 				   SNDRV_PCM_FMTBIT_S16_LE |	\
266 				   SNDRV_PCM_FMTBIT_S32_LE,	\
267 		},						\
268 		.capture = {					\
269 			.stream_name = "RX" #id "-CIF-Capture",	\
270 			.channels_min = 1,			\
271 			.channels_max = 16,			\
272 			.rates = SNDRV_PCM_RATE_8000_192000,	\
273 			.formats = SNDRV_PCM_FMTBIT_S8 |	\
274 				   SNDRV_PCM_FMTBIT_S16_LE |	\
275 				   SNDRV_PCM_FMTBIT_S32_LE,	\
276 		},						\
277 		.ops = &tegra210_amx_in_dai_ops,		\
278 	}
279 
280 #define OUT_DAI							\
281 	{							\
282 		.name = "AMX-TX-CIF",				\
283 		.playback = {					\
284 			.stream_name = "TX-CIF-Playback",	\
285 			.channels_min = 1,			\
286 			.channels_max = 16,			\
287 			.rates = SNDRV_PCM_RATE_8000_192000,	\
288 			.formats = SNDRV_PCM_FMTBIT_S8 |	\
289 				   SNDRV_PCM_FMTBIT_S16_LE |	\
290 				   SNDRV_PCM_FMTBIT_S32_LE,	\
291 		},						\
292 		.capture = {					\
293 			.stream_name = "TX-CIF-Capture",	\
294 			.channels_min = 1,			\
295 			.channels_max = 16,			\
296 			.rates = SNDRV_PCM_RATE_8000_192000,	\
297 			.formats = SNDRV_PCM_FMTBIT_S8 |	\
298 				   SNDRV_PCM_FMTBIT_S16_LE |	\
299 				   SNDRV_PCM_FMTBIT_S32_LE,	\
300 		},						\
301 		.ops = &tegra210_amx_out_dai_ops,		\
302 	}
303 
304 static struct snd_soc_dai_driver tegra210_amx_dais[] = {
305 	IN_DAI(1),
306 	IN_DAI(2),
307 	IN_DAI(3),
308 	IN_DAI(4),
309 	OUT_DAI,
310 };
311 
312 static const struct snd_soc_dapm_widget tegra210_amx_widgets[] = {
313 	SND_SOC_DAPM_AIF_IN("RX1", NULL, 0, TEGRA210_AMX_CTRL, 0, 0),
314 	SND_SOC_DAPM_AIF_IN("RX2", NULL, 0, TEGRA210_AMX_CTRL, 1, 0),
315 	SND_SOC_DAPM_AIF_IN("RX3", NULL, 0, TEGRA210_AMX_CTRL, 2, 0),
316 	SND_SOC_DAPM_AIF_IN("RX4", NULL, 0, TEGRA210_AMX_CTRL, 3, 0),
317 	SND_SOC_DAPM_AIF_OUT("TX", NULL, 0, TEGRA210_AMX_ENABLE,
318 			     TEGRA210_AMX_ENABLE_SHIFT, 0),
319 };
320 
321 #define STREAM_ROUTES(id, sname)					  \
322 	{ "RX" #id " XBAR-" sname,	NULL,	"RX" #id " XBAR-TX" },	  \
323 	{ "RX" #id "-CIF-" sname,	NULL,	"RX" #id " XBAR-" sname },\
324 	{ "RX" #id,			NULL,	"RX" #id "-CIF-" sname }, \
325 	{ "TX",				NULL,	"RX" #id },		  \
326 	{ "TX-CIF-" sname,		NULL,	"TX" },			  \
327 	{ "XBAR-" sname,		NULL,	"TX-CIF-" sname },	  \
328 	{ "XBAR-RX",			NULL,	"XBAR-" sname }
329 
330 #define AMX_ROUTES(id)			\
331 	STREAM_ROUTES(id, "Playback"),	\
332 	STREAM_ROUTES(id, "Capture")
333 
334 static const struct snd_soc_dapm_route tegra210_amx_routes[] = {
335 	AMX_ROUTES(1),
336 	AMX_ROUTES(2),
337 	AMX_ROUTES(3),
338 	AMX_ROUTES(4),
339 };
340 
341 #define TEGRA210_AMX_BYTE_MAP_CTRL(reg)					\
342 	SOC_SINGLE_EXT("Byte Map " #reg, reg, 0, 256, 0,		\
343 		       tegra210_amx_get_byte_map,			\
344 		       tegra210_amx_put_byte_map)
345 
346 static struct snd_kcontrol_new tegra210_amx_controls[] = {
347 	TEGRA210_AMX_BYTE_MAP_CTRL(0),
348 	TEGRA210_AMX_BYTE_MAP_CTRL(1),
349 	TEGRA210_AMX_BYTE_MAP_CTRL(2),
350 	TEGRA210_AMX_BYTE_MAP_CTRL(3),
351 	TEGRA210_AMX_BYTE_MAP_CTRL(4),
352 	TEGRA210_AMX_BYTE_MAP_CTRL(5),
353 	TEGRA210_AMX_BYTE_MAP_CTRL(6),
354 	TEGRA210_AMX_BYTE_MAP_CTRL(7),
355 	TEGRA210_AMX_BYTE_MAP_CTRL(8),
356 	TEGRA210_AMX_BYTE_MAP_CTRL(9),
357 	TEGRA210_AMX_BYTE_MAP_CTRL(10),
358 	TEGRA210_AMX_BYTE_MAP_CTRL(11),
359 	TEGRA210_AMX_BYTE_MAP_CTRL(12),
360 	TEGRA210_AMX_BYTE_MAP_CTRL(13),
361 	TEGRA210_AMX_BYTE_MAP_CTRL(14),
362 	TEGRA210_AMX_BYTE_MAP_CTRL(15),
363 	TEGRA210_AMX_BYTE_MAP_CTRL(16),
364 	TEGRA210_AMX_BYTE_MAP_CTRL(17),
365 	TEGRA210_AMX_BYTE_MAP_CTRL(18),
366 	TEGRA210_AMX_BYTE_MAP_CTRL(19),
367 	TEGRA210_AMX_BYTE_MAP_CTRL(20),
368 	TEGRA210_AMX_BYTE_MAP_CTRL(21),
369 	TEGRA210_AMX_BYTE_MAP_CTRL(22),
370 	TEGRA210_AMX_BYTE_MAP_CTRL(23),
371 	TEGRA210_AMX_BYTE_MAP_CTRL(24),
372 	TEGRA210_AMX_BYTE_MAP_CTRL(25),
373 	TEGRA210_AMX_BYTE_MAP_CTRL(26),
374 	TEGRA210_AMX_BYTE_MAP_CTRL(27),
375 	TEGRA210_AMX_BYTE_MAP_CTRL(28),
376 	TEGRA210_AMX_BYTE_MAP_CTRL(29),
377 	TEGRA210_AMX_BYTE_MAP_CTRL(30),
378 	TEGRA210_AMX_BYTE_MAP_CTRL(31),
379 	TEGRA210_AMX_BYTE_MAP_CTRL(32),
380 	TEGRA210_AMX_BYTE_MAP_CTRL(33),
381 	TEGRA210_AMX_BYTE_MAP_CTRL(34),
382 	TEGRA210_AMX_BYTE_MAP_CTRL(35),
383 	TEGRA210_AMX_BYTE_MAP_CTRL(36),
384 	TEGRA210_AMX_BYTE_MAP_CTRL(37),
385 	TEGRA210_AMX_BYTE_MAP_CTRL(38),
386 	TEGRA210_AMX_BYTE_MAP_CTRL(39),
387 	TEGRA210_AMX_BYTE_MAP_CTRL(40),
388 	TEGRA210_AMX_BYTE_MAP_CTRL(41),
389 	TEGRA210_AMX_BYTE_MAP_CTRL(42),
390 	TEGRA210_AMX_BYTE_MAP_CTRL(43),
391 	TEGRA210_AMX_BYTE_MAP_CTRL(44),
392 	TEGRA210_AMX_BYTE_MAP_CTRL(45),
393 	TEGRA210_AMX_BYTE_MAP_CTRL(46),
394 	TEGRA210_AMX_BYTE_MAP_CTRL(47),
395 	TEGRA210_AMX_BYTE_MAP_CTRL(48),
396 	TEGRA210_AMX_BYTE_MAP_CTRL(49),
397 	TEGRA210_AMX_BYTE_MAP_CTRL(50),
398 	TEGRA210_AMX_BYTE_MAP_CTRL(51),
399 	TEGRA210_AMX_BYTE_MAP_CTRL(52),
400 	TEGRA210_AMX_BYTE_MAP_CTRL(53),
401 	TEGRA210_AMX_BYTE_MAP_CTRL(54),
402 	TEGRA210_AMX_BYTE_MAP_CTRL(55),
403 	TEGRA210_AMX_BYTE_MAP_CTRL(56),
404 	TEGRA210_AMX_BYTE_MAP_CTRL(57),
405 	TEGRA210_AMX_BYTE_MAP_CTRL(58),
406 	TEGRA210_AMX_BYTE_MAP_CTRL(59),
407 	TEGRA210_AMX_BYTE_MAP_CTRL(60),
408 	TEGRA210_AMX_BYTE_MAP_CTRL(61),
409 	TEGRA210_AMX_BYTE_MAP_CTRL(62),
410 	TEGRA210_AMX_BYTE_MAP_CTRL(63),
411 };
412 
413 static const struct snd_soc_component_driver tegra210_amx_cmpnt = {
414 	.dapm_widgets		= tegra210_amx_widgets,
415 	.num_dapm_widgets	= ARRAY_SIZE(tegra210_amx_widgets),
416 	.dapm_routes		= tegra210_amx_routes,
417 	.num_dapm_routes	= ARRAY_SIZE(tegra210_amx_routes),
418 	.controls		= tegra210_amx_controls,
419 	.num_controls		= ARRAY_SIZE(tegra210_amx_controls),
420 };
421 
422 static bool tegra210_amx_wr_reg(struct device *dev, unsigned int reg)
423 {
424 	switch (reg) {
425 	case TEGRA210_AMX_RX_INT_MASK ... TEGRA210_AMX_RX4_CIF_CTRL:
426 	case TEGRA210_AMX_TX_INT_MASK ... TEGRA210_AMX_CG:
427 	case TEGRA210_AMX_CTRL ... TEGRA210_AMX_CYA:
428 	case TEGRA210_AMX_CFG_RAM_CTRL ... TEGRA210_AMX_CFG_RAM_DATA:
429 		return true;
430 	default:
431 		return false;
432 	}
433 }
434 
435 static bool tegra194_amx_wr_reg(struct device *dev, unsigned int reg)
436 {
437 	switch (reg) {
438 	case TEGRA194_AMX_RX1_FRAME_PERIOD ... TEGRA194_AMX_RX4_FRAME_PERIOD:
439 		return true;
440 	default:
441 		return tegra210_amx_wr_reg(dev, reg);
442 	}
443 }
444 
445 static bool tegra210_amx_rd_reg(struct device *dev, unsigned int reg)
446 {
447 	switch (reg) {
448 	case TEGRA210_AMX_RX_STATUS ... TEGRA210_AMX_CFG_RAM_DATA:
449 		return true;
450 	default:
451 		return false;
452 	}
453 }
454 
455 static bool tegra194_amx_rd_reg(struct device *dev, unsigned int reg)
456 {
457 	switch (reg) {
458 	case TEGRA194_AMX_RX1_FRAME_PERIOD ... TEGRA194_AMX_RX4_FRAME_PERIOD:
459 		return true;
460 	default:
461 		return tegra210_amx_rd_reg(dev, reg);
462 	}
463 }
464 
465 static bool tegra210_amx_volatile_reg(struct device *dev, unsigned int reg)
466 {
467 	switch (reg) {
468 	case TEGRA210_AMX_RX_STATUS:
469 	case TEGRA210_AMX_RX_INT_STATUS:
470 	case TEGRA210_AMX_RX_INT_SET:
471 	case TEGRA210_AMX_TX_STATUS:
472 	case TEGRA210_AMX_TX_INT_STATUS:
473 	case TEGRA210_AMX_TX_INT_SET:
474 	case TEGRA210_AMX_SOFT_RESET:
475 	case TEGRA210_AMX_STATUS:
476 	case TEGRA210_AMX_INT_STATUS:
477 	case TEGRA210_AMX_CFG_RAM_CTRL:
478 	case TEGRA210_AMX_CFG_RAM_DATA:
479 		return true;
480 	default:
481 		break;
482 	}
483 
484 	return false;
485 }
486 
487 static const struct regmap_config tegra210_amx_regmap_config = {
488 	.reg_bits		= 32,
489 	.reg_stride		= 4,
490 	.val_bits		= 32,
491 	.max_register		= TEGRA210_AMX_CFG_RAM_DATA,
492 	.writeable_reg		= tegra210_amx_wr_reg,
493 	.readable_reg		= tegra210_amx_rd_reg,
494 	.volatile_reg		= tegra210_amx_volatile_reg,
495 	.reg_defaults		= tegra210_amx_reg_defaults,
496 	.num_reg_defaults	= ARRAY_SIZE(tegra210_amx_reg_defaults),
497 	.cache_type		= REGCACHE_FLAT,
498 };
499 
500 static const struct regmap_config tegra194_amx_regmap_config = {
501 	.reg_bits		= 32,
502 	.reg_stride		= 4,
503 	.val_bits		= 32,
504 	.max_register		= TEGRA194_AMX_RX4_LAST_FRAME_PERIOD,
505 	.writeable_reg		= tegra194_amx_wr_reg,
506 	.readable_reg		= tegra194_amx_rd_reg,
507 	.volatile_reg		= tegra210_amx_volatile_reg,
508 	.reg_defaults		= tegra210_amx_reg_defaults,
509 	.num_reg_defaults	= ARRAY_SIZE(tegra210_amx_reg_defaults),
510 	.cache_type		= REGCACHE_FLAT,
511 };
512 
513 static const struct tegra210_amx_soc_data soc_data_tegra210 = {
514 	.regmap_conf	= &tegra210_amx_regmap_config,
515 };
516 
517 static const struct tegra210_amx_soc_data soc_data_tegra194 = {
518 	.regmap_conf	= &tegra194_amx_regmap_config,
519 	.auto_disable	= true,
520 };
521 
522 static const struct of_device_id tegra210_amx_of_match[] = {
523 	{ .compatible = "nvidia,tegra210-amx", .data = &soc_data_tegra210 },
524 	{ .compatible = "nvidia,tegra194-amx", .data = &soc_data_tegra194 },
525 	{},
526 };
527 MODULE_DEVICE_TABLE(of, tegra210_amx_of_match);
528 
529 static int tegra210_amx_platform_probe(struct platform_device *pdev)
530 {
531 	struct device *dev = &pdev->dev;
532 	struct tegra210_amx *amx;
533 	void __iomem *regs;
534 	int err;
535 	const struct of_device_id *match;
536 	struct tegra210_amx_soc_data *soc_data;
537 
538 	match = of_match_device(tegra210_amx_of_match, dev);
539 
540 	soc_data = (struct tegra210_amx_soc_data *)match->data;
541 
542 	amx = devm_kzalloc(dev, sizeof(*amx), GFP_KERNEL);
543 	if (!amx)
544 		return -ENOMEM;
545 
546 	amx->soc_data = soc_data;
547 
548 	dev_set_drvdata(dev, amx);
549 
550 	regs = devm_platform_ioremap_resource(pdev, 0);
551 	if (IS_ERR(regs))
552 		return PTR_ERR(regs);
553 
554 	amx->regmap = devm_regmap_init_mmio(dev, regs,
555 					    soc_data->regmap_conf);
556 	if (IS_ERR(amx->regmap)) {
557 		dev_err(dev, "regmap init failed\n");
558 		return PTR_ERR(amx->regmap);
559 	}
560 
561 	regcache_cache_only(amx->regmap, true);
562 
563 	err = devm_snd_soc_register_component(dev, &tegra210_amx_cmpnt,
564 					      tegra210_amx_dais,
565 					      ARRAY_SIZE(tegra210_amx_dais));
566 	if (err) {
567 		dev_err(dev, "can't register AMX component, err: %d\n", err);
568 		return err;
569 	}
570 
571 	pm_runtime_enable(dev);
572 
573 	return 0;
574 }
575 
576 static int tegra210_amx_platform_remove(struct platform_device *pdev)
577 {
578 	pm_runtime_disable(&pdev->dev);
579 
580 	return 0;
581 }
582 
583 static const struct dev_pm_ops tegra210_amx_pm_ops = {
584 	SET_RUNTIME_PM_OPS(tegra210_amx_runtime_suspend,
585 			   tegra210_amx_runtime_resume, NULL)
586 	SET_SYSTEM_SLEEP_PM_OPS(pm_runtime_force_suspend,
587 				pm_runtime_force_resume)
588 };
589 
590 static struct platform_driver tegra210_amx_driver = {
591 	.driver = {
592 		.name = "tegra210-amx",
593 		.of_match_table = tegra210_amx_of_match,
594 		.pm = &tegra210_amx_pm_ops,
595 	},
596 	.probe = tegra210_amx_platform_probe,
597 	.remove = tegra210_amx_platform_remove,
598 };
599 module_platform_driver(tegra210_amx_driver);
600 
601 MODULE_AUTHOR("Songhee Baek <sbaek@nvidia.com>");
602 MODULE_DESCRIPTION("Tegra210 AMX ASoC driver");
603 MODULE_LICENSE("GPL v2");
604