rusty-stack 0.3.0

A comprehensive ML environment installer and platform toolkit optimized for AMD GPUs with ROCm support
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56

//! Shared installer infrastructure modules.
//!
//! These modules provide common functionality used by all component installers:
//! - **package management**: unified interface for apt/dnf/pacman
//! - **distro detection**: thin facade over `platform::detection`
//! - **package mappings**: per-distro package name resolution
//! - **guard**: structured error handling, logging, progress, Python validation
//! - **utils**: color-aware output, command existence, print helpers
//! - **env_validation**: `.mlstack_env` checking with sensible defaults
//! - **rocm_env**: thin facade over `platform::linux` for ROCm paths/versions
//! - **ui_helper**: CLI arg parsing, path validation, git helpers
//!
//! # Validation Assertions
//!
//! - **VAL-INFRA-001**: Module structure compiles and integrates
//! - **VAL-INFRA-019**: No code duplication with platform modules

pub mod benchmark_common;
pub mod distro;
pub mod env_validation;
pub mod guard;
pub mod nvidia_blocklist;
pub mod package_manager;
pub mod package_mappings;
pub mod rocm_env;
pub mod sealed_token;
pub mod telemetry;
pub mod ui_helper;
pub mod utils;

// Re-export key types for convenience
pub use benchmark_common::{
    format_elapsed, is_dry_run, normalize_pci_bus_id, now_timestamp, resolve_log_dir, BenchLogger,
    BenchTimer, BenchmarkResult, BenchmarkResults,
};
pub use distro::DistroFacade;
pub use env_validation::{EnvValidationResult, EnvVars};
pub use guard::{
    detect_python_version, is_python_supported, is_python_supported_for_rocm_torch, log_error,
    log_info, log_warn, InstallerError, LogLevel, ProgressTracker, PythonVersion,
    MAX_ROCM_TORCH_PYTHON, MIN_PYTHON_VERSION,
};
pub use nvidia_blocklist::{
    contaminated_packages, filter_requirements, filter_requirements_file, is_cuda_nvidia_package,
    is_cuda_wheel_url,
};
pub use package_manager::{DryRunResult, PackageManagerFacade, PackageOperation};
pub use package_mappings::map_package_name;
pub use rocm_env::RocmEnv;
pub use sealed_token::SealedToken;
pub use telemetry::{submit_build_report, BuildReport, BuildReportArtifacts, BuildReportStatus};
pub use ui_helper::{is_system_path, UiArgError, UiArgs};
pub use utils::{
    command_exists, fix_pip_cache_ownership, get_colors, Colors, PipCacheFixResult,
    PythonPkgManager,
};