Struct TritonInstaller 

pub struct TritonInstaller { /* private fields */ }

The Triton installer.

Implementations

impl TritonInstaller

pub fn new(config: TritonConfig) -> Self

Create a new Triton installer with the given config.

pub fn with_defaults() -> Self

Create with default config.

pub fn select_branch(&self) -> TritonBranch

Determine the Triton branch to use based on channel and GPU arch.

Matches the original script logic:

• Preview channel or gfx12xx GPUs → preview branch
• Otherwise → MLIR branch with stable fallback

pub fn git_clone_url(&self) -> &'static str

Get the git clone URL.

pub fn build_git_clone_command(&self, target_dir: &str) -> ShellCommand

Construct the git clone command.

pub fn build_git_checkout_command(&self) -> ShellCommand

Construct the git checkout command for the selected branch.

pub fn build_prerequisites_command(&self) -> ShellCommand

Construct the build prerequisites pip install command.

Installs pybind11, cmake, ninja, and other build tools.

pub fn build_pip_install_command(&self, src_dir: &str) -> ShellCommand

Construct the Triton pip install from source command.

This builds Triton from the cloned source with ROCm flags:

• GPU_ARCHS=gfx1100
• TRITON_ROCM=1
• MAX_JOBS=