pub struct AiterInstaller { /* private fields */ }Expand description
The AITER installer.
Implementations§
Source§impl AiterInstaller
impl AiterInstaller
Sourcepub fn new(config: AiterConfig) -> Self
pub fn new(config: AiterConfig) -> Self
Create a new AITER installer with the given config.
Sourcepub fn with_defaults() -> Self
pub fn with_defaults() -> Self
Create with default config.
Sourcepub fn dependencies(&self) -> &[&str]
pub fn dependencies(&self) -> &[&str]
AITER depends on PyTorch and ROCm.
Sourcepub fn validate_dependencies(&self, installed_components: &[&str]) -> Result<()>
pub fn validate_dependencies(&self, installed_components: &[&str]) -> Result<()>
Validate that all dependencies are satisfied.
Sourcepub fn git_clone_url(&self) -> &str
pub fn git_clone_url(&self) -> &str
Get the git clone URL for AITER.
Sourcepub fn build_git_clone_command(&self, target_dir: &str) -> ShellCommand
pub fn build_git_clone_command(&self, target_dir: &str) -> ShellCommand
Construct the git clone command.
Sourcepub fn build_deps_install_command(&self) -> ShellCommand
pub fn build_deps_install_command(&self) -> ShellCommand
Construct the pip install command for AITER build dependencies.
The original script installs: packaging, pybind11, pandas, einops, psutil, numpy, setuptools, wheel, typing-extensions, cmake, ninja.
Sourcepub fn build_pip_install_command(&self, src_dir: &str) -> ShellCommand
pub fn build_pip_install_command(&self, src_dir: &str) -> ShellCommand
Construct the pip install command for AITER from source.
The proven working approach uses pip install --no-cache-dir .
(NOT --no-build-isolation --no-deps which skips build dependencies).
The --break-system-packages flag is added for global installs.
pub fn build_rocm_env(&self, rocm_env: &RocmEnv) -> Vec<(String, String)>
Sourcepub fn aiter_jit_env(&self) -> Vec<(String, String)>
pub fn aiter_jit_env(&self) -> Vec<(String, String)>
Construct the AITER JIT directory environment.
Sourcepub fn resolve_primary_arch(&self) -> String
pub fn resolve_primary_arch(&self) -> String
Resolve the primary GPU architecture (first item before semicolon).
Auto Trait Implementations§
impl Freeze for AiterInstaller
impl RefUnwindSafe for AiterInstaller
impl Send for AiterInstaller
impl Sync for AiterInstaller
impl Unpin for AiterInstaller
impl UnsafeUnpin for AiterInstaller
impl UnwindSafe for AiterInstaller
Blanket Implementations§
Source§impl<T> BorrowMut<T> for Twhere
T: ?Sized,
impl<T> BorrowMut<T> for Twhere
T: ?Sized,
Source§fn borrow_mut(&mut self) -> &mut T
fn borrow_mut(&mut self) -> &mut T
Source§impl<T> Instrument for T
impl<T> Instrument for T
Source§fn instrument(self, span: Span) -> Instrumented<Self>
fn instrument(self, span: Span) -> Instrumented<Self>
Source§fn in_current_span(self) -> Instrumented<Self>
fn in_current_span(self) -> Instrumented<Self>
Source§impl<T> IntoEither for T
impl<T> IntoEither for T
Source§fn into_either(self, into_left: bool) -> Either<Self, Self>
fn into_either(self, into_left: bool) -> Either<Self, Self>
self into a Left variant of Either<Self, Self>
if into_left is true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read moreSource§fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
self into a Left variant of Either<Self, Self>
if into_left(&self) returns true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read more