Skip to content

Commit 1f466f9

Browse files
committed
fix windows
1 parent c626592 commit 1f466f9

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

.github/workflows/build-wheel.yml

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -203,7 +203,12 @@ jobs:
203203
- name: Build cuda.bindings Cython tests
204204
run: |
205205
pip install $(ls ${{ env.CUDA_BINDINGS_ARTIFACTS_DIR }}/*.whl)[test]
206-
bash ${{ env.CUDA_BINDINGS_CYTHON_TESTS_DIR }}/build_tests.sh
206+
if [[ "${{ inputs.host-platform }}" == linux* ]]; then
207+
export CPLUS_INCLUDE_PATH=${CUDA_HOME}/include:${CPLUS_INCLUDE_PATH}
208+
elif [[ "${{ inputs.host-platform }}" == win* ]]; then
209+
export CL="${CL} /I\"${CUDA_HOME}\include\""
210+
fi
211+
cythonize -3 -i ${{ env.CUDA_BINDINGS_CYTHON_TESTS_DIR }}/test_*.pyx
207212

208213
# For caching
209214
echo "PY_EXT_SUFFIX=$(python -c "import sysconfig; print(sysconfig.get_config_var('EXT_SUFFIX'))")" >> $GITHUB_ENV

0 commit comments

Comments
 (0)