diff --git a/.github/workflows/whl-build-all.yaml b/.github/workflows/whl-build-all.yaml index 2edaaee..522ed2b 100644 --- a/.github/workflows/whl-build-all.yaml +++ b/.github/workflows/whl-build-all.yaml @@ -48,28 +48,28 @@ jobs: run: | ./build_scripts/build_wheel.sh python3.8 env: - CUDA_TAG: cu113 + VERSION_CUDA_TAG: cu113 EXTRA_REQUIRES: nvidia-cuda-cupti-cu11==11.7.101,nvidia-cuda-runtime-cu11==11.7.99 - name: Build Python3.9 wheel run: | ./build_scripts/build_wheel.sh python3.9 env: - CUDA_TAG: cu113 + VERSION_CUDA_TAG: cu113 EXTRA_REQUIRES: nvidia-cuda-cupti-cu11==11.7.101,nvidia-cuda-runtime-cu11==11.7.99 - name: Build Python3.10 wheel run: | ./build_scripts/build_wheel.sh python3.10 env: - CUDA_TAG: cu113 + VERSION_CUDA_TAG: cu113 EXTRA_REQUIRES: nvidia-cuda-cupti-cu11==11.7.101,nvidia-cuda-runtime-cu11==11.7.99 - name: Build Python3.11 wheel run: | ./build_scripts/build_wheel.sh python3.11 env: - CUDA_TAG: cu113 + VERSION_CUDA_TAG: cu113 EXTRA_REQUIRES: nvidia-cuda-cupti-cu11==11.7.101,nvidia-cuda-runtime-cu11==11.7.99 - name: Upload wheels to S3 @@ -111,28 +111,28 @@ jobs: run: | ./build_scripts/build_wheel.sh python3.8 env: - CUDA_TAG: cu116 + VERSION_CUDA_TAG: cu116 EXTRA_REQUIRES: nvidia-cuda-cupti-cu11==11.7.101,nvidia-cuda-runtime-cu11==11.7.99 - name: Build Python3.9 wheel run: | ./build_scripts/build_wheel.sh python3.9 env: - CUDA_TAG: cu116 + VERSION_CUDA_TAG: cu116 EXTRA_REQUIRES: nvidia-cuda-cupti-cu11==11.7.101,nvidia-cuda-runtime-cu11==11.7.99 - name: Build Python3.10 wheel run: | ./build_scripts/build_wheel.sh python3.10 env: - CUDA_TAG: cu116 + VERSION_CUDA_TAG: cu116 EXTRA_REQUIRES: nvidia-cuda-cupti-cu11==11.7.101,nvidia-cuda-runtime-cu11==11.7.99 - name: Build Python3.11 wheel run: | ./build_scripts/build_wheel.sh python3.11 env: - CUDA_TAG: cu116 + VERSION_CUDA_TAG: cu116 EXTRA_REQUIRES: nvidia-cuda-cupti-cu11==11.7.101,nvidia-cuda-runtime-cu11==11.7.99 - name: Upload wheels to S3 @@ -174,28 +174,28 @@ jobs: run: | ./build_scripts/build_wheel.sh python3.8 env: - CUDA_TAG: cu117 + VERSION_CUDA_TAG: cu117 EXTRA_REQUIRES: nvidia-cuda-cupti-cu11==11.7.101,nvidia-cuda-runtime-cu11==11.7.99 - name: Build Python3.9 wheel run: | ./build_scripts/build_wheel.sh python3.9 env: - CUDA_TAG: cu117 + VERSION_CUDA_TAG: cu117 EXTRA_REQUIRES: nvidia-cuda-cupti-cu11==11.7.101,nvidia-cuda-runtime-cu11==11.7.99 - name: Build Python3.10 wheel run: | ./build_scripts/build_wheel.sh python3.10 env: - CUDA_TAG: cu117 + VERSION_CUDA_TAG: cu117 EXTRA_REQUIRES: nvidia-cuda-cupti-cu11==11.7.101,nvidia-cuda-runtime-cu11==11.7.99 - name: Build Python3.11 wheel run: | ./build_scripts/build_wheel.sh python3.11 env: - CUDA_TAG: cu117 + VERSION_CUDA_TAG: cu117 EXTRA_REQUIRES: nvidia-cuda-cupti-cu11==11.7.101,nvidia-cuda-runtime-cu11==11.7.99 - name: Upload wheels to S3 @@ -237,28 +237,28 @@ jobs: run: | ./build_scripts/build_wheel.sh python3.8 env: - CUDA_TAG: cu118 + VERSION_CUDA_TAG: cu118 EXTRA_REQUIRES: nvidia-cuda-cupti-cu11==11.8.87,nvidia-cuda-runtime-cu11==11.8.89 - name: Build Python3.9 wheel run: | ./build_scripts/build_wheel.sh python3.9 env: - CUDA_TAG: cu118 + VERSION_CUDA_TAG: cu118 EXTRA_REQUIRES: nvidia-cuda-cupti-cu11==11.8.87,nvidia-cuda-runtime-cu11==11.8.89 - name: Build Python3.10 wheel run: | ./build_scripts/build_wheel.sh python3.10 env: - CUDA_TAG: cu118 + VERSION_CUDA_TAG: cu118 EXTRA_REQUIRES: nvidia-cuda-cupti-cu11==11.8.87,nvidia-cuda-runtime-cu11==11.8.89 - name: Build Python3.11 wheel run: | ./build_scripts/build_wheel.sh python3.11 env: - CUDA_TAG: cu118 + VERSION_CUDA_TAG: cu118 EXTRA_REQUIRES: nvidia-cuda-cupti-cu11==11.8.87,nvidia-cuda-runtime-cu11==11.8.89 - name: Upload wheels to S3 @@ -300,25 +300,25 @@ jobs: run: | ./build_scripts/build_wheel.sh python3.8 env: - CUDA_TAG: cu121 + VERSION_CUDA_TAG: cu121 - name: Build Python3.9 wheel run: | ./build_scripts/build_wheel.sh python3.9 env: - CUDA_TAG: cu121 + VERSION_CUDA_TAG: cu121 - name: Build Python3.10 wheel run: | ./build_scripts/build_wheel.sh python3.10 env: - CUDA_TAG: cu121 + VERSION_CUDA_TAG: cu121 - name: Build Python3.11 wheel run: | ./build_scripts/build_wheel.sh python3.11 env: - CUDA_TAG: cu121 + VERSION_CUDA_TAG: cu121 - name: Upload wheels to S3 run: | diff --git a/analyzer/setup.py b/analyzer/setup.py index bc23394..696579f 100644 --- a/analyzer/setup.py +++ b/analyzer/setup.py @@ -17,7 +17,7 @@ We define two additional environment arguments during build to include dependencies for different versions of CUDA we're targeting. -1. CUDA_TAG : This follows the "version number". For example, if version="1.0" and CUDA_TAG =cu121, then the +1. VERSION_CUDA_TAG : This follows the "version number". For example, if version="1.0" and VERSION_CUDA_TAG =cu121, then the version we pass into setuptools.setup would be "1.0+cu121" 2. EXTRA_REQUIRES: Depends on the CUDA version, we need additional pip libraries to provide CUPTI (among other things). @@ -44,8 +44,8 @@ "incremental" ] -CUDA_TAG = os.getenv("CUDA_TAG", default="") -if CUDA_TAG : CUDA_TAG = "+" + CUDA_TAG +VERSION_CUDA_TAG = os.getenv("VERSION_CUDA_TAG", default="") +if VERSION_CUDA_TAG : VERSION_CUDA_TAG = "+" + VERSION_CUDA_TAG EXTRA_REQUIRES = os.getenv("EXTRA_REQUIRES", default="nvidia-cuda-cupti-cu12,nvidia-cuda-runtime-cu12").split(",") PACKAGE_DATA = { @@ -149,7 +149,7 @@ def find_meta(meta): if __name__ == "__main__": setup( name=NAME, - version=VERSION + CUDA_TAG , + version=VERSION + VERSION_CUDA_TAG , description=find_meta("description"), license=find_meta("license"), author=find_meta("author"),