Skip to content

Commit c605949

Browse files
committed
build pytorch with 'numpy>2.0'
Signed-off-by: Anatoly Myachev <[email protected]>
1 parent 6f59e89 commit c605949

File tree

1 file changed

+7
-1
lines changed

1 file changed

+7
-1
lines changed

.github/actions/setup-pytorch/action.yml

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -78,7 +78,7 @@ runs:
7878
shell: bash
7979
env:
8080
# Increase this value to reset cache
81-
CACHE_NUMBER: 18
81+
CACHE_NUMBER: 19
8282
run: |
8383
ONEAPI_LINK=$(readlink /opt/intel/oneapi || true)
8484
ONEAPI_KEY=$(sha256sum /opt/intel/installed.txt 2> /dev/null | cut -d\ -f1 || true)
@@ -124,6 +124,12 @@ runs:
124124
# FIXME: Compatibility with versions of CMake older than 3.5 has been removed, this brakes compilation of third_party/protobuf:
125125
# CMake Error at third_party/protobuf/cmake/CMakeLists.txt:2 (cmake_minimum_required)
126126
pip install 'cmake<4.0.0'
127+
# Try to avoid the following error:
128+
# A module that was compiled using NumPy 1.x cannot be run in
129+
# NumPy 2.2.6 as it may crash. To support both 1.x and 2.x
130+
# versions of NumPy, modules must be compiled with NumPy 2.0.
131+
# Some module may need to rebuild instead e.g. with 'pybind11>=2.12'.
132+
pip install 'numpy>2.0'
127133
pip install -r requirements.txt
128134
USE_XCCL=1 USE_STATIC_MKL=1 python -m build --wheel --no-isolation 2>&1 | grep -v \
129135
"Double arithmetic operation is not supported on this platform with FP64 conversion emulation mode (poison FP64 kernels is enabled)." | grep -v '^$'

0 commit comments

Comments
 (0)