Skip to content

Commit

Permalink
minimal server example for julia
Browse files Browse the repository at this point in the history
  • Loading branch information
annereinarz committed May 17, 2024
1 parent de63d3b commit 6c8e437
Show file tree
Hide file tree
Showing 2 changed files with 31 additions and 0 deletions.
13 changes: 13 additions & 0 deletions models/testmodel-julia/Dockerfile
Original file line number Diff line number Diff line change
@@ -0,0 +1,13 @@
FROM ubuntu:latest

COPY minimal-server.jl /

RUN apt update && \
DEBIAN_FRONTEND="noninteractive" apt install -y curl

ENV JULIA_PATH /usr/local/julia
ENV PATH $JULIA_PATH/bin:$PATH

RUN curl -fsSL https://install.julialang.org | sh -s -- -y

#CMD julia minimal-server.jl
18 changes: 18 additions & 0 deletions models/testmodel-julia/minimal-server.jl
Original file line number Diff line number Diff line change
@@ -0,0 +1,18 @@
import Pkg

tempdir = mktempdir()
Pkg.activate(tempdir)
Pkg.add(["UMBridge"])

using UMBridge

testmodel = UMBridge.Model(
name = "forward",
inputSizes = [1],
outputSizes = [1],
supportsGradient = true,
evaluate = (input, config) -> 2 * input[1],
gradient = (outWrt, inWrt, input, sens, config) -> 2 * sens[1]
)

UMBridge.serve_models([testmodel], 4242)

0 comments on commit 6c8e437

Please sign in to comment.