49 lines
1.5 KiB
Makefile
49 lines
1.5 KiB
Makefile
CUR_DIR := $(shell dirname $(realpath $(lastword $(MAKEFILE_LIST))))
|
|
XGBOOST_TRACKER := $(realpath $(CUR_DIR)/xgboost_src/dmlc-core/tracker/)
|
|
BIN := xgboost
|
|
DATA := data
|
|
|
|
.PHONY: all test test-native test-local-cluster clean
|
|
|
|
all: occlum_workspace
|
|
|
|
$(BIN):
|
|
cp xgboost_src/$(BIN) .
|
|
|
|
$(DATA):
|
|
rm -rf $(DATA) && mkdir -p $(DATA)
|
|
cp xgboost_src/demo/binary_classification/agaricus.txt.test $(DATA)
|
|
cp xgboost_src/demo/binary_classification/agaricus.txt.train $(DATA)
|
|
cp xgboost_src/demo/binary_classification/mushroom.conf $(DATA)
|
|
sed -i 's/agaricus/.\/data\/agaricus/g' $(DATA)/mushroom.conf
|
|
sed -i 's/num_round = 2/num_round = 10/g' $(DATA)/mushroom.conf
|
|
sed -i '$$a\model_out = "./host/result.model"' $(DATA)/mushroom.conf
|
|
sed -i '$$a\verbosity = 2' $(DATA)/mushroom.conf
|
|
sed -i '$$a\nthread = 2' $(DATA)/mushroom.conf
|
|
|
|
occlum_workspace: $(BIN) $(DATA)
|
|
rm -rf occlum_workspace && mkdir occlum_workspace
|
|
cd occlum_workspace && \
|
|
occlum init && \
|
|
jq '.process.default_mmap_size = "96MB"' Occlum.json > temp_Occlum.json && \
|
|
mv temp_Occlum.json Occlum.json && \
|
|
cp ../$(BIN) image/bin && \
|
|
cp ../xgboost_src/lib/libxgboost.so image/lib && \
|
|
mkdir -p image/$(DATA) && \
|
|
cp -r ../$(DATA)/* image/$(DATA) && \
|
|
occlum build
|
|
|
|
test: occlum_workspace
|
|
@./run_xgboost.sh
|
|
|
|
test-native: $(BIN) $(DATA)
|
|
rm -rf host
|
|
mkdir host
|
|
./$(BIN) ./$(DATA)/mushroom.conf
|
|
|
|
test-local-cluster: occlum_workspace
|
|
@$(XGBOOST_TRACKER)/dmlc-submit --cluster=local --num-workers=2 --log-level=DEBUG ./run_xgboost.sh
|
|
|
|
clean:
|
|
rm -rf $(BIN) $(DATA) tmp_* host occlum_workspace
|