InfiniTensor/Makefile

.PHONY : build clean install-python test-cpp test-onnx

TYPE ?= release
CUDA ?= off

CMAKE_OPT = -DCMAKE_BUILD_TYPE=$(TYPE)

ifeq ($(CUDA), ON)
	CMAKE_OPT += -DUSE_CUDA=ON
endif

build:
	mkdir -p build/$(TYPE)
	cd build/$(TYPE) && cmake $(CMAKE_OPT) ../.. && make -j8

clean:
	rm -rf build

install-python: build
	cp build/$(TYPE)/backend*.so pyinfinitensor/src/pyinfinitensor
	pip install pyinfinitensor/

test-cpp: build
	@echo
	cd build/$(TYPE) && make test

test-onnx:
	@echo
	python3 pyinfinitensor/tests/test_onnx.py