bpf: Move ksym_tnode to bpf_ksym

[linux-2.6-microblaze.git] / lib / test_bpf.c

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