Struct MlStackInstaller 

pub struct MlStackInstaller { /* private fields */ }

The ML Stack installer.

Implementations

impl MlStackInstaller

pub fn new(config: MlStackConfig) -> Self

Create a new ML Stack installer with the given config.

pub fn with_defaults() -> Self

Create with default config.

pub fn install_order() -> Vec<MlStackComponent>

Get the ordered list of core components to install.

The order matches the original install_ml_stack.sh:

1. ROCm Config
2. PyTorch
3. ONNX Runtime
4. MIGraphX
5. Megatron-LM
6. Flash Attention
7. RCCL
8. MPI

pub fn rocm_config_path(&self) -> String

Get the ROCm config file path.

pub fn build_migraphx_install_command( &self, distro: &DistroFacade, ) -> SystemCommand

Construct the MIGraphX package install command per distro.

pub fn build_rccl_install_command(&self, distro: &DistroFacade) -> SystemCommand

Construct the RCCL package install command per distro.

pub fn build_mpi_install_command(&self, distro: &DistroFacade) -> SystemCommand

Construct the MPI system package install command per distro.

pub fn build_megatron_clone_command(&self, target_dir: &str) -> SystemCommand

Construct the Megatron-LM git clone command into target_dir.

pub fn build_megatron_install_command(&self) -> PipCommand

Construct the Megatron-LM pip install command.

pub fn prerequisite_commands(&self) -> &[&str]

Get the list of prerequisite commands to check.

pub fn check_prerequisite(&self, cmd: &str) -> bool

Check a single prerequisite.