Drivers: hv: Manage event tasklets on per-cpu basis

Now that we can potentially take vmbus interrupts on any CPU, make the
tasklets per-CPU.

Signed-off-by: K. Y. Srinivasan <kys@microsoft.com>
Reviewed-by: Haiyang Zhang <haiyangz@microsoft.com>
Signed-off-by: Greg Kroah-Hartman <gregkh@linuxfoundation.org>

1 files changed, 12 insertions, 0 deletions

diff --git a/drivers/hv/hv.c b/drivers/hv/hv.c
index e989c6fd1f8f..363532add4c7 100644
--- a/drivers/hv/hv.c
+++ b/drivers/hv/hv.c
@@ -27,6 +27,7 @@
 #include <linux/vmalloc.h>
 #include <linux/hyperv.h>
 #include <linux/version.h>
+#include <linux/interrupt.h>
 #include <asm/hyperv.h>
 #include "hyperv_vmbus.h"
 
@@ -137,6 +138,8 @@ int hv_init(void)
                sizeof(void *) * NR_CPUS);
         memset(hv_context.vp_index, 0,
                sizeof(int) * NR_CPUS);
+        memset(hv_context.event_dpc, 0,
+               sizeof(void *) * NR_CPUS);
 
         max_leaf = query_hypervisor_info();
 
@@ -285,6 +288,15 @@ void hv_synic_init(void *irqarg)
         /* Check the version */
         rdmsrl(HV_X64_MSR_SVERSION, version);
 
+        hv_context.event_dpc[cpu] = (struct tasklet_struct *)
+                                        kmalloc(sizeof(struct tasklet_struct),
+                                                GFP_ATOMIC);
+        if (hv_context.event_dpc[cpu] == NULL) {
+                pr_err("Unable to allocate event dpc\n");
+                goto cleanup;
+        }
+        tasklet_init(hv_context.event_dpc[cpu], vmbus_on_event, cpu);
+
         hv_context.synic_message_page[cpu] =
                 (void *)get_zeroed_page(GFP_ATOMIC);