X-Git-Url: http://git.liburcu.org/?a=blobdiff_plain;f=ltt-relay-alloc.c;fp=ltt-relay-alloc.c;h=43246a73d38f7e3f9e6b929b66c6af03e99e0f38;hb=031e36b31fc9aa1ca6dcaa35091bffa3b4b59555;hp=6ff552efacc06cb5506f23090a34833533e5783e;hpb=65da3e9c6fbc641fb475e539ec11dc6b3d4896e9;p=lttng-modules.git

diff --git a/ltt-relay-alloc.c b/ltt-relay-alloc.c
index 6ff552ef..43246a73 100644
--- a/ltt-relay-alloc.c
+++ b/ltt-relay-alloc.c
@@ -716,9 +716,14 @@ void *ltt_relay_offset_address(struct ltt_chanbuf_alloc *bufa, size_t offset)
 }
 EXPORT_SYMBOL_GPL(ltt_relay_offset_address);
 
+static struct notifier_block ltt_relay_hotcpu = {
+	.notifier_call = ltt_relay_hotcpu_callback,
+	.priority = 5,
+};
+
 static __init int ltt_relay_alloc_init(void)
 {
-	hotcpu_notifier(ltt_relay_hotcpu_callback, 5);
+	register_cpu_notifier(&ltt_relay_hotcpu);
 	ltt_relay_init();
 	ltt_ascii_init();
 	return 0;
@@ -728,6 +733,7 @@ static void __exit ltt_relay_alloc_exit(void)
 {
 	ltt_ascii_exit();
 	ltt_relay_exit();
+	unregister_cpu_notifier(&ltt_relay_hotcpu);
 }
 
 module_init(ltt_relay_alloc_init);