diff --git a/include/linux/filter.h b/include/linux/filter.h
index 45ba2a61f9e55146f1c092d8936757200cd7f313..0418d88aa0f3d09a76c10335be84ff65c5399a58 100644
--- a/include/linux/filter.h
+++ b/include/linux/filter.h
@@ -578,12 +578,13 @@ DECLARE_STATIC_KEY_FALSE(bpf_stats_enabled_key);
 	if (static_branch_unlikely(&bpf_stats_enabled_key)) {		\
 		struct bpf_prog_stats *__stats;				\
 		u64 __start = sched_clock();				\
+		unsigned long flags;					\
 		__ret = dfunc(ctx, (prog)->insnsi, (prog)->bpf_func);	\
 		__stats = this_cpu_ptr(prog->aux->stats);		\
-		u64_stats_update_begin(&__stats->syncp);		\
+		flags = u64_stats_update_begin_irqsave(&__stats->syncp);\
 		__stats->cnt++;						\
 		__stats->nsecs += sched_clock() - __start;		\
-		u64_stats_update_end(&__stats->syncp);			\
+		u64_stats_update_end_irqrestore(&__stats->syncp, flags);\
 	} else {							\
 		__ret = dfunc(ctx, (prog)->insnsi, (prog)->bpf_func);	\
 	}								\