zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-11-05 11:28:12 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

rebase

Browse Source
This commit is contained in:
CircleCI Docs
2025-03-20 22:37:22 +00:00
parent a943912d4c
commit cecec56a99
858 changed files with 18494 additions and 17475 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
docs/build/html/search/all_2.js vendored
View File

@@ -42,15 +42,15 @@ var searchData=
['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_42',['binary',['../namespacemlx_1_1core_1_1metal.html#a269d591ec02e2f7c0f7a718fbfa37f73',1,'mlx::core::metal']]],
['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_48',['binary_op',['../namespacemlx_1_1core.html#ae0c47c0bd95bb8c44339159e04c0f604',1,'mlx::core::binary_op(const array &a, const array &b, array &out, BinaryOpType bopt)'],['../namespacemlx_1_1core.html#a5160ef5819f58cf040c9613ecce548f1',1,'mlx::core::binary_op(const array &a, const array &b, array &out, BinaryOpType bopt)']]],
['binary_5fop_5fdims_49',['binary_op_dims',['../namespacemlx_1_1core.html#aeac6fa9529eedba76b27de9d098de963',1,'mlx::core']]],
['binary_5fop_5fdispatch_5fdims_50',['binary_op_dispatch_dims',['../namespacemlx_1_1core.html#ad1229ffbba21bdb81f63482a4651bc5a',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']]],
@@ -89,12 +89,12 @@ var searchData=
['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']]],
['blockm_89',['blockM',['../struct_g_e_m_v_kernel.html#a188ef7ed5c1d74d9b540b6d4ebf12f2e',1,'GEMVKernel::blockM'],['../struct_g_e_m_v_t_kernel.html#aa3f9e771c59770a72b73fb673eb7bf71',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']]],
['blockn_94',['blockN',['../struct_g_e_m_v_kernel.html#af22b6e5ed1a9d350866aaafa35d63a8a',1,'GEMVKernel::blockN'],['../struct_g_e_m_v_t_kernel.html#a2091b9804b702cbb995899312e3da417',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,'']]],