File tree Expand file tree Collapse file tree 2 files changed +13
-6
lines changed Expand file tree Collapse file tree 2 files changed +13
-6
lines changed Original file line number Diff line number Diff line change @@ -27,16 +27,16 @@ jobs:
27
27
runs-on : ${{ matrix.runs-on }}
28
28
strategy :
29
29
matrix :
30
- os : [ linux-intel, linux-arm, macos-arm ]
30
+ os : [ linux-intel, linux-arm, macos-arm, windows ]
31
31
include :
32
32
- archs : auto64
33
33
platform : auto
34
34
- os : linux-intel
35
35
runs-on : ubuntu-latest
36
36
- os : linux-arm
37
37
runs-on : ubuntu-24.04-arm
38
- # - os: windows
39
- # runs-on: windows-latest
38
+ - os : windows
39
+ runs-on : windows-latest
40
40
- os : macos-arm
41
41
# macos-14+ (including latest) are ARM64 runners
42
42
runs-on : macos-latest
Original file line number Diff line number Diff line change @@ -19,9 +19,16 @@ execute_process(
19
19
find_package (nanobind CONFIG REQUIRED )
20
20
21
21
# Find jaxlib include dir
22
- execute_process (
23
- COMMAND /usr/bin/env python3 -c "import inspect; import jaxlib; import pathlib; p = pathlib.Path(inspect.getfile(jaxlib)); print(p.parent / 'include')"
24
- OUTPUT_STRIP_TRAILING_WHITESPACE OUTPUT_VARIABLE jaxlib_INCLUDE_DIR )
22
+ IF (WIN32 )
23
+ execute_process (
24
+ COMMAND python -c "import inspect; import jaxlib; import pathlib; p = pathlib.Path(inspect.getfile(jaxlib)); print(p.parent / 'include')"
25
+ OUTPUT_STRIP_TRAILING_WHITESPACE OUTPUT_VARIABLE jaxlib_INCLUDE_DIR )
26
+ ELSE ()
27
+ execute_process (
28
+ COMMAND /usr/bin/env python3 -c "import inspect; import jaxlib; import pathlib; p = pathlib.Path(inspect.getfile(jaxlib)); print(p.parent / 'include')"
29
+ OUTPUT_STRIP_TRAILING_WHITESPACE OUTPUT_VARIABLE jaxlib_INCLUDE_DIR )
30
+ ENDIF ()
31
+
25
32
26
33
# We are now ready to compile the actual extension module
27
34
nanobind_add_module (
You can’t perform that action at this time.
0 commit comments