AO-WebGPU-Demo/build.sh at main · elliotsayes/AO-WebGPU-Demo · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
#!/bin/bash
set -e

SCRIPT_DIR=$( cd -- "$( dirname -- "${BASH_SOURCE[0]}" )" &> /dev/null && pwd )

WEBGPU_DIR="${SCRIPT_DIR}/wasm_webgpu"
PROCESS_DIR="${SCRIPT_DIR}/aos/process"
LIBS_DIR="${PROCESS_DIR}/libs"

rm -rf ${LIBS_DIR}
mkdir -p ${LIBS_DIR}

# AO_IMAGE="p3rmaw3b/ao:0.1.3"
AO_IMAGE="p3rmaw3b/ao:dev"
AO_IMAGE_SYNC="p3rmaw3b/ao:webgpu-sync"

EMXX_CFLAGS="-s SUPPORT_LONGJMP=1 -s USE_WEBGPU=1"
EMXX_CFLAGS_LUA="-s SUPPORT_LONGJMP=1 -s USE_WEBGPU=1 /lua-5.3.4/src/liblua.a -I/lua-5.3.4/src"

# Build wasm_webgpu libraries with emscripten
docker run -v ${WEBGPU_DIR}:/wasm_webgpu ${AO_IMAGE} sh -c \
		"cd /wasm_webgpu && emcc -c hellotriangle.cpp -o hellotriangle.o ${EMXX_CFLAGS} && emar r hellotriangle.a hellotriangle.o"
docker run -v ${WEBGPU_DIR}:/wasm_webgpu ${AO_IMAGE} sh -c \
		"cd /wasm_webgpu && emcc -c helloboat.cpp -o helloboat.o ${EMXX_CFLAGS} && emar r helloboat.a helloboat.o"

# Build lsokoldemo into a library with emscripten
docker run -v ${WEBGPU_DIR}:/wasm_webgpu ${AO_IMAGE} sh -c \
		"cd /wasm_webgpu && emcc -c lsokoldemo.c -o lsokoldemo.o ${EMXX_CFLAGS_LUA} && emar rcs lsokoldemo.a lsokoldemo.o"

# Fix permissions
sudo chmod -R 777 ${WEBGPU_DIR}

# # Copy to the libs directory
cp ${WEBGPU_DIR}/hellotriangle.a $LIBS_DIR/hellotriangle.a
cp ${WEBGPU_DIR}/helloboat.a $LIBS_DIR/helloboat.a
cp ${WEBGPU_DIR}/lsokoldemo.a $LIBS_DIR/lsokoldemo.a

# Copy config.yml to the process directory
cp ${SCRIPT_DIR}/config.yml ${PROCESS_DIR}/config.yml

# Build the process module
cd ${PROCESS_DIR}
docker run -e DEBUG=1 --platform linux/amd64 -v ./:/src ${AO_IMAGE_SYNC} ao-build-module
# Copy the sync glue js
cp ${PROCESS_DIR}/process.js ${SCRIPT_DIR}/ao/loader/src/formats/emscripten-webgpu-sync.cjs
docker run -e DEBUG=1 --platform linux/amd64 -v ./:/src ${AO_IMAGE} ao-build-module
# Copy the unsafe glue js
cp ${PROCESS_DIR}/process.js ${SCRIPT_DIR}/ao/loader/src/formats/emscripten-webgpu-unsafe.cjs

cd ${SCRIPT_DIR}/ao/loader
npm install
npm run format-file src/formats/emscripten-webgpu-sync.cjs
npm run format-file src/formats/emscripten-webgpu-unsafe.cjs
npm run patch:webgpu
npm run build

# Copy the process module to the tests directory
cp ${PROCESS_DIR}/process.wasm ${SCRIPT_DIR}/tests/process.wasm