diff --git a/Makefile b/Makefile index 3661da78..4de85392 100644 --- a/Makefile +++ b/Makefile @@ -5,7 +5,7 @@ CUDA ?= off CMAKE_OPT = -DCMAKE_BUILD_TYPE=$(TYPE) -ifeq ($(CUDA), on) +ifeq ($(CUDA), ON) CMAKE_OPT += -DUSE_CUDA=ON endif diff --git a/README.md b/README.md index 8e109331..1f6c07fa 100644 --- a/README.md +++ b/README.md @@ -5,16 +5,27 @@ ``` bash # Enter the root of InfiniTensor source test/script/env_lotus.sh -mkdir build && cd build -cmake -DUSE_CUDA=ON .. && make -j 12 +make CUDA=ON ``` +### Make Commands + +- `make`/`make build`: Builds the project; +- `make install-python`: Builds the project then install the python frontend; +- `make test-cpp`: Builds the project then run cpp unit tests; +- `make test-onnx`: Run python unit tests; + +--- + +> Sets env: `CUDA=ON` to enable cuda. + ### CMake Options There are several configurable CMake options, see the [CMakeLists.txt file](/CMakeLists.txt#L5). - If `USE_BACKTRACE` is `ON`, `libdw-dev` have to be installed. See the README of [backward-cpp](https://github.com/bombela/backward-cpp) for details. - If `USE_PROTOBUF` is `ON`, `protobuf` have to be installed. See the README of [protobuf](https://github.com/protocolbuffers/protobuf) for details. +- If `USE_CUDA` is `ON`, `cuda` have to be installed. ## Contributor Guide