54 lines
1.6 KiB
Makefile
54 lines
1.6 KiB
Makefile
# Environment variables passed via elixir_make
|
|
# ROOT_DIR
|
|
# BUILD_ARCHIVE
|
|
# BUILD_INTERNAL_FLAGS
|
|
|
|
# System vars
|
|
TEMP ?= $(HOME)/.cache
|
|
|
|
# Public configuration
|
|
BUILD_MODE ?= opt # can also be dbg
|
|
BUILD_CACHE ?= $(TEMP)/xla_extension
|
|
OPENXLA_GIT_REPO ?= https://github.com/openxla/xla.git
|
|
|
|
OPENXLA_GIT_REV ?= fd58925adee147d38c25a085354e15427a12d00a
|
|
|
|
# Private configuration
|
|
BAZEL_FLAGS = --define "framework_shared_object=false" -c $(BUILD_MODE)
|
|
|
|
OPENXLA_NS = xla-$(OPENXLA_GIT_REV)
|
|
OPENXLA_DIR = $(BUILD_CACHE)/$(OPENXLA_NS)
|
|
OPENXLA_XLA_EXTENSION_NS = xla/extension
|
|
OPENXLA_XLA_EXTENSION_DIR = $(OPENXLA_DIR)/$(OPENXLA_XLA_EXTENSION_NS)
|
|
OPENXLA_XLA_BUILD_ARCHIVE = $(OPENXLA_DIR)/bazel-bin/$(OPENXLA_XLA_EXTENSION_NS)/xla_extension.tar.gz
|
|
|
|
$(BUILD_ARCHIVE): $(OPENXLA_DIR) extension/BUILD
|
|
rm -f $(OPENXLA_XLA_EXTENSION_DIR) && \
|
|
ln -s "$(ROOT_DIR)/extension" $(OPENXLA_XLA_EXTENSION_DIR) && \
|
|
cd $(OPENXLA_DIR) && \
|
|
bazel build $(BAZEL_FLAGS) $(BUILD_FLAGS) $(BUILD_INTERNAL_FLAGS) //$(OPENXLA_XLA_EXTENSION_NS):xla_extension && \
|
|
mkdir -p $(dir $(BUILD_ARCHIVE)) && \
|
|
cp -f $(OPENXLA_XLA_BUILD_ARCHIVE) $(BUILD_ARCHIVE)
|
|
|
|
# Clones OPENXLA
|
|
$(OPENXLA_DIR):
|
|
mkdir -p $(OPENXLA_DIR) && \
|
|
cp -r extension/patches $(OPENXLA_DIR) && \
|
|
cd $(OPENXLA_DIR) && \
|
|
git init && \
|
|
git remote add origin $(OPENXLA_GIT_REPO) && \
|
|
git fetch --depth 1 origin $(OPENXLA_GIT_REV) && \
|
|
git checkout FETCH_HEAD && \
|
|
bash patches/apply.sh && \
|
|
rm $(OPENXLA_DIR)/.bazelversion
|
|
|
|
# Print OPENXLA Dir
|
|
PTD:
|
|
@ echo $(OPENXLA_DIR)
|
|
|
|
clean:
|
|
cd $(OPENXLA_DIR) && bazel clean --expunge
|
|
rm -f $(OPENXLA_XLA_EXTENSION_DIR)
|
|
rm -rf $(OPENXLA_DIR)
|
|
rm -rf $(TARGET_DIR)
|