Skip to content

Commit d33a63b

Browse files
author
stoo
committed
tweaking compiler flags
1 parent 35e4066 commit d33a63b

File tree

1 file changed

+9
-4
lines changed

1 file changed

+9
-4
lines changed

.github/workflows/build-windows-wheel-test.yml

Lines changed: 9 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -367,14 +367,17 @@ jobs:
367367
exit 1
368368
}
369369
}
370-
371370
# Set environment variables for subsequent steps
372-
echo "INCLUDE=$include_paths;$env:INCLUDE" >> $env:GITHUB_ENV
371+
echo "INCLUDE=$include_paths;C:\sparsehash\src;$env:INCLUDE" >> $env:GITHUB_ENV
373372
echo "LIB=$lib_paths;$env:LIB" >> $env:GITHUB_ENV
374373
echo "PATH=$cuda_root\bin;$env:PATH" >> $env:GITHUB_ENV
375374
echo "CUDA_PATH=$cuda_root" >> $env:GITHUB_ENV
376375
echo "CUDA_HOME=$cuda_root" >> $env:GITHUB_ENV
377376
echo "CUDA_TOOLKIT_ROOT_DIR=$cuda_root" >> $env:GITHUB_ENV
377+
378+
# Set additional CUDA compile flags
379+
echo "NVCC_FLAGS=--use-local-env -Xcompiler /MD -Xcompiler /O2" >> $env:GITHUB_ENV
380+
echo "CUDA_NVCC_FLAGS=--use-local-env -Xcompiler /MD -Xcompiler /O2" >> $env:GITHUB_ENV
378381
379382
# Verify CUDA setup
380383
Write-Host "CUDA Environment Setup:"
@@ -402,19 +405,21 @@ jobs:
402405
# Verify NVCC
403406
Write-Host "NVCC version:"
404407
nvcc --version
405-
406408
# Set build environment (Windows)
407409
- name: Set build environment (Windows)
408410
if: runner.os == 'Windows'
409411
shell: pwsh
410412
run: |
411413
# Set environment variables for optimized Windows build
412-
echo "CL=/O1 /MP4" >> $env:GITHUB_ENV
414+
echo "CL=/O2 /MP4 /DWIN32 /D_WINDOWS" >> $env:GITHUB_ENV
413415
echo "DISTUTILS_USE_SDK=1" >> $env:GITHUB_ENV
414416
echo "MSSdk=1" >> $env:GITHUB_ENV
415417
echo "FORCE_CUDA=1" >> $env:GITHUB_ENV
416418
echo "TORCH_CUDA_ARCH_LIST=7.5;8.0;8.6;8.9" >> $env:GITHUB_ENV
417419
echo "MAX_JOBS=4" >> $env:GITHUB_ENV
420+
# Add MSVC specific defines
421+
echo "_CRT_SECURE_NO_WARNINGS=1" >> $env:GITHUB_ENV
422+
echo "_SILENCE_TR1_NAMESPACE_DEPRECATION_WARNING=1" >> $env:GITHUB_ENV
418423
419424
- name: Set build environment (Linux)
420425
if: runner.os == 'Linux'

0 commit comments

Comments
 (0)