mirror of https://github.com/maderix/ANE.git
72 lines
2.1 KiB
Makefile
72 lines
2.1 KiB
Makefile
CC = xcrun clang
|
|
|
|
ANE_COMPAT = -Wno-deprecated-declarations
|
|
SEC_FLAGS = -fstack-protector-strong -Wformat-security
|
|
|
|
CFLAGS = -O2 -Wall $(ANE_COMPAT) -fobjc-arc $(SEC_FLAGS)
|
|
CFLAGS_DEBUG = -O0 -g -Wall $(ANE_COMPAT) -fobjc-arc -fsanitize=address,undefined
|
|
FRAMEWORKS = -framework Foundation -framework CoreML -framework IOSurface
|
|
LDFLAGS = $(FRAMEWORKS) -ldl
|
|
|
|
HEADERS_LARGE = stories_config.h stories_io.h stories_mil.h stories_cpu_ops.h
|
|
|
|
HEADERS_ANE = $(HEADERS_LARGE) ane_rmsnorm_bwd.h ane_classifier.h
|
|
|
|
train: train.m ane_runtime.h ane_mil_gen.h model.h forward.h backward.h
|
|
$(CC) $(CFLAGS) -o $@ train.m $(LDFLAGS)
|
|
|
|
train_large: train_large.m $(HEADERS_LARGE)
|
|
$(CC) $(CFLAGS) -o $@ train_large.m $(LDFLAGS) -framework Accelerate
|
|
|
|
train_large_ane: train_large_ane.m $(HEADERS_ANE)
|
|
$(CC) $(CFLAGS) -o $@ train_large_ane.m $(LDFLAGS) -framework Accelerate
|
|
|
|
PROBES = test_weight_reload test_perf_stats test_qos_sweep test_ane_advanced
|
|
|
|
test_rmsnorm_bwd: test_rmsnorm_bwd.m $(HEADERS_ANE)
|
|
$(CC) $(CFLAGS) -o $@ $< $(LDFLAGS) -framework Accelerate
|
|
|
|
test_classifier: test_classifier.m $(HEADERS_ANE)
|
|
$(CC) $(CFLAGS) -o $@ $< $(LDFLAGS) -framework Accelerate
|
|
|
|
test_weight_reload: test_weight_reload.m
|
|
$(CC) $(CFLAGS) -o $@ $< $(LDFLAGS)
|
|
|
|
test_perf_stats: test_perf_stats.m
|
|
$(CC) $(CFLAGS) -o $@ $< $(LDFLAGS)
|
|
|
|
test_qos_sweep: test_qos_sweep.m
|
|
$(CC) $(CFLAGS) -o $@ $< $(LDFLAGS)
|
|
|
|
test_ane_advanced: test_ane_advanced.m
|
|
$(CC) $(CFLAGS) -o $@ $< $(LDFLAGS)
|
|
|
|
test_chaining: test_chaining.m
|
|
$(CC) $(CFLAGS) -o $@ $< $(LDFLAGS)
|
|
|
|
probes: $(PROBES)
|
|
|
|
data: tokenize
|
|
@bash download_data.sh
|
|
|
|
tokenize:
|
|
python3 tokenize.py
|
|
|
|
setup: data
|
|
@echo "=== Setup complete ==="
|
|
@echo "Data: tinystories_data00.bin"
|
|
@echo "To train: make train_large && ./train_large"
|
|
@echo "Override paths: ANE_MODEL_PATH=... ANE_DATA_PATH=... ./train_large"
|
|
|
|
verify-flags:
|
|
@echo "=== Active CFLAGS ==="
|
|
@echo "$(CFLAGS)"
|
|
@echo "=== Compiler version ==="
|
|
@xcrun clang --version
|
|
|
|
clean:
|
|
rm -f train train_large train_large_ane $(PROBES) test_rmsnorm_bwd test_classifier
|
|
|
|
.PHONY: clean tokenize probes verify-flags data setup
|
|
|