Skip to main content

apple_mps/
lib.rs

1#![doc = include_str!("../README.md")]
2#![cfg_attr(docsrs, feature(doc_cfg))]
3#![allow(clippy::missing_const_for_fn)]
4#![allow(clippy::missing_errors_doc)]
5#![allow(clippy::module_name_repetitions)]
6#![allow(clippy::must_use_candidate)]
7
8pub mod core;
9pub mod error;
10pub mod ffi;
11pub mod filters;
12pub mod image;
13pub mod matrix;
14pub mod ndarray;
15pub mod neural;
16pub mod ray;
17pub mod state;
18
19pub use crate::core::{
20    device_options, hint_temporary_memory_high_water_mark, preferred_device,
21    set_heap_cache_duration, supports_mtl_device, CommandBuffer as MpsCommandBuffer, Predicate,
22    PreferredDevice,
23};
24pub use crate::error::{Error, Result};
25pub use crate::filters::{
26    HistogramInfo, ImageAdd, ImageBilinearScale, ImageBox, ImageConvolution, ImageGaussianBlur,
27    ImageHistogram, ImageLanczosScale, ImageMedian, ImageReduceRowMax, ImageReduceRowMean,
28    ImageReduceRowMin, ImageReduceRowSum, ImageScaleAndAdd, ImageSobel, ImageStatisticsMean,
29    ImageStatisticsMinAndMax, ImageThresholdBinary, ScaleTransform,
30};
31pub use crate::image::{
32    feature_channel_format, image_edge_mode, image_layout, kernel_options, Image, ImageDescriptor,
33    ImageReadWriteParams, ImageRegion,
34};
35pub use crate::matrix::{
36    data_type, data_type_size, Matrix, MatrixDescriptor, MatrixMultiplication,
37    MatrixMultiplicationDescriptor, Vector, VectorDescriptor,
38};
39pub use crate::ndarray::{NDArray, NDArrayDescriptor, NDArrayIdentity, NDArrayMatrixMultiplication};
40pub use crate::neural::{
41    cnn_accumulator_precision_option, cnn_convolution_flags, cnn_convolution_weights_layout,
42    nn_regularization_type, rnn_bidirectional_combine_mode, rnn_sequence_direction,
43    CnnConvolution, CnnConvolutionDescriptor, CnnConvolutionWeightsAndBiasesState,
44    CnnNeuronReluNode, CnnPoolingMaxNode, CnnSoftMaxNode, CnnUpsamplingNearestNode,
45    GruDescriptor, LstmDescriptor, NNGraph, NNImageNode, NNOptimizer, NNOptimizerAdam,
46    NNOptimizerDescriptor, NNOptimizerRmsProp, NNOptimizerStochasticGradientDescent,
47    RnnDescriptor, RnnImageInferenceLayer, RnnRecurrentImageState, RnnSingleGateDescriptor,
48};
49pub use crate::ray::{
50    acceleration_structure_status, acceleration_structure_usage, cull_mode, intersection_data_type,
51    intersection_type, polygon_type, ray_data_type, winding, PolygonAccelerationStructure,
52    RayIntersector, SVGF,
53};
54pub use crate::state::{
55    state_batch_increment_read_count, state_batch_resource_size, state_batch_synchronize,
56    state_resource_type, State, StateResourceList, StateTextureInfo,
57};