trustformers-core 0.1.1

Core traits and utilities for TrustformeRS
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

//! Common imports and utilities for Metal backend modules
//!
//! This module provides shared imports and helper functions used across all Metal backend modules.

// Re-export all common types and traits needed by Metal backend modules
pub(crate) use crate::device::Device;
pub(crate) use crate::errors::{Result, TrustformersError};
pub(crate) use crate::tensor::Tensor;

#[cfg(all(target_os = "macos", feature = "metal"))]
pub(crate) use metal::{
    foreign_types::ForeignType, Buffer, CommandQueue, CompileOptions, Device as MetalDevice,
    MTLResourceOptions,
};

#[cfg(all(target_os = "macos", feature = "metal"))]
pub(crate) use std::collections::HashMap;

#[cfg(all(target_os = "macos", feature = "metal"))]
pub(crate) use std::mem;

#[cfg(all(target_os = "macos", feature = "metal"))]
pub(crate) use std::sync::Arc;

// Import scirs2-core MPS for GPU-to-GPU optimized operations
#[cfg(all(target_os = "macos", feature = "metal"))]
pub(crate) use scirs2_core::gpu::backends::MPSOperations;

// Re-export objc2 types commonly used in Metal backend
#[cfg(all(target_os = "macos", feature = "metal"))]
pub(crate) use objc2::rc::Retained;
#[cfg(all(target_os = "macos", feature = "metal"))]
pub(crate) use objc2::runtime::ProtocolObject;
#[cfg(all(target_os = "macos", feature = "metal"))]
pub(crate) use objc2_metal::MTLBuffer as ObjC2Buffer;