From 48e0eba3721d4e1e0009c279ad22481e93d150f9 Mon Sep 17 00:00:00 2001 From: Cheng Date: Wed, 30 Jul 2025 14:20:21 +0900 Subject: [PATCH] Use uv in cuda build --- .circleci/config.yml | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/.circleci/config.yml b/.circleci/config.yml index 4c37b22a3..600fea694 100644 --- a/.circleci/config.yml +++ b/.circleci/config.yml @@ -225,17 +225,17 @@ jobs: curl -sL https://github.com/ccache/ccache/releases/download/v4.11.3/ccache-4.11.3-linux-x86_64.tar.xz | tar xJf - sudo mv ccache-4.11.3-linux-x86_64/ccache /usr/bin/ccache rm -rf ccache-4.11.3-linux-x86_64 + curl -LsSf https://astral.sh/uv/install.sh | sh - run: name: Install Python package command: | - python3 -m venv env - source env/bin/activate + uv venv CMAKE_ARGS="-DMLX_BUILD_CUDA=ON -DCMAKE_CUDA_COMPILER=`which nvcc`" \ - pip install -e ".[dev]" -v + uv pip install -e ".[dev]" - run: name: Run Python tests command: | - source env/bin/activate + source .venv/bin/activate LOW_MEMORY=1 DEVICE=cpu python -m unittest discover python/tests -v LOW_MEMORY=1 DEVICE=gpu python -m tests discover python/tests -v - run: