MLX
Loading...
Searching...
No Matches
Classes | Namespaces | Macros | Typedefs | Enumerations | Functions
gemm.h File Reference
#include <Metal/Metal.hpp>

Go to the source code of this file.

Classes

class  MPS::MatrixDescriptor
 
class  MPS::Matrix
 
class  MPS::Kernel
 
class  MPS::MatrixMultiplication
 
class  MPS::VectorDescriptor
 
class  MPS::Vector
 
class  MPS::MatrixVectorMultiplication
 

Namespaces

namespace  MTL
 
namespace  MTL::Private
 
namespace  MTL::Private::Class
 
namespace  MTL::Private::Selector
 
namespace  MPS
 

Macros

#define _MPS_PRIVATE_CLS(symbol)   (MTL::Private::Class::s_k##symbol)
 
#define _MPS_PRIVATE_SEL(accessor)   (MTL::Private::Selector::s_k##accessor)
 

Typedefs

typedef enum MPS::DataType MPS::DataType
 

Enumerations

enum  MPS::DataType : uint32_t {
  MPS::DataTypeFloatBit = 0x10000000 , MPS::DataTypeAlternateEncodingBit = 0x80000000 , MPS::DataTypeFloat16 = DataTypeFloatBit | 16 , MPS::DataTypeFloat32 = DataTypeFloatBit | 32 ,
  MPS::DataTypeBFloat16 = DataTypeAlternateEncodingBit | DataTypeFloat16
}
 

Functions

 MTL::Private::Class::_MTL_PRIVATE_DEF_CLS (MPSMatrixDescriptor)
 
 MTL::Private::Class::_MTL_PRIVATE_DEF_CLS (MPSMatrix)
 
 MTL::Private::Class::_MTL_PRIVATE_DEF_CLS (MPSVectorDescriptor)
 
 MTL::Private::Class::_MTL_PRIVATE_DEF_CLS (MPSVector)
 
 MTL::Private::Class::_MTL_PRIVATE_DEF_CLS (MPSKernel)
 
 MTL::Private::Class::_MTL_PRIVATE_DEF_CLS (MPSMatrixMultiplication)
 
 MTL::Private::Class::_MTL_PRIVATE_DEF_CLS (MPSMatrixVectorMultiplication)
 
 MTL::Private::Selector::_MTL_PRIVATE_DEF_SEL (matrixDescriptorWithRows_columns_rowBytes_dataType, "matrixDescriptorWithRows:columns:rowBytes:dataType:")
 
 MTL::Private::Selector::_MTL_PRIVATE_DEF_SEL (matrixDescriptorWithRows_columns_matrices_rowBytes_matrixBytes_dataType, "matrixDescriptorWithRows:columns:matrices:rowBytes:matrixBytes:dataType:")
 
 MTL::Private::Selector::_MTL_PRIVATE_DEF_SEL (rows, "rows")
 
 MTL::Private::Selector::_MTL_PRIVATE_DEF_SEL (initWithBuffer_descriptor, "initWithBuffer:descriptor:")
 
 MTL::Private::Selector::_MTL_PRIVATE_DEF_SEL (initWithDevice_, "initWithDevice:transposeLeft:transposeRight:" "resultRows:resultColumns:interiorColumns:alpha:beta:")
 
 MTL::Private::Selector::_MTL_PRIVATE_DEF_SEL (encodeToCommandBuffer_leftMatrix_rightMatrix_resultMatrix, "encodeToCommandBuffer:leftMatrix:rightMatrix:resultMatrix:")
 
 MTL::Private::Selector::_MTL_PRIVATE_DEF_SEL (setLeftMatrixOrigin_, "setLeftMatrixOrigin:")
 
 MTL::Private::Selector::_MTL_PRIVATE_DEF_SEL (setRightMatrixOrigin_, "setRightMatrixOrigin:")
 
 MTL::Private::Selector::_MTL_PRIVATE_DEF_SEL (setResultMatrixOrigin_, "setResultMatrixOrigin:")
 
 MTL::Private::Selector::_MTL_PRIVATE_DEF_SEL (setBatchStart_, "setBatchStart:")
 
 MTL::Private::Selector::_MTL_PRIVATE_DEF_SEL (setBatchSize_, "setBatchSize:")
 
 MTL::Private::Selector::_MTL_PRIVATE_DEF_SEL (vectorDescriptorWithLength_dataType, "vectorDescriptorWithLength:dataType:")
 
 MTL::Private::Selector::_MTL_PRIVATE_DEF_SEL (vectorDescriptorWithLength_vectors_vectorBytes_dataType, "vectorDescriptorWithLength:vectors:vectorBytes:dataType:")
 
 MTL::Private::Selector::_MTL_PRIVATE_DEF_SEL (initWithDevice_transpose_rows_columns_alpha_beta, "initWithDevice:transpose:rows:columns:alpha:beta:")
 
 MTL::Private::Selector::_MTL_PRIVATE_DEF_SEL (encodeToCommandBuffer_inputMatrix_inputVector_resultVector, "encodeToCommandBuffer:inputMatrix:inputVector:resultVector:")
 

Macro Definition Documentation

◆ _MPS_PRIVATE_CLS

#define _MPS_PRIVATE_CLS ( symbol)    (MTL::Private::Class::s_k##symbol)

◆ _MPS_PRIVATE_SEL

#define _MPS_PRIVATE_SEL ( accessor)    (MTL::Private::Selector::s_k##accessor)