forked from NVIDIA/TensorRT
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Makefile
56 lines (45 loc) · 1.93 KB
/
Makefile
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
#
# SPDX-FileCopyrightText: Copyright (c) 1993-2022 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
# SPDX-License-Identifier: Apache-2.0
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
#
.ONESHELL: # Run all rules in the same shell to make it easier to run tests with environment variables
.PHONY: test_venv test leak_check clean build install docs
NPROC ?= 8
BUILD_DIR := build
TESTS_DIR := tests
PYTEST_OPTS := -v -x --durations=5 --failed-first --new-first --script-launch-mode=subprocess
# Tests also check that docs can build
test: docs install
export PYTHONPATH=$(CURDIR):$${PYTHON_PATH}
export POLYGRAPHY_INTERNAL_CORRECTNESS_CHECKS=1
# Some tests need to be run serially - we annotate those with a `serial` marker.
python3 -m pytest $(TESTS_DIR) -m "serial" $(PYTEST_OPTS) && \
python3 -m pytest $(TESTS_DIR) -n $(NPROC) --dist=loadscope -m "not serial" $(PYTEST_OPTS)
leak_check:
export PYTHONPATH=$(CURDIR):$${PYTHON_PATH}
export POLYGRAPHY_INTERNAL_CORRECTNESS_CHECKS=1
export PYTHONMALLOC=malloc
valgrind --leak-check=full python3 -m pytest $(TESTS_DIR) -v --durations=5 2>&1 | tee leak-check.log
clean:
rm -rf dist/ $(BUILD_DIR)/ polygraphy.egg-info/
build:
python3 setup.py bdist_wheel
install_deps: build
- python3 -m pip install colored wheel
install: install_deps
- python3 -m pip install --force-reinstall $(CURDIR)/dist/*.whl
docs:
mkdir -p $(BUILD_DIR)/docs
python3 `which sphinx-build` docs $(BUILD_DIR)/docs/ -j $(NPROC) -W