X-Git-Url: http://git.lttng.org/?p=lttng-modules.git;a=blobdiff_plain;f=lttng-context-need-reschedule.c;h=7f8deec24461d00cce6ea410a4653a4244f9fff9;hp=9a51f0c60714df4a47f05dd1b56be406f6c17a06;hb=cd4486798c2b046ea93b89439cd705e93c40b349;hpb=ceabb767180e064629b5a9ab4ed14449da864763

diff --git a/lttng-context-need-reschedule.c b/lttng-context-need-reschedule.c
index 9a51f0c6..7f8deec2 100644
--- a/lttng-context-need-reschedule.c
+++ b/lttng-context-need-reschedule.c
@@ -11,10 +11,10 @@
 #include <linux/slab.h>
 #include <linux/sched.h>
 #include <linux/irqflags.h>
-#include <lttng-events.h>
-#include <wrapper/ringbuffer/frontend_types.h>
+#include <lttng/events.h>
+#include <ringbuffer/frontend_types.h>
 #include <wrapper/vmalloc.h>
-#include <lttng-tracer.h>
+#include <lttng/tracer.h>
 
 static
 size_t need_reschedule_get_size(size_t offset)
@@ -68,7 +68,7 @@ int lttng_add_need_reschedule_to_ctx(struct lttng_ctx **ctx)
 	field->record = need_reschedule_record;
 	field->get_value = need_reschedule_get_value;
 	lttng_context_update(*ctx);
-	wrapper_vmalloc_sync_all();
+	wrapper_vmalloc_sync_mappings();
 	return 0;
 }
 EXPORT_SYMBOL_GPL(lttng_add_need_reschedule_to_ctx);