forked from jiuyuan/InfiniTensor
35 lines
632 B
Makefile
35 lines
632 B
Makefile
.PHONY : build clean install-python test-cpp test-onnx
|
|
|
|
TYPE ?= release
|
|
CUDA ?= off
|
|
INTELCPU ?= off
|
|
|
|
CMAKE_OPT = -DCMAKE_BUILD_TYPE=$(TYPE)
|
|
|
|
ifeq ($(CUDA), ON)
|
|
CMAKE_OPT += -DUSE_CUDA=ON
|
|
endif
|
|
|
|
ifeq ($(INTELCPU), ON)
|
|
CMAKE_OPT += -DUSE_INTELCPU=ON -DCMAKE_CXX_COMPILER=dpcpp
|
|
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
|