diff --git a/.github/workflows/canary.yml b/.github/workflows/canary.yml index ab61cb1..7d52ae9 100644 --- a/.github/workflows/canary.yml +++ b/.github/workflows/canary.yml @@ -66,6 +66,7 @@ jobs: nexfort-ref: ${{ inputs.nexfort-ref }} python-publish: false upload_oss: true + canary: "1" os: ${{ matrix.os }} python: ${{ matrix.python }} torch_version: ${{ matrix.torch_version }} diff --git a/.github/workflows/wheels_build.yml b/.github/workflows/wheels_build.yml index 65e79ac..de4cc34 100644 --- a/.github/workflows/wheels_build.yml +++ b/.github/workflows/wheels_build.yml @@ -47,6 +47,11 @@ on: python-publish: required: false type: boolean + canary: + type: string + required: false + default: "0" + description: "Canay compile" secrets: GH_NEX_PRIV_TOKEN: required: true @@ -103,6 +108,11 @@ on: python-publish: required: false type: boolean + canary: + type: string + required: false + default: "0" + description: "Canay compile" secrets: OSS_ACCESS_KEY_ID: required: true @@ -121,6 +131,8 @@ env: DISTUTILS_USE_SDK: 1 # otherwise distutils will complain on windows about multiple versions of msvc TWINE_USERNAME: __token__ OSS_BUCKET: nexfort-whl + NEXFORT_COMPILE_CANARY: inputs.canary + NEXFORT_BRANCH_NAME: inputs.nexfort-ref jobs: generate-build-date: diff --git a/pyproject.toml b/pyproject.toml index 7c0cdbb..4123073 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -58,6 +58,8 @@ environment-pass = [ "PIP_FIND_LINKS", "PIP_NO_BUILD_ISOLATION", "PY", + "NEXFORT_COMPILE_CANARY", + "NEXFORT_BRANCH_NAME", "NEXFORT_BUILD_CYTHONIZE", "NEXFORT_BUILD_WITH_CUTLASS", "TORCH_CUDA_ARCH_LIST", diff --git a/pyproject_cu124.toml b/pyproject_cu124.toml index ab33f41..69fa656 100644 --- a/pyproject_cu124.toml +++ b/pyproject_cu124.toml @@ -58,6 +58,8 @@ environment-pass = [ "PIP_FIND_LINKS", "PIP_NO_BUILD_ISOLATION", "PY", + "NEXFORT_COMPILE_CANARY", + "NEXFORT_BRANCH_NAME", "NEXFORT_BUILD_CYTHONIZE", "NEXFORT_BUILD_WITH_CUTLASS", "TORCH_CUDA_ARCH_LIST",