Make sure all tasklets are drained before unloading the LinuxKPI.
Else use-after-free may happen.

This change also partially cherry picks support for the
taskqgroup_drain_all() function.

Sponsored by: Mellanox Technologies // NVIDIA Networking

(cherry picked from commit 209d4919c5270fe9e3c1d809cf91355d222dc2f3)