|
78 | 78 | cmake ../native \ |
79 | 79 | -DCMAKE_BUILD_TYPE=Release \ |
80 | 80 | -DPYBIND11_FINDPYTHON=ON \ |
| 81 | + -Dpybind11_DIR=$(python -c "import pybind11; print(pybind11.get_cmake_dir())") \ |
81 | 82 | -DCMAKE_CUDA_ARCHITECTURES="80;86;89;90" \ |
82 | 83 | -DPYGPUKIT_MODULE_SUFFIX="_cu129" |
83 | 84 | cmake --build . --config Release -j$(nproc) |
@@ -125,6 +126,7 @@ jobs: |
125 | 126 | cmake ../native \ |
126 | 127 | -DCMAKE_BUILD_TYPE=Release \ |
127 | 128 | -DPYBIND11_FINDPYTHON=ON \ |
| 129 | + -Dpybind11_DIR=$(python -c "import pybind11; print(pybind11.get_cmake_dir())") \ |
128 | 130 | -DCMAKE_CUDA_ARCHITECTURES="80;86;89;90;100;120" \ |
129 | 131 | -DPYGPUKIT_MODULE_SUFFIX="_cu131" |
130 | 132 | cmake --build . --config Release -j$(nproc) |
@@ -290,11 +292,15 @@ jobs: |
290 | 292 | set "CUDA_PATH=C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.9" |
291 | 293 | set "PATH=%CUDA_PATH%\bin;%PATH%" |
292 | 294 |
|
| 295 | + @REM Get pybind11 cmake dir |
| 296 | + for /f "delims=" %%i in ('python -c "import pybind11; print(pybind11.get_cmake_dir())"') do set PYBIND11_DIR=%%i |
| 297 | +
|
293 | 298 | mkdir build-cu12 |
294 | 299 | cd build-cu12 |
295 | 300 | cmake ..\native -G Ninja ^ |
296 | 301 | -DCMAKE_BUILD_TYPE=Release ^ |
297 | 302 | -DPYBIND11_FINDPYTHON=ON ^ |
| 303 | + -Dpybind11_DIR="%PYBIND11_DIR%" ^ |
298 | 304 | -DCMAKE_CUDA_ARCHITECTURES="80;86;89;90" ^ |
299 | 305 | -DPYGPUKIT_MODULE_SUFFIX="_cu129" |
300 | 306 | cmake --build . --config Release |
@@ -354,11 +360,15 @@ jobs: |
354 | 360 | set "CUDA_PATH=C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v13.1" |
355 | 361 | set "PATH=%CUDA_PATH%\bin;%PATH%" |
356 | 362 |
|
| 363 | + @REM Get pybind11 cmake dir |
| 364 | + for /f "delims=" %%i in ('python -c "import pybind11; print(pybind11.get_cmake_dir())"') do set PYBIND11_DIR=%%i |
| 365 | +
|
357 | 366 | mkdir build-cu13 |
358 | 367 | cd build-cu13 |
359 | 368 | cmake ..\native -G Ninja ^ |
360 | 369 | -DCMAKE_BUILD_TYPE=Release ^ |
361 | 370 | -DPYBIND11_FINDPYTHON=ON ^ |
| 371 | + -Dpybind11_DIR="%PYBIND11_DIR%" ^ |
362 | 372 | -DCMAKE_CUDA_ARCHITECTURES="80;86;89;90;100;120" ^ |
363 | 373 | -DPYGPUKIT_MODULE_SUFFIX="_cu131" |
364 | 374 | cmake --build . --config Release |
|
0 commit comments