pyproject.toml

[build-system]
requires = ["hatchling >= 1.26"]
build-backend = "hatchling.build"

[project]
name = "ml-tuning"
version = "0.1.0.dev0"
description = "A repo to store all code related to machine learning based tuning, which aims to provide a faster tuning process to find best kernel parameters for GEMM."
readme = "README.md"
requires-python = ">=3.10"
dependencies = [
    "numpy>=2.2.6",
    "ollama>=0.6.0",
    "pandas>=2.3.3",
    "pydantic>=2.12.3",
    "pyyaml>=6.0.3",
    "rocisa",
    "tabulate>=0.9.0",
    "tensile",
    "typer>=0.19.2",
    "typing-extensions>=4.15.0",
]

[project.scripts]
confgen = "config_generator.main:app"

[tool.hatch.build]
packages = ["config_generator"]

[tool.uv.sources]
rocisa = { path = "../rocm-libraries/projects/hipblaslt/tensilelite/rocisa" }
tensile = { path = "../rocm-libraries/projects/hipblaslt/tensilelite" }