zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-06-24 09:21:16 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Change the linux test to ubuntu 24.04

Browse Source
This commit is contained in:
Angelos Katharopoulos 2025-01-20 20:55:10 -08:00
parent 6fe2b82926
commit 4515866024
2 changed files with 17 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

33
.circleci/config.yml
View File

@ -70,8 +70,8 @@ jobs:
git push -f origin gh-pages git push -f origin gh-pages
linux_build_and_test: linux_build_and_test:
docker: machine:
- image: cimg/python:3.9 image: ubuntu-2404:2024.11.1
steps: steps:
- checkout - checkout
@ -84,34 +84,33 @@ jobs:
- run: - run:
name: Install dependencies name: Install dependencies
command: | command: |
pip install --upgrade cmake sudo add-apt-repository -y ppa:deadsnakes/ppa
pip install nanobind==2.4.0 sudo apt-get update -y
pip install numpy sudo apt-get install -y python3.9 python3.9-distutils python3.9-dev
python3.9 -m pip install --upgrade cmake
python3.9 -m pip install nanobind==2.4.0
python3.9 -m pip install numpy
sudo apt-get update sudo apt-get update
sudo apt-get install libblas-dev liblapack-dev liblapacke-dev sudo apt-get install libopenblas-dev liblapacke-dev openmpi-bin libopenmpi-dev
sudo add-apt-repository ppa:ubuntu-toolchain-r/ppa -y
sudo apt-get update
sudo apt install g++-13 gcc-13
sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-13 100 --slave /usr/bin/g++ g++ /usr/bin/g++-13
- run: - run:
name: Install Python package name: Install Python package
command: | command: |
CMAKE_ARGS="-DMLX_BUILD_METAL=OFF" \ CMAKE_ARGS="-DMLX_BUILD_METAL=OFF -DPython_EXECUTABLE=/usr/bin/python3.9" \
CMAKE_BUILD_PARALLEL_LEVEL=`nproc` \ CMAKE_BUILD_PARALLEL_LEVEL=`nproc` \
python3 setup.py build_ext --inplace python3.9 setup.py build_ext --inplace
CMAKE_ARGS="-DMLX_BUILD_METAL=OFF" \ CMAKE_ARGS="-DMLX_BUILD_METAL=OFF -DPython_EXECUTABLE=/usr/bin/python3.9" \
CMAKE_BUILD_PARALLEL_LEVEL=`nproc` \ CMAKE_BUILD_PARALLEL_LEVEL=`nproc` \
python3 setup.py develop python3.9 setup.py develop
- run: - run:
name: Generate package stubs name: Generate package stubs
command: | command: |
echo "stubs" echo "stubs"
pip install typing_extensions python3.9 -m pip install typing_extensions
python setup.py generate_stubs python3.9 setup.py generate_stubs
- run: - run:
name: Run Python tests name: Run Python tests
command: | command: |
python3 -m unittest discover python/tests -v python3.9 -m unittest discover python/tests -v
- run: - run:
name: Build CPP only name: Build CPP only
command: | command: |

1
mlx/backend/common/jit_compiler.cpp
View File

@ -2,6 +2,7 @@
#include "mlx/backend/common/jit_compiler.h" #include "mlx/backend/common/jit_compiler.h"
#include <algorithm>
#include <sstream> #include <sstream>
#include <vector> #include <vector>