diff --git a/.github/workflows/build_windows.yml b/.github/workflows/build_windows.yml
index ad21a7b..3401f61 100644
--- a/.github/workflows/build_windows.yml
+++ b/.github/workflows/build_windows.yml
@@ -28,9 +28,6 @@ jobs:
   build_windows_wheels:
     name: Build wheels and Upload (Windows x86_64, GitHub hosted runner, ${{ inputs.flash-attn-version }}, ${{ inputs.python-version }}, ${{ inputs.torch-version }}, ${{ inputs.cuda-version }})
     runs-on: windows-2022
-    env:
-      MAX_JOBS: 2
-      NVCC_THREADS: 2
     steps:
       - uses: actions/checkout@v4
 
@@ -73,7 +70,16 @@ jobs:
 
       - name: Build wheels
         shell: pwsh
+        env:
+          MAX_JOBS: 2
+          NVCC_THREADS: 2
         run: |
+          $env:MAX_JOBS = "${{ env.MAX_JOBS }}"
+          $env:NVCC_THREADS = "${{ env.NVCC_THREADS }}"
+          $env:FLASH_ATTENTION_FORCE_BUILD = "TRUE"
+          $env:NVCC_FLAGS = "-w --disable-warnings"
+          $env:CXXFLAGS = "/w"
+          $env:CFLAGS = "/w"
           .\build_windows.ps1 -FlashAttnVersion "${{ inputs.flash-attn-version }}" -PythonVersion "${{ inputs.python-version }}" -TorchVersion "${{ inputs.torch-version }}" -CudaVersion "${{ inputs.cuda-version }}"
           $wheelName = Get-ChildItem -Path "flash-attention\dist\*.whl" | Select-Object -First 1 | ForEach-Object { $_.Name }
           echo "wheel_name=$wheelName" >> $env:GITHUB_ENV
diff --git a/.github/workflows/build_windows_code_build.yml b/.github/workflows/build_windows_code_build.yml
index b54ff96..1730949 100644
--- a/.github/workflows/build_windows_code_build.yml
+++ b/.github/workflows/build_windows_code_build.yml
@@ -76,6 +76,12 @@ jobs:
           MAX_JOBS: 6
           NVCC_THREADS: 4
         run: |
+          $env:MAX_JOBS = "${{ env.MAX_JOBS }}"
+          $env:NVCC_THREADS = "${{ env.NVCC_THREADS }}"
+          $env:FLASH_ATTENTION_FORCE_BUILD = "TRUE"
+          $env:NVCC_FLAGS = "-w --disable-warnings"
+          $env:CXXFLAGS = "/w"
+          $env:CFLAGS = "/w"
           .\build_windows.ps1 -FlashAttnVersion "${{ inputs.flash-attn-version }}" -PythonVersion "${{ inputs.python-version }}" -TorchVersion "${{ inputs.torch-version }}" -CudaVersion "${{ inputs.cuda-version }}"
           $wheelName = Get-ChildItem -Path "flash-attention\dist\*.whl" | Select-Object -First 1 | ForEach-Object { $_.Name }
           echo "wheel_name=$wheelName" >> $env:GITHUB_ENV
diff --git a/build_windows.ps1 b/build_windows.ps1
index c6f2a45..d67b5d6 100644
--- a/build_windows.ps1
+++ b/build_windows.ps1
@@ -61,6 +61,9 @@ $env:BUILD_TARGET = "cuda"
 # Use environment variables from workflow if available, otherwise use defaults
 if (-not $env:MAX_JOBS) { $env:MAX_JOBS = "2" }
 if (-not $env:NVCC_THREADS) { $env:NVCC_THREADS = "2" }
+Write-Host "Environment variables:"
+Write-Host "  MAX_JOBS: $env:MAX_JOBS"
+Write-Host "  NVCC_THREADS: $env:NVCC_THREADS"
 $env:FLASH_ATTENTION_FORCE_BUILD = "TRUE"
 $env:NVCC_FLAGS = "-w --disable-warnings"
 $env:CXXFLAGS = "/w"