Check out and build a specific version of bitsandbytes

learning-at-home · Feb 11, 2025 · fd69b64 · fd69b64
1 parent 801bb4f
commit fd69b64
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/.github/workflows/run-tests-on-modal.yml b/.github/workflows/run-tests-on-modal.yml
@@ -10,7 +10,7 @@ concurrency:
   cancel-in-progress: true
 
 jobs:
-  test:
+  run_tests:
     runs-on: ubuntu-latest
     strategy:
       matrix:
@@ -45,7 +45,7 @@ jobs:
         run: |
           modal run modal_ci.py::run_tests
 
-  codecov:
+  measure_coverage:
     runs-on: ubuntu-latest
     env:
       MODAL_TOKEN_ID: ${{ secrets.MODAL_TOKEN_ID }}

diff --git a/modal_ci.py b/modal_ci.py
@@ -9,7 +9,12 @@
     .apt_install(["git", "procps", "build-essential", "cmake"])
     .pip_install_from_requirements("requirements-dev.txt")
     .pip_install_from_requirements("requirements.txt")
-    .pip_install("bitsandbytes==0.45.2")
+    .run_commands(
+        [
+            "git clone --branch 0.45.2 --depth 1 https://github.com/bitsandbytes-foundation/bitsandbytes.git",
+            "cd bitsandbytes && cmake -DCOMPUTE_BACKEND=cpu -S . && make && pip --no-cache install . ",
+        ]
+    )
     .add_local_dir("hivemind", remote_path="/root/hivemind/hivemind")
     .add_local_file("requirements.txt", remote_path="/root/hivemind/requirements.txt")
     .add_local_file("requirements-dev.txt", remote_path="/root/hivemind/requirements-dev.txt")
@@ -23,6 +28,7 @@
 
 codecov_secret = modal.Secret.from_dict({"CODECOV_TOKEN": os.getenv("CODECOV_TOKEN")})
 
+
 def setup_environment():
     os.chdir("/root/hivemind")