Cmake and ninja move

This commit is contained in:
Daniel Han 2025-07-01 00:13:00 -07:00
commit b221ffc216
2 changed files with 5 additions and 3 deletions

View file

@ -1,8 +1,6 @@
[build-system]
# Should be mirrored in requirements/build.txt
requires = [
"cmake>=3.26",
"ninja",
"packaging>=24.2",
"setuptools>=77.0.3,<80.0.0",
"setuptools-scm>=8.0",
@ -13,7 +11,7 @@ build-backend = "setuptools.build_meta"
name = "unsloth"
description = "2-5X faster LLM finetuning"
readme = "README.md"
requires-python = ">=3.9,<3.13"
requires-python = ">=3.9"
dynamic = ["version", "dependencies", "optional-dependencies"]
license = { file = "LICENSE" }
keywords = ["ai", "llm",]

View file

@ -770,6 +770,10 @@ if HAS_HIP:
os.mkdir('thirdparties')
os.chdir('thirdparties')
# Cmake and ninja
subprocess.check_call(['pip', 'install', 'cmake>=3.26'])
subprocess.check_call(['pip', 'install', 'ninja'])
# Extract ROCm GPU arch from environment variable. If unset, then detect ROCm arch from rocminfo
# Refer to https://github.com/bitsandbytes-foundation/bitsandbytes/blob/1abd5e781013a085f86586b30a248dc769909668/bitsandbytes/cuda_specs.py#L81
# TODO(billishyahao): need to triage rocminfo unavailable observation from https://github.com/bitsandbytes-foundation/bitsandbytes/issues/1444