Merge pull request #67 from shwina/add-nvtx-dir

Add support for `NVTX_DIR`, drop Python 3.7

python/docs/install.rst

@@ -17,3 +17,13 @@ Or `conda`:
 ::
 
    conda install -c conda-forge nvtx
+
+On Windows, or another system for which pre-built packages are not
+published, you can try to install from source. This requires: (1) that
+Cython is installed, and (2) that the ``NVTX_PREFIX`` environment
+variable is set to the directory containing ``nvtx3``, the C header
+library. On systems with CUDA installed, this is typically the CUDA
+include directory (e.g., ``/usr/local/cuda/include``).
+::
+
+   NVTX_PREFIX=/path/to/nvtx/prefix/ python -m pip install nvtx --no-binary nvtx

python/setup.py

@@ -23,6 +23,9 @@
 include_dirs = [os.path.dirname(sysconfig.get_path("include")),]
 library_dirs = [get_python_lib()]
 
+if nvtx_include_dir := os.getenv("NVTX_PREFIX"):
+    include_dirs.insert(0, nvtx_include_dir)
+
 extensions = [
     Extension(
         "*",
@@ -55,7 +58,7 @@
 
 setup(
     name="nvtx",
-    version="0.2.5",
+    version="0.2.6",
     description="PyNVTX - Python code annotation library",
     url="https://github.com/NVIDIA/nvtx",
     author="NVIDIA Corporation",

python/tools/build-wheels.sh

@@ -14,7 +14,7 @@ function repair_wheel {
 export C_INCLUDE_PATH=/io/c/include
 
 # Compile wheels
-for PY_VERSION in 37 38 39 310; do
+for PY_VERSION in 38 39 310; do
     PYBIN=/opt/python/cp${PY_VERSION}*/bin/
     ${PYBIN}/pip wheel /io/python/ --no-deps -w wheels/
 done
@@ -25,7 +25,7 @@ for whl in wheels/*.whl; do
 done
 
 # Install packages and test
-for PY_VERSION in 37 38 39 310; do
+for PY_VERSION in 38 39 310; do
     PYBIN=/opt/python/cp${PY_VERSION}*/bin
     ${PYBIN}/pip install nvtx --no-index -f /io/python/wheels/
     ${PYBIN}/pip install -r /io/python/test_requirements.txt