1 files changed, 4 insertions, 3 deletions

diff --git a/arch/arm64/net/bpf_jit_comp.c b/arch/arm64/net/bpf_jit_comp.c
index ef35e866caf7..dd428807cb30 100644
--- a/arch/arm64/net/bpf_jit_comp.c
+++ b/arch/arm64/net/bpf_jit_comp.c
@@ -762,7 +762,7 @@ void bpf_jit_compile(struct bpf_prog *prog)
         /* Nothing to do here. We support Internal BPF. */
 }
 
-void bpf_int_jit_compile(struct bpf_prog *prog)
+struct bpf_prog *bpf_int_jit_compile(struct bpf_prog *prog)
 {
         struct bpf_binary_header *header;
         struct jit_ctx ctx;
@@ -770,14 +770,14 @@ void bpf_int_jit_compile(struct bpf_prog *prog)
         u8 *image_ptr;
 
         if (!bpf_jit_enable)
-                return;
+                return prog;
 
         memset(&ctx, 0, sizeof(ctx));
         ctx.prog = prog;
 
         ctx.offset = kcalloc(prog->len, sizeof(int), GFP_KERNEL);
         if (ctx.offset == NULL)
-                return;
+                return prog;
 
         /* 1. Initial fake pass to compute ctx->idx. */
 
@@ -828,6 +828,7 @@ void bpf_int_jit_compile(struct bpf_prog *prog)
         prog->jited = 1;
 out:
         kfree(ctx.offset);
+        return prog;
 }
 
 void bpf_jit_free(struct bpf_prog *prog)