treeboost 0.1.0-beta.1

[[bench]]
harness = false
name = "competitors"
path = "benches/competitors.rs"

[[bench]]
harness = false
name = "correctness"
path = "benches/correctness.rs"

[[bench]]
harness = false
name = "profile"
path = "benches/profile.rs"

[[bin]]
name = "treeboost"
path = "src/main.rs"

[dependencies.bincode]
version = "1.3"

[dependencies.bytemuck]
features = ["derive"]
version = "1.19"

[dependencies.chrono]
version = "0.4"

[dependencies.clap]
features = ["derive"]
version = "4.5"

[dependencies.csv]
version = "1.3"

[dependencies.cudarc]
default-features = false
features = ["driver", "nvrtc", "dynamic-linking", "cuda-version-from-build-system"]
optional = true
version = "=0.18.2"

[dependencies.faer]
version = "0.23"

[dependencies.numpy]
optional = true
version = "0.23"

[dependencies.ordered-float]
version = "4.5"

[dependencies.polars]
features = ["parquet", "lazy", "dtype-full"]
version = "0.45"

[dependencies.pollster]
optional = true
version = "0.4"

[dependencies.pyo3]
features = ["extension-module"]
optional = true
version = "0.23"

[dependencies.rand]
version = "0.8"

[dependencies.rayon]
version = "1.10"

[dependencies.rkyv]
version = "0.8"

[dependencies.rustc-hash]
version = "2.1"

[dependencies.serde]
features = ["derive"]
version = "1.0"

[dependencies.serde_json]
version = "1.0"

[dependencies.tdigest]
version = "0.2"

[dependencies.thiserror]
version = "2.0"

[dependencies.wgpu]
optional = true
version = "27"

[dev-dependencies.approx]
version = "0.5"

[dev-dependencies.criterion]
features = ["html_reports"]
version = "0.5"

[dev-dependencies.forust-ml]
version = "0.4"

[dev-dependencies.gbdt]
features = ["enable_training"]
version = "0.1"

[dev-dependencies.tempfile]
version = "3.15"

[[example]]
name = "4bit_benchmark"
path = "examples/4bit_benchmark.rs"

[[example]]
name = "autotuner"
path = "examples/autotuner.rs"

[[example]]
name = "backend_benchmark"
path = "examples/backend_benchmark.rs"

[[example]]
name = "cache_pollution_study"
path = "examples/cache_pollution_study.rs"

[[example]]
name = "detailed_breakdown"
path = "examples/detailed_breakdown.rs"

[[example]]
name = "era_histogram_benchmark"
path = "examples/era_histogram_benchmark.rs"

[[example]]
name = "era_split_comparison"
path = "examples/era_split_comparison.rs"

[[example]]
name = "find_crossover"
path = "examples/find_crossover.rs"

[[example]]
name = "full_gpu_benchmark"
path = "examples/full_gpu_benchmark.rs"

[[example]]
name = "fused_benchmark"
path = "examples/fused_benchmark.rs"

[[example]]
name = "histogram_benchmark"
path = "examples/histogram_benchmark.rs"

[[example]]
name = "histogram_scaling"
path = "examples/histogram_scaling.rs"

[[example]]
name = "large_dataset_profiler"
path = "examples/large_dataset_profiler.rs"

[[example]]
name = "register_cache_benchmark"
path = "examples/register_cache_benchmark.rs"

[[example]]
name = "scalar_poc"
path = "examples/scalar_poc.rs"

[[example]]
name = "simd_benchmark"
path = "examples/simd_benchmark.rs"

[[example]]
name = "soa_histogram_poc"
path = "examples/soa_histogram_poc.rs"

[[example]]
name = "subgroup_benchmark"
path = "examples/subgroup_benchmark.rs"

[[example]]
name = "tensor_tile_poc"
path = "examples/tensor_tile_poc.rs"

[[example]]
name = "training_breakdown"
path = "examples/training_breakdown.rs"

[[example]]
name = "training_loop_deep_dive"
path = "examples/training_loop_deep_dive.rs"

[[example]]
name = "workgroup_size_benchmark"
path = "examples/workgroup_size_benchmark.rs"

[features]
cuda = ["dep:cudarc"]
default = []
gpu = ["dep:wgpu", "dep:pollster"]
python = ["dep:pyo3", "dep:numpy"]

[lib]
crate-type = ["cdylib", "rlib"]
name = "treeboost"
path = "src/lib.rs"

[package]
authors = ["Farhan Syah"]
autobenches = false
autobins = false
autoexamples = false
autolib = false
autotests = false
build = false
categories = ["science", "algorithms"]
description = "High-performance Gradient Boosted Decision Tree engine for large-scale tabular data"
edition = "2021"
keywords = ["machine-learning", "gbdt", "gradient-boosting", "decision-tree", "xgboost"]
license = "Apache-2.0"
name = "treeboost"
readme = "README.md"
repository = "https://github.com/ml-rust/treeboost"
version = "0.1.0-beta.1"

[profile.bench]
codegen-units = 1
lto = "thin"

[profile.release]
codegen-units = 1
lto = "thin"
opt-level = 3

[[test]]
name = "integration"
path = "tests/integration.rs"