rten/Makefile at main · robertknight/rten · 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
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
.PHONY: all
all:
	cargo build --workspace

.PHONY: schema
schema: rten-model-file/src/schema_generated.rs rten-convert/rten_convert/schema_generated.py

.PHONY: clean
clean:
	rm -rf dist/*
	rm -rf target/

.PHONY: check
check: checkformatting test lint docs

.PHONY: checkformatting
checkformatting:
	cargo fmt --check

.PHONY: docs
docs:
	RUSTDOCFLAGS='-D warnings' cargo doc -p rten --features mmap,random

.PHONY: lint
lint:
	cargo clippy --workspace

.PHONY: miri
miri:
	# - Only the tensor lib is currently tested. Testing the main crate will
	#   require changes to prevent tests taking too long to run.
	cargo +nightly miri test -p rten-tensor

# Run tests for all crates with all features enabled that do not require
# nightly Rust.
.PHONY: test
test:
	cargo test --no-fail-fast --workspace --features all-ops,mmap,text-decoder,serde

# Default to running tests for the main crate unless otherwise specified.
PACKAGE ?= rten
CRATE := $(subst -,_,$(PACKAGE))

# Support enabling WASM relaxed-simd extension by passing `RELAXED=y`.
ifeq ($(RELAXED),y)
WASM_TARGET_FEATURES := -C target-feature=+simd128,+relaxed-simd
else
WASM_TARGET_FEATURES := -C target-feature=+simd128
endif

.PHONY: wasm
wasm:
	RUSTFLAGS="$(WASM_TARGET_FEATURES)" cargo build --features=wasm_api --release --target wasm32-unknown-unknown
	wasm-bindgen target/wasm32-unknown-unknown/release/rten.wasm --out-dir dist/ --target web --weak-refs
	# This makes the binary smaller but also removes all symbols. Comment this
	# out to get a release WASM build with symbols.
	tools/optimize-wasm.sh dist/rten_bg.wasm

.PHONY: wasm-nosimd
wasm-nosimd:
	cargo build --release --target wasm32-unknown-unknown
	wasm-bindgen target/wasm32-unknown-unknown/release/rten.wasm --out-dir dist/ --out-name rten-nosimd --target web --weak-refs
	tools/optimize-wasm.sh dist/rten-nosimd_bg.wasm

.PHONY: wasm-all
wasm-all: wasm wasm-nosimd

# Run wasm tests using `make wasm-test PACKAGE={package_name}`
#
# WASM tests run with `--nocapture` as otherwise assertion failure panic messages
# are not printed if a test assert fails.
.PHONY: wasm-test
wasm-test:
	rm -f target/wasm32-wasi/debug/deps/$(CRATE)-*.wasm
	RUSTFLAGS="$(WASM_TARGET_FEATURES)" cargo build --target wasm32-wasip1 --tests -p $(PACKAGE)
	wasmtime --dir . target/wasm32-wasip1/debug/deps/${CRATE}-*.wasm --nocapture

# Run wasm benchmark using `make wasm-test PACKAGE={package_name} BENCH={bench_name}`
.PHONY: wasm-bench
wasm-bench:
	rm -f target/wasm32-wasi/release/deps/$(CRATE)-*.wasm
	RUSTFLAGS="$(WASM_TARGET_FEATURES)" cargo build --target wasm32-wasip1 --tests -p $(PACKAGE) -r
	wasmtime --dir . target/wasm32-wasip1/release/deps/$(CRATE)-*.wasm --nocapture --ignored $(BENCH)

MODEL_SCHEMA=rten-model-file/src/schema.fbs

rten-model-file/src/schema_generated.rs: $(MODEL_SCHEMA)
	flatc -o rten-model-file/src/ --rust $(MODEL_SCHEMA)
	cargo fmt

rten-convert/rten_convert/schema_generated.py: $(MODEL_SCHEMA)
	flatc -o rten-convert/rten_convert --gen-onefile --gen-object-api --python $(MODEL_SCHEMA)

.PHONY: gen-pytorch-references
gen-pytorch-references:
	python -m pytorch-ref-tests.rnn