diff mbox series

[3/5] gpio: mvebu: add pwm support for Armada 8K/7K

Message ID db0d6d619a0686eef1b15ca7409d73813440856f.1605694661.git.baruch@tkos.co.il
State New
Headers show
Series gpio: mvebu: Armada 8K/7K PWM support | expand

Commit Message

Baruch Siach Nov. 18, 2020, 10:30 a.m. UTC
Use the pwm-offset DT property to store the location of PWM signal
duration registers.

Since we have more than two GPIO chips per system, we can't use the
alias id to differentiate between them. Use the offset value for that.

Move mvebu_pwm_probe() call before irq support code. The AP80x does not
provide irq support, but does provide PWM. Don't skip PWM probe because
of that.

Signed-off-by: Baruch Siach <baruch@tkos.co.il>
---
 drivers/gpio/gpio-mvebu.c | 112 +++++++++++++++++++++++++-------------
 1 file changed, 75 insertions(+), 37 deletions(-)

Comments

Andrew Lunn Nov. 18, 2020, 11:18 p.m. UTC | #1
On Wed, Nov 18, 2020 at 12:30:44PM +0200, Baruch Siach wrote:
> Use the pwm-offset DT property to store the location of PWM signal

> duration registers.

> 

> Since we have more than two GPIO chips per system, we can't use the

> alias id to differentiate between them. Use the offset value for that.

> 

> Move mvebu_pwm_probe() call before irq support code. The AP80x does not

> provide irq support, but does provide PWM. Don't skip PWM probe because

> of that.

> 

> Signed-off-by: Baruch Siach <baruch@tkos.co.il>

> ---

>  drivers/gpio/gpio-mvebu.c | 112 +++++++++++++++++++++++++-------------

>  1 file changed, 75 insertions(+), 37 deletions(-)

> 

> diff --git a/drivers/gpio/gpio-mvebu.c b/drivers/gpio/gpio-mvebu.c

> index 946571e70928..8602afd21673 100644

> --- a/drivers/gpio/gpio-mvebu.c

> +++ b/drivers/gpio/gpio-mvebu.c

> @@ -70,7 +70,12 @@

>   */

>  #define PWM_BLINK_ON_DURATION_OFF	0x0

>  #define PWM_BLINK_OFF_DURATION_OFF	0x4

> +#define PWM_BLINK_COUNTER_B_OFF		0x8

>  

> +/* Armada 8k variant gpios register offsets */

> +#define AP80X_GPIO0_OFF_A8K		0x1040

> +#define CP11X_GPIO0_OFF_A8K		0x100

> +#define CP11X_GPIO1_OFF_A8K		0x140

>  

>  /* The MV78200 has per-CPU registers for edge mask and level mask */

>  #define GPIO_EDGE_MASK_MV78200_OFF(cpu)	  ((cpu) ? 0x30 : 0x18)

> @@ -93,6 +98,7 @@

>  

>  struct mvebu_pwm {

>  	struct regmap		*regs;

> +	u32			 offset;

>  	unsigned long		 clk_rate;

>  	struct gpio_desc	*gpiod;

>  	struct pwm_chip		 chip;

> @@ -283,12 +289,12 @@ mvebu_gpio_write_level_mask(struct mvebu_gpio_chip *mvchip, u32 val)

>   */

>  static unsigned int mvebu_pwmreg_blink_on_duration(struct mvebu_pwm *mvpwm)

>  {

> -	return PWM_BLINK_ON_DURATION_OFF;

> +	return mvpwm->offset + PWM_BLINK_ON_DURATION_OFF;

>  }

>  

>  static unsigned int mvebu_pwmreg_blink_off_duration(struct mvebu_pwm *mvpwm)

>  {

> -	return PWM_BLINK_OFF_DURATION_OFF;

> +	return mvpwm->offset + PWM_BLINK_OFF_DURATION_OFF;

>  }

>  

>  /*

> @@ -781,51 +787,80 @@ static int mvebu_pwm_probe(struct platform_device *pdev,

>  	struct device *dev = &pdev->dev;

>  	struct mvebu_pwm *mvpwm;

>  	void __iomem *base;

> +	u32 offset;

>  	u32 set;

>  

> -	if (!of_device_is_compatible(mvchip->chip.of_node,

> -				     "marvell,armada-370-gpio"))

> -		return 0;

> -

> -	/*

> -	 * There are only two sets of PWM configuration registers for

> -	 * all the GPIO lines on those SoCs which this driver reserves

> -	 * for the first two GPIO chips. So if the resource is missing

> -	 * we can't treat it as an error.

> -	 */

> -	if (!platform_get_resource_byname(pdev, IORESOURCE_MEM, "pwm"))

> +	if (of_device_is_compatible(mvchip->chip.of_node,

> +				    "marvell,armada-370-gpio")) {

> +		/*

> +		 * There are only two sets of PWM configuration registers for

> +		 * all the GPIO lines on those SoCs which this driver reserves

> +		 * for the first two GPIO chips. So if the resource is missing

> +		 * we can't treat it as an error.

> +		 */

> +		if (!platform_get_resource_byname(pdev, IORESOURCE_MEM, "pwm"))

> +			return 0;

> +		offset = 0;

> +	} else if (mvchip->soc_variant == MVEBU_GPIO_SOC_VARIANT_A8K) {

> +		int ret = of_property_read_u32(dev->of_node, "pwm-offset",

> +					       &offset);

> +		if (ret < 0)

> +			return 0;


It would look more uniform if this was

	if (of_device_is_compatible(mvchip->chip.of_node,
				    "marvell,armada-8k-gpio")) {

> +	} else {

>  		return 0;

> +	}

>  

>  	if (IS_ERR(mvchip->clk))

>  		return PTR_ERR(mvchip->clk);

>  

> -	/*

> -	 * Use set A for lines of GPIO chip with id 0, B for GPIO chip

> -	 * with id 1. Don't allow further GPIO chips to be used for PWM.

> -	 */

> -	if (id == 0)

> -		set = 0;

> -	else if (id == 1)

> -		set = U32_MAX;

> -	else

> -		return -EINVAL;

> -	regmap_write(mvchip->regs,

> -		     GPIO_BLINK_CNT_SELECT_OFF + mvchip->offset, set);

> -

>  	mvpwm = devm_kzalloc(dev, sizeof(struct mvebu_pwm), GFP_KERNEL);

>  	if (!mvpwm)

>  		return -ENOMEM;

>  	mvchip->mvpwm = mvpwm;

>  	mvpwm->mvchip = mvchip;

> +	mvpwm->offset = offset;

> +

> +	if (mvchip->soc_variant == MVEBU_GPIO_SOC_VARIANT_A8K) {

> +		mvpwm->regs = mvchip->regs;

> +

> +		switch (mvchip->offset) {

> +		case AP80X_GPIO0_OFF_A8K:

> +		case CP11X_GPIO0_OFF_A8K:

> +			/* Blink counter A */

> +			set = 0;

> +			break;

> +		case CP11X_GPIO1_OFF_A8K:

> +			/* Blink counter B */

> +			set = U32_MAX;

> +			mvpwm->offset += PWM_BLINK_COUNTER_B_OFF;

> +			break;

> +		default:

> +			return -EINVAL;

> +		}

> +	} else {

> +		base = devm_platform_ioremap_resource_byname(pdev, "pwm");

> +		if (IS_ERR(base))

> +			return PTR_ERR(base);

>  

> -	base = devm_platform_ioremap_resource_byname(pdev, "pwm");

> -	if (IS_ERR(base))

> -		return PTR_ERR(base);

> +		mvpwm->regs = devm_regmap_init_mmio(&pdev->dev, base,

> +						    &mvebu_gpio_regmap_config);

> +		if (IS_ERR(mvpwm->regs))

> +			return PTR_ERR(mvpwm->regs);

> +

> +		/*

> +		 * Use set A for lines of GPIO chip with id 0, B for GPIO chip

> +		 * with id 1. Don't allow further GPIO chips to be used for PWM.

> +		 */

> +		if (id == 0)

> +			set = 0;

> +		else if (id == 1)

> +			set = U32_MAX;

> +		else

> +			return -EINVAL;

> +	}

>  

> -	mvpwm->regs = devm_regmap_init_mmio(&pdev->dev, base,

> -					    &mvebu_gpio_regmap_config);

> -	if (IS_ERR(mvpwm->regs))

> -		return PTR_ERR(mvpwm->regs);

> +	regmap_write(mvchip->regs,

> +		     GPIO_BLINK_CNT_SELECT_OFF + mvchip->offset, set);

>  

>  	mvpwm->clk_rate = clk_get_rate(mvchip->clk);

>  	if (!mvpwm->clk_rate) {

> @@ -1200,6 +1235,13 @@ static int mvebu_gpio_probe(struct platform_device *pdev)

>  

>  	devm_gpiochip_add_data(&pdev->dev, &mvchip->chip, mvchip);

>  

> +	/* Some MVEBU SoCs have simple PWM support for GPIO lines */

> +	if (IS_ENABLED(CONFIG_PWM)) {

> +		err = mvebu_pwm_probe(pdev, mvchip, id);

> +		if (err)

> +			return err;

> +	}

> +


The existing error handling looks odd here. Why is there no goto
err_domain when probing the PWMs fails? I wonder if this a bug from me
from a long time again?

	Andrew
Baruch Siach Nov. 19, 2020, 6:21 a.m. UTC | #2
Hi Andrew,

On Thu, Nov 19 2020, Andrew Lunn wrote:
> On Wed, Nov 18, 2020 at 12:30:44PM +0200, Baruch Siach wrote:

>> Use the pwm-offset DT property to store the location of PWM signal

>> duration registers.

>> 

>> Since we have more than two GPIO chips per system, we can't use the

>> alias id to differentiate between them. Use the offset value for that.

>> 

>> Move mvebu_pwm_probe() call before irq support code. The AP80x does not

>> provide irq support, but does provide PWM. Don't skip PWM probe because

>> of that.

>> 

>> Signed-off-by: Baruch Siach <baruch@tkos.co.il>


[snip]

>> @@ -781,51 +787,80 @@ static int mvebu_pwm_probe(struct platform_device *pdev,

>>  	struct device *dev = &pdev->dev;

>>  	struct mvebu_pwm *mvpwm;

>>  	void __iomem *base;

>> +	u32 offset;

>>  	u32 set;

>>  

>> -	if (!of_device_is_compatible(mvchip->chip.of_node,

>> -				     "marvell,armada-370-gpio"))

>> -		return 0;

>> -

>> -	/*

>> -	 * There are only two sets of PWM configuration registers for

>> -	 * all the GPIO lines on those SoCs which this driver reserves

>> -	 * for the first two GPIO chips. So if the resource is missing

>> -	 * we can't treat it as an error.

>> -	 */

>> -	if (!platform_get_resource_byname(pdev, IORESOURCE_MEM, "pwm"))

>> +	if (of_device_is_compatible(mvchip->chip.of_node,

>> +				    "marvell,armada-370-gpio")) {

>> +		/*

>> +		 * There are only two sets of PWM configuration registers for

>> +		 * all the GPIO lines on those SoCs which this driver reserves

>> +		 * for the first two GPIO chips. So if the resource is missing

>> +		 * we can't treat it as an error.

>> +		 */

>> +		if (!platform_get_resource_byname(pdev, IORESOURCE_MEM, "pwm"))

>> +			return 0;

>> +		offset = 0;

>> +	} else if (mvchip->soc_variant == MVEBU_GPIO_SOC_VARIANT_A8K) {

>> +		int ret = of_property_read_u32(dev->of_node, "pwm-offset",

>> +					       &offset);

>> +		if (ret < 0)

>> +			return 0;

>

> It would look more uniform if this was

>

> 	if (of_device_is_compatible(mvchip->chip.of_node,

> 				    "marvell,armada-8k-gpio")) {


Right. However I use soc_variant again below. I think that
of_device_is_compatible is too verbose for that.

In fact, I'd rather use soc_variant for marvell,armada-370-gpio as
well. The trouble is that marvell,armada-370-gpio is not equivalent to
MVEBU_GPIO_SOC_VARIANT_ORION. Changing that is more intrusive.

>> +	} else {

>>  		return 0;

>> +	}


[snip]

>> @@ -1200,6 +1235,13 @@ static int mvebu_gpio_probe(struct platform_device *pdev)

>>  

>>  	devm_gpiochip_add_data(&pdev->dev, &mvchip->chip, mvchip);

>>  

>> +	/* Some MVEBU SoCs have simple PWM support for GPIO lines */

>> +	if (IS_ENABLED(CONFIG_PWM)) {

>> +		err = mvebu_pwm_probe(pdev, mvchip, id);

>> +		if (err)

>> +			return err;

>> +	}

>> +

>

> The existing error handling looks odd here. Why is there no goto

> err_domain when probing the PWMs fails? I wonder if this a bug from me

> from a long time again?


What would you release under the err_domain label? As far as I can see
all resources are allocated using devres, and released automatically on
failure exit.

baruch

-- 
                                                     ~. .~   Tk Open Systems
=}------------------------------------------------ooO--U--Ooo------------{=
   - baruch@tkos.co.il - tel: +972.52.368.4656, http://www.tkos.co.il -
Andrew Lunn Nov. 19, 2020, 1:34 p.m. UTC | #3
> >> @@ -1200,6 +1235,13 @@ static int mvebu_gpio_probe(struct platform_device *pdev)

> >>  

> >>  	devm_gpiochip_add_data(&pdev->dev, &mvchip->chip, mvchip);

> >>  

> >> +	/* Some MVEBU SoCs have simple PWM support for GPIO lines */

> >> +	if (IS_ENABLED(CONFIG_PWM)) {

> >> +		err = mvebu_pwm_probe(pdev, mvchip, id);

> >> +		if (err)

> >> +			return err;

> >> +	}

> >> +

> >

> > The existing error handling looks odd here. Why is there no goto

> > err_domain when probing the PWMs fails? I wonder if this a bug from me

> > from a long time again?

> 

> What would you release under the err_domain label? As far as I can see

> all resources are allocated using devres, and released automatically on

> failure exit.


The IRQ domain is still registers. So once the memory is automatically
freed, don't we have a potential use after free?

       Andrew
Baruch Siach Nov. 19, 2020, 1:47 p.m. UTC | #4
Hi Andrew,

On Thu, Nov 19 2020, Andrew Lunn wrote:
>> >> @@ -1200,6 +1235,13 @@ static int mvebu_gpio_probe(struct platform_device *pdev)

>> >>  

>> >>  	devm_gpiochip_add_data(&pdev->dev, &mvchip->chip, mvchip);

>> >>  

>> >> +	/* Some MVEBU SoCs have simple PWM support for GPIO lines */

>> >> +	if (IS_ENABLED(CONFIG_PWM)) {

>> >> +		err = mvebu_pwm_probe(pdev, mvchip, id);

>> >> +		if (err)

>> >> +			return err;

>> >> +	}

>> >> +

>> >

>> > The existing error handling looks odd here. Why is there no goto

>> > err_domain when probing the PWMs fails? I wonder if this a bug from me

>> > from a long time again?

>> 

>> What would you release under the err_domain label? As far as I can see

>> all resources are allocated using devres, and released automatically on

>> failure exit.

>

> The IRQ domain is still registers. So once the memory is automatically

> freed, don't we have a potential use after free?


This patch moves PWM registration before IRQ domain registration for
another reason as mentioned in the commit log. So this might
incidentally fix the bug.

Would you prefer a separate patch for that with a 'Fixes:
757642f9a584e8' tag?

baruch

-- 
                                                     ~. .~   Tk Open Systems
=}------------------------------------------------ooO--U--Ooo------------{=
   - baruch@tkos.co.il - tel: +972.52.368.4656, http://www.tkos.co.il -
Bartosz Golaszewski Dec. 1, 2020, 6:16 p.m. UTC | #5
On Thu, Nov 19, 2020 at 2:47 PM Baruch Siach <baruch@tkos.co.il> wrote:
>

> Hi Andrew,

>

> On Thu, Nov 19 2020, Andrew Lunn wrote:

> >> >> @@ -1200,6 +1235,13 @@ static int mvebu_gpio_probe(struct platform_device *pdev)

> >> >>

> >> >>   devm_gpiochip_add_data(&pdev->dev, &mvchip->chip, mvchip);

> >> >>

> >> >> + /* Some MVEBU SoCs have simple PWM support for GPIO lines */

> >> >> + if (IS_ENABLED(CONFIG_PWM)) {

> >> >> +         err = mvebu_pwm_probe(pdev, mvchip, id);

> >> >> +         if (err)

> >> >> +                 return err;

> >> >> + }

> >> >> +

> >> >

> >> > The existing error handling looks odd here. Why is there no goto

> >> > err_domain when probing the PWMs fails? I wonder if this a bug from me

> >> > from a long time again?

> >>

> >> What would you release under the err_domain label? As far as I can see

> >> all resources are allocated using devres, and released automatically on

> >> failure exit.

> >

> > The IRQ domain is still registers. So once the memory is automatically

> > freed, don't we have a potential use after free?

>

> This patch moves PWM registration before IRQ domain registration for

> another reason as mentioned in the commit log. So this might

> incidentally fix the bug.

>

> Would you prefer a separate patch for that with a 'Fixes:

> 757642f9a584e8' tag?

>

> baruch

>


Baruch: does this series conflict with the fix you sent? I'm thinking
about how to take it through the next and fixes trees.

Bartosz
Baruch Siach Dec. 1, 2020, 6:21 p.m. UTC | #6
Hi Bartosz,

On Tue, Dec 01 2020, Bartosz Golaszewski wrote:
> On Thu, Nov 19, 2020 at 2:47 PM Baruch Siach <baruch@tkos.co.il> wrote:

>> On Thu, Nov 19 2020, Andrew Lunn wrote:

>> >> >> @@ -1200,6 +1235,13 @@ static int mvebu_gpio_probe(struct platform_device *pdev)

>> >> >>

>> >> >>   devm_gpiochip_add_data(&pdev->dev, &mvchip->chip, mvchip);

>> >> >>

>> >> >> + /* Some MVEBU SoCs have simple PWM support for GPIO lines */

>> >> >> + if (IS_ENABLED(CONFIG_PWM)) {

>> >> >> +         err = mvebu_pwm_probe(pdev, mvchip, id);

>> >> >> +         if (err)

>> >> >> +                 return err;

>> >> >> + }

>> >> >> +

>> >> >

>> >> > The existing error handling looks odd here. Why is there no goto

>> >> > err_domain when probing the PWMs fails? I wonder if this a bug from me

>> >> > from a long time again?

>> >>

>> >> What would you release under the err_domain label? As far as I can see

>> >> all resources are allocated using devres, and released automatically on

>> >> failure exit.

>> >

>> > The IRQ domain is still registers. So once the memory is automatically

>> > freed, don't we have a potential use after free?

>>

>> This patch moves PWM registration before IRQ domain registration for

>> another reason as mentioned in the commit log. So this might

>> incidentally fix the bug.

>>

>> Would you prefer a separate patch for that with a 'Fixes:

>> 757642f9a584e8' tag?

>

> Baruch: does this series conflict with the fix you sent? I'm thinking

> about how to take it through the next and fixes trees.


Yes, It conflicts.

I can send in a single series v3 of the fix along with the other patches
rebased on top. Would that work for you?

Thanks,
baruch

-- 
                                                     ~. .~   Tk Open Systems
=}------------------------------------------------ooO--U--Ooo------------{=
   - baruch@tkos.co.il - tel: +972.52.368.4656, http://www.tkos.co.il -
diff mbox series

Patch

diff --git a/drivers/gpio/gpio-mvebu.c b/drivers/gpio/gpio-mvebu.c
index 946571e70928..8602afd21673 100644
--- a/drivers/gpio/gpio-mvebu.c
+++ b/drivers/gpio/gpio-mvebu.c
@@ -70,7 +70,12 @@ 
  */
 #define PWM_BLINK_ON_DURATION_OFF	0x0
 #define PWM_BLINK_OFF_DURATION_OFF	0x4
+#define PWM_BLINK_COUNTER_B_OFF		0x8
 
+/* Armada 8k variant gpios register offsets */
+#define AP80X_GPIO0_OFF_A8K		0x1040
+#define CP11X_GPIO0_OFF_A8K		0x100
+#define CP11X_GPIO1_OFF_A8K		0x140
 
 /* The MV78200 has per-CPU registers for edge mask and level mask */
 #define GPIO_EDGE_MASK_MV78200_OFF(cpu)	  ((cpu) ? 0x30 : 0x18)
@@ -93,6 +98,7 @@ 
 
 struct mvebu_pwm {
 	struct regmap		*regs;
+	u32			 offset;
 	unsigned long		 clk_rate;
 	struct gpio_desc	*gpiod;
 	struct pwm_chip		 chip;
@@ -283,12 +289,12 @@  mvebu_gpio_write_level_mask(struct mvebu_gpio_chip *mvchip, u32 val)
  */
 static unsigned int mvebu_pwmreg_blink_on_duration(struct mvebu_pwm *mvpwm)
 {
-	return PWM_BLINK_ON_DURATION_OFF;
+	return mvpwm->offset + PWM_BLINK_ON_DURATION_OFF;
 }
 
 static unsigned int mvebu_pwmreg_blink_off_duration(struct mvebu_pwm *mvpwm)
 {
-	return PWM_BLINK_OFF_DURATION_OFF;
+	return mvpwm->offset + PWM_BLINK_OFF_DURATION_OFF;
 }
 
 /*
@@ -781,51 +787,80 @@  static int mvebu_pwm_probe(struct platform_device *pdev,
 	struct device *dev = &pdev->dev;
 	struct mvebu_pwm *mvpwm;
 	void __iomem *base;
+	u32 offset;
 	u32 set;
 
-	if (!of_device_is_compatible(mvchip->chip.of_node,
-				     "marvell,armada-370-gpio"))
-		return 0;
-
-	/*
-	 * There are only two sets of PWM configuration registers for
-	 * all the GPIO lines on those SoCs which this driver reserves
-	 * for the first two GPIO chips. So if the resource is missing
-	 * we can't treat it as an error.
-	 */
-	if (!platform_get_resource_byname(pdev, IORESOURCE_MEM, "pwm"))
+	if (of_device_is_compatible(mvchip->chip.of_node,
+				    "marvell,armada-370-gpio")) {
+		/*
+		 * There are only two sets of PWM configuration registers for
+		 * all the GPIO lines on those SoCs which this driver reserves
+		 * for the first two GPIO chips. So if the resource is missing
+		 * we can't treat it as an error.
+		 */
+		if (!platform_get_resource_byname(pdev, IORESOURCE_MEM, "pwm"))
+			return 0;
+		offset = 0;
+	} else if (mvchip->soc_variant == MVEBU_GPIO_SOC_VARIANT_A8K) {
+		int ret = of_property_read_u32(dev->of_node, "pwm-offset",
+					       &offset);
+		if (ret < 0)
+			return 0;
+	} else {
 		return 0;
+	}
 
 	if (IS_ERR(mvchip->clk))
 		return PTR_ERR(mvchip->clk);
 
-	/*
-	 * Use set A for lines of GPIO chip with id 0, B for GPIO chip
-	 * with id 1. Don't allow further GPIO chips to be used for PWM.
-	 */
-	if (id == 0)
-		set = 0;
-	else if (id == 1)
-		set = U32_MAX;
-	else
-		return -EINVAL;
-	regmap_write(mvchip->regs,
-		     GPIO_BLINK_CNT_SELECT_OFF + mvchip->offset, set);
-
 	mvpwm = devm_kzalloc(dev, sizeof(struct mvebu_pwm), GFP_KERNEL);
 	if (!mvpwm)
 		return -ENOMEM;
 	mvchip->mvpwm = mvpwm;
 	mvpwm->mvchip = mvchip;
+	mvpwm->offset = offset;
+
+	if (mvchip->soc_variant == MVEBU_GPIO_SOC_VARIANT_A8K) {
+		mvpwm->regs = mvchip->regs;
+
+		switch (mvchip->offset) {
+		case AP80X_GPIO0_OFF_A8K:
+		case CP11X_GPIO0_OFF_A8K:
+			/* Blink counter A */
+			set = 0;
+			break;
+		case CP11X_GPIO1_OFF_A8K:
+			/* Blink counter B */
+			set = U32_MAX;
+			mvpwm->offset += PWM_BLINK_COUNTER_B_OFF;
+			break;
+		default:
+			return -EINVAL;
+		}
+	} else {
+		base = devm_platform_ioremap_resource_byname(pdev, "pwm");
+		if (IS_ERR(base))
+			return PTR_ERR(base);
 
-	base = devm_platform_ioremap_resource_byname(pdev, "pwm");
-	if (IS_ERR(base))
-		return PTR_ERR(base);
+		mvpwm->regs = devm_regmap_init_mmio(&pdev->dev, base,
+						    &mvebu_gpio_regmap_config);
+		if (IS_ERR(mvpwm->regs))
+			return PTR_ERR(mvpwm->regs);
+
+		/*
+		 * Use set A for lines of GPIO chip with id 0, B for GPIO chip
+		 * with id 1. Don't allow further GPIO chips to be used for PWM.
+		 */
+		if (id == 0)
+			set = 0;
+		else if (id == 1)
+			set = U32_MAX;
+		else
+			return -EINVAL;
+	}
 
-	mvpwm->regs = devm_regmap_init_mmio(&pdev->dev, base,
-					    &mvebu_gpio_regmap_config);
-	if (IS_ERR(mvpwm->regs))
-		return PTR_ERR(mvpwm->regs);
+	regmap_write(mvchip->regs,
+		     GPIO_BLINK_CNT_SELECT_OFF + mvchip->offset, set);
 
 	mvpwm->clk_rate = clk_get_rate(mvchip->clk);
 	if (!mvpwm->clk_rate) {
@@ -1200,6 +1235,13 @@  static int mvebu_gpio_probe(struct platform_device *pdev)
 
 	devm_gpiochip_add_data(&pdev->dev, &mvchip->chip, mvchip);
 
+	/* Some MVEBU SoCs have simple PWM support for GPIO lines */
+	if (IS_ENABLED(CONFIG_PWM)) {
+		err = mvebu_pwm_probe(pdev, mvchip, id);
+		if (err)
+			return err;
+	}
+
 	/* Some gpio controllers do not provide irq support */
 	if (!have_irqs)
 		return 0;
@@ -1257,10 +1299,6 @@  static int mvebu_gpio_probe(struct platform_device *pdev)
 						 mvchip);
 	}
 
-	/* Some MVEBU SoCs have simple PWM support for GPIO lines */
-	if (IS_ENABLED(CONFIG_PWM))
-		return mvebu_pwm_probe(pdev, mvchip, id);
-
 	return 0;
 
 err_domain: