mirror of
https://github.com/ml-explore/mlx.git
synced 2025-06-27 03:11:16 +08:00
127 lines
35 KiB
JavaScript
127 lines
35 KiB
JavaScript
var searchData=
|
|
[
|
|
['b_0',['B',['../structmlx_1_1steel_1_1_attn_params.html#a1cba7fedbd02e157922619195997cf4f',1,'mlx::steel::AttnParams']]],
|
|
['b_1',['b',['../unionbool4__or__uint.html#a47d77eac47598fe420f8f04a615f76ca',1,'bool4_or_uint::b'],['../structmlx_1_1core_1_1_dtype.html#adb1ea8b45a0c53e04a0e73b168702715a92eb5ffee6ae2fec3ad71c777531578f',1,'mlx::core::Dtype::b']]],
|
|
['b_5fstr_5fk_2',['B_str_k',['../structmlx_1_1steel_1_1_block_m_m_a.html#aa71400922babd388177f228c2c82b211',1,'mlx::steel::BlockMMA']]],
|
|
['b_5fstr_5fn_3',['B_str_n',['../structmlx_1_1steel_1_1_block_m_m_a.html#a49538190209e522ddbef45fe95563d17',1,'mlx::steel::BlockMMA']]],
|
|
['backward_4',['BACKWARD',['../namespacepocketfft_1_1detail.html#a9d1eaa7469c018c39e745733eab9a9c3',1,'pocketfft::detail::BACKWARD'],['../namespacepocketfft.html#a9d1eaa7469c018c39e745733eab9a9c3',1,'pocketfft::BACKWARD']]],
|
|
['barrier_5',['barrier',['../structmlx_1_1core_1_1metal_1_1_command_encoder.html#a7f028c6ca48e75bf2c1806b5b8cfc90e',1,'mlx::core::metal::CommandEncoder::barrier()'],['../structmlx_1_1core_1_1_command_encoder.html#a7f028c6ca48e75bf2c1806b5b8cfc90e',1,'mlx::core::CommandEncoder::barrier()']]],
|
|
['base_6',['Base',['../classmlx_1_1core_1_1_log.html#a044a23e8b1422984628e1cd5ab506421',1,'mlx::core::Log']]],
|
|
['base_5fsimd_2eh_7',['base_simd.h',['../base__simd_8h.html',1,'']]],
|
|
['base_5fwh_8',['base_wh',['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_general.html#aca37adba6f148579eb1cd0a7800a5cfe',1,'mlx::steel::Conv2DInputBlockLoaderGeneral::base_wh'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_general.html#a6c46564bf1a96a02791dd432cc9c883e',1,'mlx::steel::Conv2DWeightBlockLoaderGeneral::base_wh']]],
|
|
['base_5fww_9',['base_ww',['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_general.html#a32d020c6715d06f7de360877fcb7b6e4',1,'mlx::steel::Conv2DInputBlockLoaderGeneral::base_ww'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_general.html#a230f0e581f9b8227b9ee68760b3b1503',1,'mlx::steel::Conv2DWeightBlockLoaderGeneral::base_ww']]],
|
|
['basemmafrag_10',['BaseMMAFrag',['../structmlx_1_1steel_1_1_base_m_m_a_frag.html',1,'mlx::steel']]],
|
|
['basemmafrag_3c_20accumtype_2c_20kfragsize_2c_20kfragsize_20_3e_11',['BaseMMAFrag< AccumType, kFragSize, kFragSize >',['../structmlx_1_1steel_1_1_base_m_m_a_frag.html',1,'mlx::steel']]],
|
|
['basemmafrag_3c_20t_2c_208_2c_208_20_3e_12',['BaseMMAFrag< T, 8, 8 >',['../structmlx_1_1steel_1_1_base_m_m_a_frag_3_01_t_00_018_00_018_01_4.html',1,'mlx::steel']]],
|
|
['batch_5fndim_13',['batch_ndim',['../structmlx_1_1steel_1_1_g_e_m_m_params.html#a640dc138a8bf7b2b5bed6a436b429c2f',1,'mlx::steel::GEMMParams']]],
|
|
['batch_5fsize_14',['batch_size',['../struct_read_writer.html#a689f4890c1d2ce33fc6da7550beec735',1,'ReadWriter']]],
|
|
['batch_5fstride_5fa_15',['batch_stride_a',['../structmlx_1_1steel_1_1_g_e_m_m_params.html#aa9efc32581bba343b096123482507eee',1,'mlx::steel::GEMMParams']]],
|
|
['batch_5fstride_5fb_16',['batch_stride_b',['../structmlx_1_1steel_1_1_g_e_m_m_params.html#a5fba5117664fbab81e22abf1e8c8fbc8',1,'mlx::steel::GEMMParams']]],
|
|
['batch_5fstride_5fc_17',['batch_stride_c',['../structmlx_1_1steel_1_1_g_e_m_m_add_m_m_params.html#ac977827a0de9650b7110df029324fd60',1,'mlx::steel::GEMMAddMMParams']]],
|
|
['batch_5fstride_5fd_18',['batch_stride_d',['../structmlx_1_1steel_1_1_g_e_m_m_params.html#a4fc69951d02a9ce2d3e11b356557deb9',1,'mlx::steel::GEMMParams']]],
|
|
['bcols_19',['BCOLS',['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_large_filter.html#a29fbeeacdf5b6feeb74815ced255fa5a',1,'mlx::steel::Conv2DInputBlockLoaderLargeFilter::BCOLS'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_filter.html#ac3b40db720055350bba59d614ea1dd79',1,'mlx::steel::Conv2DInputBlockLoaderSmallFilter::BCOLS'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader.html#a86519729ef0561686bb86e474c95b93d',1,'mlx::steel::Conv2DWeightBlockLoader::BCOLS'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_channels.html#a9229d22e0a02d96825eb5a57c8cb95ac',1,'mlx::steel::Conv2DInputBlockLoaderSmallChannels::BCOLS'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_small_channels.html#a8b6cf53a10514310d01f4d6459053a57',1,'mlx::steel::Conv2DWeightBlockLoaderSmallChannels::BCOLS'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_general.html#a3d6272d000f8ea79d9b3b5228bdca20f',1,'mlx::steel::Conv2DInputBlockLoaderGeneral::BCOLS'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_general.html#a1843921cd67926002bb0dcccf3048eb6',1,'mlx::steel::Conv2DWeightBlockLoaderGeneral::BCOLS']]],
|
|
['bcols_5fpacked_20',['BCOLS_PACKED',['../struct_quantized_block_loader.html#a1392a5278cf6e090ea80ebe7c4ac5fbb',1,'QuantizedBlockLoader']]],
|
|
['begin_21',['begin',['../classmlx_1_1core_1_1array.html#a76b258b169d7d73419ebbf85340fb914',1,'mlx::core::array']]],
|
|
['bernoulli_22',['bernoulli',['../namespacemlx_1_1core_1_1random.html#a1c601b637f60071dfc85cad19a841744',1,'mlx::core::random::bernoulli(const array &p, const Shape &shape, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1random.html#aaa49f6c2af5496822fa09435e54275cb',1,'mlx::core::random::bernoulli(const array &p, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1random.html#aa9e360f9cb7bd23221352ed9e31d83c2',1,'mlx::core::random::bernoulli(T p, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1random.html#a2778876cd2318fec69cd1f3fc0955d68',1,'mlx::core::random::bernoulli(T p, const Shape &shape, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1random.html#ad7eb4467e2f9d5f74a5607b29a935b6e',1,'mlx::core::random::bernoulli(const std::optional< array > &key=std::nullopt, StreamOrDevice s={})']]],
|
|
['beta_23',['beta',['../structmlx_1_1steel_1_1_transform_axpby.html#a5fc726f085bafd1acbc391886f7fb8b6',1,'mlx::steel::TransformAxpby::beta'],['../structmlx_1_1steel_1_1_g_e_m_m_add_m_m_params.html#ac0ce4d8a6014f8adb29fd0a0bb23139f',1,'mlx::steel::GEMMAddMMParams::beta']]],
|
|
['bf16_2eh_24',['bf16.h',['../backend_2metal_2kernels_2jit_2bf16_8h.html',1,'(Global Namespace)'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html',1,'(Global Namespace)'],['../backend_2metal_2kernels_2metal__3__1_2bf16_8h.html',1,'(Global Namespace)'],['../types_2bf16_8h.html',1,'(Global Namespace)']]],
|
|
['bf16_5fmath_2eh_25',['bf16_math.h',['../bf16__math_8h.html',1,'']]],
|
|
['bfloat16_26',['bfloat16',['../structmlx_1_1core_1_1_dtype.html#ade845ef5dcebead13a37fe696436e1daa444fe01f3a7a54d1809aef0912846a47',1,'mlx::core::Dtype::bfloat16'],['../namespacemlx_1_1core.html#a514cf8b4e6f0a6af3a867e752f4338f7',1,'mlx::core::bfloat16']]],
|
|
['bfloat16_5ft_27',['bfloat16_t',['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a7782de82393104dd4ad754ce3b316e82',1,'bfloat16_t: bf16.h'],['../backend_2metal_2kernels_2metal__3__1_2bf16_8h.html#a58e15a77da988b9104fee00cdf8b280e',1,'bfloat16_t: bf16.h'],['../namespacemlx_1_1core.html#acb5d16c9b83778c7621c38e522e0060b',1,'mlx::core::bfloat16_t']]],
|
|
['bfloat16_5fto_5fuint16_28',['bfloat16_to_uint16',['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a1420e191fa60d707dce327d0938e3088',1,'bfloat16_to_uint16(const bfloat16_t x): bf16.h'],['../backend_2metal_2kernels_2metal__3__1_2bf16_8h.html#a1420e191fa60d707dce327d0938e3088',1,'bfloat16_to_uint16(const bfloat16_t x): bf16.h']]],
|
|
['bfloat_5fbinop_29',['bfloat_binop',['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a7694892a131c0e31e5153c088cccb707',1,'bfloat_binop: bf16.h'],['../types_2bf16_8h.html#a7694892a131c0e31e5153c088cccb707',1,'bfloat_binop: bf16.h']]],
|
|
['bfloat_5fbinop_5fbase_30',['bfloat_binop_base',['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a78c92beda4436da9a2e520fa98c59f70',1,'bfloat_binop_base: bf16.h'],['../types_2bf16_8h.html#a78c92beda4436da9a2e520fa98c59f70',1,'bfloat_binop_base: bf16.h']]],
|
|
['bfloat_5fbinop_5fhelper_31',['bfloat_binop_helper',['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ac7ff36230dab17e8f17b7a7c80888594',1,'bfloat_binop_helper: bf16.h'],['../types_2bf16_8h.html#ac7ff36230dab17e8f17b7a7c80888594',1,'bfloat_binop_helper: bf16.h']]],
|
|
['bfloat_5fbitop_32',['bfloat_bitop',['../types_2bf16_8h.html#aac9ba86d4bf05bcda1936494f9b9b4d3',1,'bf16.h']]],
|
|
['bfloat_5fbits_5fto_5ffloat_33',['bfloat_bits_to_float',['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a3b33ae338dc4f223d0f3c748de07bad1',1,'bf16.h']]],
|
|
['bfloat_5fcompop_34',['bfloat_compop',['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a330a0883503cb640f1cf628a7ca50239',1,'bfloat_compop: bf16.h'],['../types_2bf16_8h.html#a330a0883503cb640f1cf628a7ca50239',1,'bfloat_compop: bf16.h']]],
|
|
['bfloat_5finplace_5fbitop_35',['bfloat_inplace_bitop',['../types_2bf16_8h.html#af13b46bc58e6e6f675ae47aabec37711',1,'bf16.h']]],
|
|
['bfloat_5finplace_5fop_36',['bfloat_inplace_op',['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a4ac82467fbc674e990090f482b9c1e5c',1,'bfloat_inplace_op: bf16.h'],['../types_2bf16_8h.html#aee905053f51f76e0c1af94199714d514',1,'bfloat_inplace_op: bf16.h']]],
|
|
['bfloat_5finplace_5fop_5faddr_5fspace_5fhelper_37',['bfloat_inplace_op_addr_space_helper',['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#af30a2cbd2c3415516203b83bd21872f8',1,'bfloat_inplace_op_addr_space_helper: bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a1457da931c28fa4e2500daa4e6441e8b',1,'bfloat_inplace_op_addr_space_helper: bf16.h']]],
|
|
['bfloat_5finplace_5fop_5fhelper_38',['bfloat_inplace_op_helper',['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a2846fd11b5e19b435e9f7ef0998c9b1d',1,'bfloat_inplace_op_helper: bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#afe5988aa8147be2bafda6a5b7792fe15',1,'bfloat_inplace_op_helper: bf16.h']]],
|
|
['bfs_5fmax_5fwidth_39',['bfs_max_width',['../namespacemlx_1_1core_1_1env.html#ac3266e1259a64c8b56bdc6c7029179f2',1,'mlx::core::env']]],
|
|
['bi_40',['bi',['../struct_quantized_block_loader.html#a85041d72225a2095659c70509291a906',1,'QuantizedBlockLoader::bi'],['../structmlx_1_1steel_1_1_block_loader.html#a9ef13742bcdf07532d8f09394928a8af',1,'mlx::steel::BlockLoader::bi'],['../structmlx_1_1steel_1_1_block_loader_t.html#a6964273994b06d6cf8ef7e59fb10bb35',1,'mlx::steel::BlockLoaderT::bi'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_large_filter.html#a8e53b0a9951cb840d922cc285b257ee3',1,'mlx::steel::Conv2DInputBlockLoaderLargeFilter::bi'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_filter.html#ae3af75287f279d2cdeef189126740d4c',1,'mlx::steel::Conv2DInputBlockLoaderSmallFilter::bi'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader.html#a8c5e74003600132954cb953616e1a026',1,'mlx::steel::Conv2DWeightBlockLoader::bi'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_channels.html#a9eb024e2fc6f07345f87fbf7141c0d16',1,'mlx::steel::Conv2DInputBlockLoaderSmallChannels::bi'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_small_channels.html#ae3b9f21f72e5e6c541c9978f55d354c7',1,'mlx::steel::Conv2DWeightBlockLoaderSmallChannels::bi'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_general.html#a32a3a91fa715b82f36e05ceb10933d09',1,'mlx::steel::Conv2DInputBlockLoaderGeneral::bi'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_general.html#a4c91f848856ab0872bdfd37c62d4b0ba',1,'mlx::steel::Conv2DWeightBlockLoaderGeneral::bi']]],
|
|
['biases_41',['biases',['../struct_quantized_block_loader.html#a17d01a6aba0833b073586ef2c09d0fbd',1,'QuantizedBlockLoader']]],
|
|
['binary_42',['binary',['../namespacemlx_1_1core.html#ae374861abd45cf019c3e6be2026f3798',1,'mlx::core::binary()'],['../namespacemlx_1_1core_1_1metal.html#a269d591ec02e2f7c0f7a718fbfa37f73',1,'mlx::core::metal::binary()']]],
|
|
['binary_2eh_43',['binary.h',['../common_2binary_8h.html',1,'(Global Namespace)'],['../cpu_2binary_8h.html',1,'(Global Namespace)'],['../metal_2binary_8h.html',1,'(Global Namespace)'],['../metal_2kernels_2binary_8h.html',1,'(Global Namespace)']]],
|
|
['binary_5fg_44',['binary_g',['../metal_2kernels_2binary_8h.html#ab6b062acfb0497230e9476482d9dac20',1,'binary_g(device const T *a, device const T *b, device U *c, constant const int *shape, constant const int64_t *a_strides, constant const int64_t *b_strides, constant const int &ndim, uint3 index, uint3 grid_dim): binary.h'],['../metal_2kernels_2binary__two_8h.html#a84dbd99589c7a18ff4d91056dc6fe17a',1,'binary_g(device const T *a, device const T *b, device U *c, device U *d, constant const int *shape, constant const int64_t *a_strides, constant const int64_t *b_strides, constant const int &ndim, uint3 index, uint3 grid_dim): binary_two.h']]],
|
|
['binary_5fg_5fnd1_45',['binary_g_nd1',['../metal_2kernels_2binary_8h.html#aba4dd8bf59ed391789110e08ecfec6c2',1,'binary_g_nd1(device const T *a, device const T *b, device U *c, constant const int64_t &a_stride, constant const int64_t &b_stride, uint index): binary.h'],['../metal_2kernels_2binary__two_8h.html#a1f2e8e7dc1998bab3864184878a75776',1,'binary_g_nd1(device const T *a, device const T *b, device U *c, device U *d, constant const int64_t &a_stride, constant const int64_t &b_stride, uint index): binary_two.h']]],
|
|
['binary_5fg_5fnd2_46',['binary_g_nd2',['../metal_2kernels_2binary_8h.html#a73869f3771f16108f37120e485ce6e0b',1,'binary_g_nd2(device const T *a, device const T *b, device U *c, constant const int64_t a_strides[2], constant const int64_t b_strides[2], uint2 index, uint2 grid_dim): binary.h'],['../metal_2kernels_2binary__two_8h.html#a0d31d8d6c8a1845d3329ef2e23e3fff6',1,'binary_g_nd2(device const T *a, device const T *b, device U *c, device U *d, constant const int64_t a_strides[2], constant const int64_t b_strides[2], uint2 index, uint2 grid_dim): binary_two.h']]],
|
|
['binary_5fg_5fnd3_47',['binary_g_nd3',['../metal_2kernels_2binary_8h.html#ab6eb3c1f7349a52d5befff14e796320a',1,'binary_g_nd3(device const T *a, device const T *b, device U *c, constant const int64_t a_strides[3], constant const int64_t b_strides[3], uint3 index, uint3 grid_dim): binary.h'],['../metal_2kernels_2binary__two_8h.html#aae95e8f5fa772f780fc88054a2e35878',1,'binary_g_nd3(device const T *a, device const T *b, device U *c, device U *d, constant const int64_t a_strides[3], constant const int64_t b_strides[3], uint3 index, uint3 grid_dim): binary_two.h']]],
|
|
['binary_5fop_48',['binary_op',['../namespacemlx_1_1core.html#a9c1c1fdf9a0840a16a4d10a8f74f761d',1,'mlx::core::binary_op(const array &a, const array &b, array &out, Op op)'],['../namespacemlx_1_1core.html#a2aca3458c56605a74d07ec39876549bc',1,'mlx::core::binary_op(const array &a, const array &b, array &out, Op op)']]],
|
|
['binary_5fop_5fdims_49',['binary_op_dims',['../namespacemlx_1_1core.html#a7ca09ebf776fe32db580f9038587ec31',1,'mlx::core']]],
|
|
['binary_5fop_5fdispatch_5fdims_50',['binary_op_dispatch_dims',['../namespacemlx_1_1core.html#a66c9ee5018168b9101de52e0122d9755',1,'mlx::core']]],
|
|
['binary_5fop_5fgpu_51',['binary_op_gpu',['../namespacemlx_1_1core.html#ad884f4a36308b5b4f8a5d990d2e086df',1,'mlx::core::binary_op_gpu(const std::vector< array > &inputs, std::vector< array > &outputs, const std::string &op, const Stream &s)'],['../namespacemlx_1_1core.html#a094876ea5a2a2445ab64efc8222da202',1,'mlx::core::binary_op_gpu(const std::vector< array > &inputs, array &out, const std::string &op, const Stream &s)']]],
|
|
['binary_5fop_5fgpu_5finplace_52',['binary_op_gpu_inplace',['../namespacemlx_1_1core.html#a8616c0b7b0fc118a75400bc86404c367',1,'mlx::core::binary_op_gpu_inplace(const std::vector< array > &inputs, std::vector< array > &outputs, const std::string &op, const Stream &s)'],['../namespacemlx_1_1core.html#a7e6af6624e322e7ad60a3873a66e18a3',1,'mlx::core::binary_op_gpu_inplace(const std::vector< array > &inputs, array &out, const std::string &op, const Stream &s)']]],
|
|
['binary_5fops_53',['binary_ops',['../namespacemlx_1_1core_1_1metal.html#a8db7f9cc781d4bfb08423a401665f322',1,'mlx::core::metal']]],
|
|
['binary_5fops_2eh_54',['binary_ops.h',['../cpu_2binary__ops_8h.html',1,'(Global Namespace)'],['../metal_2kernels_2binary__ops_8h.html',1,'(Global Namespace)']]],
|
|
['binary_5fsingle_55',['BINARY_SINGLE',['../cpu_2binary__ops_8h.html#ad606fe7b9586413e760cfe461c9be119',1,'binary_ops.h']]],
|
|
['binary_5fss_56',['binary_ss',['../metal_2kernels_2binary_8h.html#a242b8b29a852c255467e50628c6dccf5',1,'binary_ss(device const T *a, device const T *b, device U *c, uint index): binary.h'],['../metal_2kernels_2binary__two_8h.html#af8a791ac7ca88d32cd8f4e9ac0f9ab4f',1,'binary_ss(device const T *a, device const T *b, device U *c, device U *d, uint index): binary_two.h']]],
|
|
['binary_5fsv_57',['binary_sv',['../metal_2kernels_2binary_8h.html#a4116c35f2e4632366d1611d5a95ba141',1,'binary_sv(device const T *a, device const T *b, device U *c, uint index): binary.h'],['../metal_2kernels_2binary__two_8h.html#ab18c6ecf5065275c93701efd095c916c',1,'binary_sv(device const T *a, device const T *b, device U *c, device U *d, uint index): binary_two.h']]],
|
|
['binary_5fsv2_58',['binary_sv2',['../metal_2kernels_2binary_8h.html#aa8c48b1b21d8f5a181f5443de2346589',1,'binary_sv2(device const T *a, device const T *b, device U *c, uint2 index, uint2 grid_dim): binary.h'],['../metal_2kernels_2binary__two_8h.html#a08822ff98ea6f61a98b49a9e9a38b891',1,'binary_sv2(device const T *a, device const T *b, device U *c, device U *d, uint2 index, uint2 grid_dim): binary_two.h']]],
|
|
['binary_5ftwo_59',['binary_two',['../namespacemlx_1_1core_1_1metal.html#aed047eec38b030ec5f29b9da54abf8cb',1,'mlx::core::metal']]],
|
|
['binary_5ftwo_2eh_60',['binary_two.h',['../cpu_2binary__two_8h.html',1,'(Global Namespace)'],['../metal_2kernels_2binary__two_8h.html',1,'(Global Namespace)']]],
|
|
['binary_5fvs_61',['binary_vs',['../metal_2kernels_2binary_8h.html#a649851d133358dd5832a73b1061b3313',1,'binary_vs(device const T *a, device const T *b, device U *c, uint index): binary.h'],['../metal_2kernels_2binary__two_8h.html#a12dbda74fa460812177ccb9aeee6e1ca',1,'binary_vs(device const T *a, device const T *b, device U *c, device U *d, uint index): binary_two.h']]],
|
|
['binary_5fvs2_62',['binary_vs2',['../metal_2kernels_2binary_8h.html#a48bd82eb10f9c623ce7d28daec4fa512',1,'binary_vs2(device const T *a, device const T *b, device U *c, uint2 index, uint2 grid_dim): binary.h'],['../metal_2kernels_2binary__two_8h.html#a273d2f31691f2c64623c2a97eab344be',1,'binary_vs2(device const T *a, device const T *b, device U *c, device U *d, uint2 index, uint2 grid_dim): binary_two.h']]],
|
|
['binary_5fvv_63',['binary_vv',['../metal_2kernels_2binary_8h.html#add6a9aeee3cb0ba909574f27fa9ecd5b',1,'binary_vv(device const T *a, device const T *b, device U *c, uint index): binary.h'],['../metal_2kernels_2binary__two_8h.html#ab4324f594c007a6895540b77ad5d89d9',1,'binary_vv(device const T *a, device const T *b, device U *c, device U *d, uint index): binary_two.h']]],
|
|
['binary_5fvv2_64',['binary_vv2',['../metal_2kernels_2binary_8h.html#a19dbbf8fea68b64bdd25dc8d36865171',1,'binary_vv2(device const T *a, device const T *b, device U *c, uint2 index, uint2 grid_dim): binary.h'],['../metal_2kernels_2binary__two_8h.html#a12e80730e43dfaa4c79ce8d5f99edc50',1,'binary_vv2(device const T *a, device const T *b, device U *c, device U *d, uint2 index, uint2 grid_dim): binary_two.h']]],
|
|
['binaryoptype_65',['BinaryOpType',['../namespacemlx_1_1core.html#a546e3d3c8957fbf2758f9504f4a2d0b6',1,'mlx::core']]],
|
|
['bits_66',['bits',['../namespacemlx_1_1core_1_1random.html#ad7d1c0b530906538dd8fb31b17382f2b',1,'mlx::core::random::bits(const Shape &shape, int width, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1random.html#a035d36774135faabad33d8f64a879df7',1,'mlx::core::random::bits(const Shape &shape, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})']]],
|
|
['bits_5f_67',['bits_',['../struct___m_l_x___b_float16.html#a4113263b63e3757ea8334cc4f0f5c3c8',1,'_MLX_BFloat16::bits_'],['../structmlx_1_1core_1_1___m_l_x___b_float16.html#aca48963f820065c3d8ecab24265ab3fc',1,'mlx::core::_MLX_BFloat16::bits_'],['../structmlx_1_1core_1_1___m_l_x___float16.html#a5203fe52424fd32bce6eb7917dd9288b',1,'mlx::core::_MLX_Float16::bits_']]],
|
|
['bits_5fto_5fbfloat_68',['bits_to_bfloat',['../struct___m_l_x___b_float16.html#a91ccb774773b65f8d4c1aea3f1c6e1ca',1,'_MLX_BFloat16']]],
|
|
['bits_5fto_5fbfloat_5fstruct_69',['bits_to_bfloat_struct',['../struct___m_l_x___b_float16_1_1bits__to__bfloat__struct.html',1,'_MLX_BFloat16']]],
|
|
['bitwise_5fand_70',['bitwise_and',['../group__ops.html#ga752fd2707dabb05d0308ba3d55346ada',1,'mlx::core']]],
|
|
['bitwise_5finvert_71',['bitwise_invert',['../group__ops.html#gaf1182ae7c049fbc9ee190f3e0fffbf83',1,'mlx::core']]],
|
|
['bitwise_5for_72',['bitwise_or',['../group__ops.html#ga8af4f22c08c11c4ffab7e3d45e0f3cd6',1,'mlx::core']]],
|
|
['bitwise_5fxor_73',['bitwise_xor',['../group__ops.html#ga3188638fba3a60e264baf69956a1e08b',1,'mlx::core']]],
|
|
['bitwiseand_74',['BitwiseAnd',['../struct_bitwise_and.html',1,'BitwiseAnd'],['../structmlx_1_1core_1_1detail_1_1_bitwise_and.html',1,'mlx::core::detail::BitwiseAnd']]],
|
|
['bitwisebinary_75',['BitwiseBinary',['../classmlx_1_1core_1_1_bitwise_binary.html',1,'mlx::core::BitwiseBinary'],['../classmlx_1_1core_1_1_bitwise_binary.html#a0d8b3a94951621ffcdebc6fda748a172',1,'mlx::core::BitwiseBinary::BitwiseBinary()']]],
|
|
['bitwiseinvert_76',['BitwiseInvert',['../struct_bitwise_invert.html',1,'BitwiseInvert'],['../classmlx_1_1core_1_1_bitwise_invert.html',1,'mlx::core::BitwiseInvert'],['../structmlx_1_1core_1_1detail_1_1_bitwise_invert.html',1,'mlx::core::detail::BitwiseInvert'],['../classmlx_1_1core_1_1_bitwise_invert.html#aaa0180570a82e93988b982b93cd91623',1,'mlx::core::BitwiseInvert::BitwiseInvert()']]],
|
|
['bitwiseor_77',['BitwiseOr',['../struct_bitwise_or.html',1,'BitwiseOr'],['../structmlx_1_1core_1_1detail_1_1_bitwise_or.html',1,'mlx::core::detail::BitwiseOr']]],
|
|
['bitwisexor_78',['BitwiseXor',['../struct_bitwise_xor.html',1,'BitwiseXor'],['../structmlx_1_1core_1_1detail_1_1_bitwise_xor.html',1,'mlx::core::detail::BitwiseXor']]],
|
|
['bj_79',['bj',['../struct_quantized_block_loader.html#ae2add92b2aaf3414e91f0470b9b0cc00',1,'QuantizedBlockLoader::bj'],['../structmlx_1_1steel_1_1_block_loader.html#a78c326e75ee35a484685771143047cd4',1,'mlx::steel::BlockLoader::bj'],['../structmlx_1_1steel_1_1_block_loader_t.html#aca83e49c31095badc8a46eb3c8e00957',1,'mlx::steel::BlockLoaderT::bj'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_large_filter.html#a7ae9e41f50c0c63c35b63086a1c22cc3',1,'mlx::steel::Conv2DInputBlockLoaderLargeFilter::bj'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_filter.html#a6fd3dd7b74d91609fa9dd61c657a0e32',1,'mlx::steel::Conv2DInputBlockLoaderSmallFilter::bj'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader.html#a6f2fdcaf5a67567cca38ae3d8120ab37',1,'mlx::steel::Conv2DWeightBlockLoader::bj'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_channels.html#a7cf448573d41fbc67f8dfc65b7aef2b2',1,'mlx::steel::Conv2DInputBlockLoaderSmallChannels::bj'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_small_channels.html#adaa261fc2e8e694aedab4ebd60b52e5e',1,'mlx::steel::Conv2DWeightBlockLoaderSmallChannels::bj'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_general.html#ace16704025bc6e6204c306a357f3a8b8',1,'mlx::steel::Conv2DInputBlockLoaderGeneral::bj'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_general.html#acec010e10d5733654963407af38d4f67',1,'mlx::steel::Conv2DWeightBlockLoaderGeneral::bj']]],
|
|
['block_5fmasked_5fgemm_80',['block_masked_gemm',['../steel__gemm__masked_8h.html#a3642df61cba8eff47a08a20834137eac',1,'block_masked_gemm(const device T *A, const device T *B, device T *D, const constant GEMMParams *params, const constant int *batch_shape, const constant int64_t *batch_strides, const device out_mask_t *out_mask, const device op_mask_t *lhs_mask, const device op_mask_t *rhs_mask, const constant int *mask_strides, uint simd_lane_id, uint simd_group_id, uint3 tid, uint3 lid): steel_gemm_masked.h'],['../steel__gemm__masked_8h.html#ac59086cee8cf396751da691fe80cda7b',1,'block_masked_gemm(const device T *A, const device T *B, device T *D, const constant GEMMParams *params, const constant int *batch_shape, const constant int64_t *batch_strides, const device bool *out_mask, const device bool *lhs_mask, const device bool *rhs_mask, const constant int *mask_strides, uint simd_lane_id, uint simd_group_id, uint3 tid, uint3 lid): steel_gemm_masked.h']]],
|
|
['block_5fmasked_5fmm_81',['block_masked_mm',['../group__ops.html#ga6b76c8ea46b19e6866af155fa5910be6',1,'mlx::core']]],
|
|
['block_5fmerge_5fsort_5ft_82',['block_merge_sort_t',['../struct_kernel_merge_sort.html#a37b94702630a4f97254d1da847d420c0',1,'KernelMergeSort::block_merge_sort_t'],['../struct_kernel_multi_block_merge_sort.html#aaf08efb1c540b276b0f1bbdeb514544d',1,'KernelMultiBlockMergeSort::block_merge_sort_t']]],
|
|
['block_5fsort_83',['block_sort',['../struct_kernel_merge_sort.html#a90102e73c3f6e9652e9870d57b129e71',1,'KernelMergeSort::block_sort()'],['../struct_kernel_multi_block_merge_sort.html#a56b3d8bf726e5619325dd07752666f8a',1,'KernelMultiBlockMergeSort::block_sort()'],['../sort_8h.html#a93f14092416169c4449141043ac45ffd',1,'block_sort(const device T *inp, device U *out, const constant int &size_sorted_axis, const constant int &in_stride_sorted_axis, const constant int &out_stride_sorted_axis, const constant int &in_stride_segment_axis, const constant int &out_stride_segment_axis, uint3 tid, uint3 lid): sort.h']]],
|
|
['block_5fsort_5fnc_84',['block_sort_nc',['../sort_8h.html#ad272ef4422c75565a79b7f5e751bb247',1,'sort.h']]],
|
|
['blockloader_85',['BlockLoader',['../structmlx_1_1steel_1_1_block_loader.html',1,'mlx::steel::BlockLoader< T, BROWS, BCOLS, dst_ld, reduction_dim, tgp_size, alignment, n_reads, TCOLS, TROWS >'],['../structmlx_1_1steel_1_1_block_loader.html#a37aca066e63dff238865b5923a2d4335',1,'mlx::steel::BlockLoader::BlockLoader(const device T *src_, const int src_ld_, threadgroup T *dst_, ushort simd_group_id, ushort simd_lane_id)'],['../structmlx_1_1steel_1_1_block_loader.html#a37aca066e63dff238865b5923a2d4335',1,'mlx::steel::BlockLoader::BlockLoader(const device T *src_, const int src_ld_, threadgroup T *dst_, ushort simd_group_id, ushort simd_lane_id)']]],
|
|
['blockloader_3c_20t_2c_20transpose_5fa_20_3f_20bk_20_3abm_2c_20transpose_5fa_20_3f_20bm_20_3abk_2c_20transpose_5fa_20_3f_20bm_2btgp_5fpadding_5fa_20_3abk_2btgp_5fpadding_5fa_2c_20_21transpose_5fa_2c_20tgp_5fsize_20_3e_86',['BlockLoader< T, transpose_a ? BK :BM, transpose_a ? BM :BK, transpose_a ? BM+tgp_padding_a :BK+tgp_padding_a, !transpose_a, tgp_size >',['../structmlx_1_1steel_1_1_block_loader.html',1,'mlx::steel']]],
|
|
['blockloader_3c_20t_2c_20transpose_5fb_20_3f_20bn_20_3abk_2c_20transpose_5fb_20_3f_20bk_20_3abn_2c_20transpose_5fb_20_3f_20bk_2btgp_5fpadding_5fb_20_3abn_2btgp_5fpadding_5fb_2c_20transpose_5fb_2c_20tgp_5fsize_20_3e_87',['BlockLoader< T, transpose_b ? BN :BK, transpose_b ? BK :BN, transpose_b ? BK+tgp_padding_b :BN+tgp_padding_b, transpose_b, tgp_size >',['../structmlx_1_1steel_1_1_block_loader.html',1,'mlx::steel']]],
|
|
['blockloadert_88',['BlockLoaderT',['../structmlx_1_1steel_1_1_block_loader_t.html',1,'mlx::steel::BlockLoaderT< T, BROWS, BCOLS, kDstStrRow, kDstStrCol, reduction_dim, tgp_size, n_reads, TCOLS, TROWS >'],['../structmlx_1_1steel_1_1_block_loader_t.html#a076616a7c67ad1b847e0e6b046077ee2',1,'mlx::steel::BlockLoaderT::BlockLoaderT()']]],
|
|
['blockm_89',['blockM',['../struct_g_e_m_v_kernel.html#a7281520100658811076400060663903c',1,'GEMVKernel::blockM'],['../struct_g_e_m_v_t_kernel.html#a2ae8ce535d59cccf453381b4485a77f0',1,'GEMVTKernel::blockM']]],
|
|
['blockmaskedmm_90',['BlockMaskedMM',['../classmlx_1_1core_1_1_block_masked_m_m.html',1,'mlx::core::BlockMaskedMM'],['../classmlx_1_1core_1_1_block_masked_m_m.html#ad26509deb5306d0c5eb72477e9a57477',1,'mlx::core::BlockMaskedMM::BlockMaskedMM()']]],
|
|
['blockmergesort_91',['BlockMergeSort',['../struct_block_merge_sort.html',1,'']]],
|
|
['blockmma_92',['BlockMMA',['../structmlx_1_1steel_1_1_block_m_m_a.html',1,'mlx::steel::BlockMMA< T, U, BM, BN, BK, WM, WN, transpose_a, transpose_b, lda_tgp, ldb_tgp, AccumType, Epilogue >'],['../structmlx_1_1steel_1_1_block_m_m_a.html#aa14406b7298456ac45d23dd3c4642dd8',1,'mlx::steel::BlockMMA::BlockMMA(ushort simd_group_id, ushort simd_lane_id)'],['../structmlx_1_1steel_1_1_block_m_m_a.html#aa14406b7298456ac45d23dd3c4642dd8',1,'mlx::steel::BlockMMA::BlockMMA(ushort simd_group_id, ushort simd_lane_id)']]],
|
|
['blockmma_3c_20t_2c_20u_2c_20bm_2c_20bn_2c_20bk_2c_20wm_2c_20wn_2c_20transpose_5fa_2c_20transpose_5fb_2c_20transpose_5fa_20_3f_20bm_2btgp_5fpadding_5fa_20_3abk_2btgp_5fpadding_5fa_2c_20transpose_5fb_20_3f_20bk_2btgp_5fpadding_5fb_20_3abn_2btgp_5fpadding_5fb_2c_20accumtype_2c_20epilogue_20_3e_93',['BlockMMA< T, U, BM, BN, BK, WM, WN, transpose_a, transpose_b, transpose_a ? BM+tgp_padding_a :BK+tgp_padding_a, transpose_b ? BK+tgp_padding_b :BN+tgp_padding_b, AccumType, Epilogue >',['../structmlx_1_1steel_1_1_block_m_m_a.html',1,'mlx::steel']]],
|
|
['blockn_94',['blockN',['../struct_g_e_m_v_kernel.html#a2fef17f9c9aa0bdf530ad3554fb0988b',1,'GEMVKernel::blockN'],['../struct_g_e_m_v_t_kernel.html#a60be87666006ba0bf88bc8e6902da42a',1,'GEMVTKernel::blockN']]],
|
|
['blockswizzle_95',['BlockSwizzle',['../structmlx_1_1steel_1_1_block_swizzle.html',1,'mlx::steel']]],
|
|
['bluestein_5ffft_96',['bluestein_fft',['../backend_2metal_2kernels_2fft_8h.html#a0abc609e9756475800e996775a96a87e',1,'fft.h']]],
|
|
['bool4_5for_5fuint_97',['bool4_or_uint',['../unionbool4__or__uint.html',1,'']]],
|
|
['bool_5f_98',['bool_',['../structmlx_1_1core_1_1_dtype.html#ade845ef5dcebead13a37fe696436e1daa467afb5838aa377d55cce81f84c5512b',1,'mlx::core::Dtype::bool_'],['../namespacemlx_1_1core.html#a113d2bac7e4aa6a4cb4a5c3242527b82',1,'mlx::core::bool_']]],
|
|
['bool_5fconstant_99',['bool_constant',['../namespacemlx_1_1steel.html#adbb34bcf0d2dca6b9fb803d591d00da9',1,'mlx::steel']]],
|
|
['broadcast_100',['Broadcast',['../classmlx_1_1core_1_1_broadcast.html',1,'mlx::core::Broadcast'],['../classmlx_1_1core_1_1_broadcast.html#a004cce3029c0427569830016f99648cb',1,'mlx::core::Broadcast::Broadcast()']]],
|
|
['broadcast_5farrays_101',['broadcast_arrays',['../group__ops.html#gab783890428b596f715dc7dd2057eae99',1,'mlx::core']]],
|
|
['broadcast_5fshapes_102',['broadcast_shapes',['../namespacemlx_1_1core.html#aefa6a2ec6439e9619cafd227a1dc14ab',1,'mlx::core']]],
|
|
['broadcast_5fto_103',['broadcast_to',['../group__ops.html#ga2fd5891f11593b7f09550e884f969013',1,'mlx::core']]],
|
|
['broadcastaxes_104',['BroadcastAxes',['../classmlx_1_1core_1_1_broadcast_axes.html',1,'mlx::core::BroadcastAxes'],['../classmlx_1_1core_1_1_broadcast_axes.html#a5136f33489670cdc0802e46725288195',1,'mlx::core::BroadcastAxes::BroadcastAxes()']]],
|
|
['brows_105',['BROWS',['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_large_filter.html#ac070c6bd5be85b1ae805e18890db4fd4',1,'mlx::steel::Conv2DInputBlockLoaderLargeFilter::BROWS'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_filter.html#a10591ea957605a9c662f93d59ff3410d',1,'mlx::steel::Conv2DInputBlockLoaderSmallFilter::BROWS'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader.html#ae9b86b05b23153ea1abaeead456c491c',1,'mlx::steel::Conv2DWeightBlockLoader::BROWS'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_channels.html#a343984fb74ec579a4404278dbbc7e7b5',1,'mlx::steel::Conv2DInputBlockLoaderSmallChannels::BROWS'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_small_channels.html#acc8140aae84694f62e6324dbb6a614a4',1,'mlx::steel::Conv2DWeightBlockLoaderSmallChannels::BROWS'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_general.html#aba1e1c8012e4e50f0e9bcfb9486c1781',1,'mlx::steel::Conv2DInputBlockLoaderGeneral::BROWS'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_general.html#a015a0c56de74a0c4d51953a7e94fbba8',1,'mlx::steel::Conv2DWeightBlockLoaderGeneral::BROWS']]],
|
|
['bs_5foffset_106',['Bs_offset',['../structmlx_1_1steel_1_1_block_m_m_a.html#a92f6aeee432f53638447eac842f43eca',1,'mlx::steel::BlockMMA']]],
|
|
['bs_5fqmm_5fn_107',['bs_qmm_n',['../quantized_8h.html#a323d2d70799c8d9dffe4b64a4285a799',1,'quantized.h']]],
|
|
['bs_5fqmm_5ft_108',['bs_qmm_t',['../quantized_8h.html#a693057a0c311a73ac0759e005b9806c1',1,'quantized.h']]],
|
|
['bs_5fqmv_109',['bs_qmv',['../quantized_8h.html#aaf4fb9c4318c5cd27d118004dbdeba61',1,'quantized.h']]],
|
|
['bs_5fqmv_5ffast_110',['bs_qmv_fast',['../quantized_8h.html#a359282a9f71e487e5d86d246896ab33d',1,'quantized.h']]],
|
|
['bs_5fqvm_111',['bs_qvm',['../quantized_8h.html#a2c53419ba5019d4722c0f4c2026b1142',1,'quantized.h']]],
|
|
['btile_112',['Btile',['../structmlx_1_1steel_1_1_block_m_m_a.html#a44fca27c821764317263047a780977b0',1,'mlx::steel::BlockMMA']]],
|
|
['buf_113',['buf',['../struct_read_writer.html#a23bac3c96dd0265ddbee1f256be45ff5',1,'ReadWriter']]],
|
|
['buffer_114',['Buffer',['../classmlx_1_1core_1_1allocator_1_1_buffer.html',1,'mlx::core::allocator::Buffer'],['../classmlx_1_1core_1_1metal_1_1_buffer.html',1,'mlx::core::metal::Buffer'],['../classmlx_1_1core_1_1allocator_1_1_buffer.html#ac4fc2cc6aa1368cfb74aff329d9a1300',1,'mlx::core::allocator::Buffer::Buffer()'],['../classmlx_1_1core_1_1metal_1_1_buffer.html#ac4fc2cc6aa1368cfb74aff329d9a1300',1,'mlx::core::metal::Buffer::Buffer()']]],
|
|
['buffer_115',['buffer',['../structmlx_1_1core_1_1array_1_1_data.html#a9a51e2d12ba505027cc0fca86bdd39ad',1,'mlx::core::array::Data::buffer'],['../structmlx_1_1core_1_1metal_1_1_device_stream.html#a99183c92599edfeb75f7fa0f37e1d9eb',1,'mlx::core::metal::DeviceStream::buffer'],['../classmlx_1_1core_1_1array.html#ab3daf04c27c4593d9d73c397b8484a08',1,'mlx::core::array::buffer()'],['../classmlx_1_1core_1_1array.html#a634466ce661485394f2fdc3bd6796bcd',1,'mlx::core::array::buffer() const']]],
|
|
['buffer_5fops_116',['buffer_ops',['../structmlx_1_1core_1_1metal_1_1_device_stream.html#ab6048b329e65a59033834f3bdd351782',1,'mlx::core::metal::DeviceStream']]],
|
|
['buffer_5fsize_117',['buffer_size',['../classmlx_1_1core_1_1array.html#a914577c63755b2e862d2da68bbf8e3dd',1,'mlx::core::array']]],
|
|
['buffer_5fsizes_118',['buffer_sizes',['../structmlx_1_1core_1_1metal_1_1_device_stream.html#ae7054233303b06329c67177382ded459',1,'mlx::core::metal::DeviceStream']]],
|
|
['buffers_119',['buffers',['../struct_indices.html#ad705070a740579c07d109ae4f3d86e76',1,'Indices']]],
|
|
['build_5fcommand_120',['build_command',['../classmlx_1_1core_1_1_jit_compiler.html#a10a5cde91ab929ccbdbdf4c4d940f156',1,'mlx::core::JitCompiler']]],
|
|
['build_5flib_5fname_121',['build_lib_name',['../namespacemlx_1_1core.html#a3ef23f334cb9f68a2c50524bc67c913b',1,'mlx::core']]],
|
|
['bytes_5fper_5fpack_122',['bytes_per_pack',['../struct_quantized_block_loader.html#ad00fe6d8bd395206a41693a8ed65d4db',1,'QuantizedBlockLoader']]]
|
|
];
|