refine_canary_compile_info

.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 }}

.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:

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",

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",