Cmake and ninja move
This commit is contained in:
parent
0ceeab0821
commit
b221ffc216
2 changed files with 5 additions and 3 deletions
|
|
@ -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",]
|
||||
|
|
|
|||
4
setup.py
4
setup.py
|
|
@ -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
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue