1 files changed, 7 insertions, 1 deletions

diff --git a/kernel/bpf/syscall.c b/kernel/bpf/syscall.c
index d124e702e040..54fba06942f5 100644
--- a/kernel/bpf/syscall.c
+++ b/kernel/bpf/syscall.c
@@ -592,6 +592,12 @@ static int map_update_elem(union bpf_attr *attr)
         if (copy_from_user(value, uvalue, value_size) != 0)
                 goto free_value;
 
+        /* Need to create a kthread, thus must support schedule */
+        if (map->map_type == BPF_MAP_TYPE_CPUMAP) {
+                err = map->ops->map_update_elem(map, key, value, attr->flags);
+                goto out;
+        }
+
         /* must increment bpf_prog_active to avoid kprobe+bpf triggering from
          * inside bpf map update or delete otherwise deadlocks are possible
          */
@@ -622,7 +628,7 @@ static int map_update_elem(union bpf_attr *attr)
         }
         __this_cpu_dec(bpf_prog_active);
         preempt_enable();
-
+out:
         if (!err)
                 trace_bpf_map_update_elem(map, ufd, key, value);
 free_value: