1 files changed, 2 insertions, 0 deletions

diff --git a/lib/test_bpf.c b/lib/test_bpf.c
index f3e570722a7e..0845f635f404 100644
--- a/lib/test_bpf.c
+++ b/lib/test_bpf.c
@@ -6668,12 +6668,14 @@ static int __run_one(const struct bpf_prog *fp, const void *data,
         u64 start, finish;
         int ret = 0, i;
 
+        preempt_disable();
         start = ktime_get_ns();
 
         for (i = 0; i < runs; i++)
                 ret = BPF_PROG_RUN(fp, data);
 
         finish = ktime_get_ns();
+        preempt_enable();
 
         *duration = finish - start;
         do_div(*duration, runs);