1 files changed, 13 insertions, 5 deletions

diff --git a/arch/arm/mach-omap2/omap-wakeupgen.c b/arch/arm/mach-omap2/omap-wakeupgen.c
index 813c61558a5f..3664562f9148 100644
--- a/arch/arm/mach-omap2/omap-wakeupgen.c
+++ b/arch/arm/mach-omap2/omap-wakeupgen.c
@@ -33,8 +33,12 @@
 #include "omap4-sar-layout.h"
 #include "common.h"
 
-#define MAX_NR_REG_BANKS        5
-#define MAX_IRQS                160
+#define AM43XX_NR_REG_BANKS     7
+#define AM43XX_IRQS             224
+#define MAX_NR_REG_BANKS        AM43XX_NR_REG_BANKS
+#define MAX_IRQS                AM43XX_IRQS
+#define DEFAULT_NR_REG_BANKS    5
+#define DEFAULT_IRQS            160
 #define WKG_MASK_ALL            0x00000000
 #define WKG_UNMASK_ALL          0xffffffff
 #define CPU_ENA_OFFSET          0x400
@@ -47,8 +51,8 @@ static void __iomem *wakeupgen_base;
 static void __iomem *sar_base;
 static DEFINE_RAW_SPINLOCK(wakeupgen_lock);
 static unsigned int irq_target_cpu[MAX_IRQS];
-static unsigned int irq_banks = MAX_NR_REG_BANKS;
-static unsigned int max_irqs = MAX_IRQS;
+static unsigned int irq_banks = DEFAULT_NR_REG_BANKS;
+static unsigned int max_irqs = DEFAULT_IRQS;
 static unsigned int omap_secure_apis;
 
 /*
@@ -418,12 +422,16 @@ int __init omap_wakeupgen_init(void)
                 irq_banks = OMAP4_NR_BANKS;
                 max_irqs = OMAP4_NR_IRQS;
                 omap_secure_apis = 1;
+        } else if (soc_is_am43xx()) {
+                irq_banks = AM43XX_NR_REG_BANKS;
+                max_irqs = AM43XX_IRQS;
         }
 
         /* Clear all IRQ bitmasks at wakeupGen level */
         for (i = 0; i < irq_banks; i++) {
                 wakeupgen_writel(0, i, CPU0_ID);
-                wakeupgen_writel(0, i, CPU1_ID);
+                if (!soc_is_am43xx())
+                        wakeupgen_writel(0, i, CPU1_ID);
         }
 
         /*