zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-09-18 01:50:16 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

rebase

Browse Source
This commit is contained in:
CircleCI Docs
2024-12-06 21:22:39 +00:00
parent 8ba8a39e43
commit 04b749a588
896 changed files with 12293 additions and 11212 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
docs/build/html/functions_s.html vendored
View File

@@ -89,6 +89,7 @@ $(function(){ initResizable(false); });
<h3><a id="index_s" name="index_s"></a>- s -</h3><ul>
<li>sanity_check()&#160;:&#160;<a class="el" href="structpocketfft_1_1detail_1_1util.html#a3d2f5f00d4fed5c09bd1509ffa9a44cd">pocketfft::detail::util</a></li>
<li>scale&#160;:&#160;<a class="el" href="structmlx_1_1steel_1_1_attn_params.html#ad81bcd32e6ff8fec0000eca505fb6826">mlx::steel::AttnParams</a>, <a class="el" href="struct_scale_op.html#a02043fac21c68fb8d6863a01f45ede4b">ScaleOp&lt; OutT, InT &gt;</a>, <a class="el" href="struct_transform_scale.html#aa56b8e107acf16fdf77006625c2b8bc6">TransformScale&lt; T &gt;</a></li>
<li>scale_&#160;:&#160;<a class="el" href="classmlx_1_1core_1_1fast_1_1_ro_p_e.html#aa3b6ec6bb0766fb2f4f805d49d540f40">mlx::core::fast::RoPE</a>, <a class="el" href="classmlx_1_1core_1_1fast_1_1_scaled_dot_product_attention.html#a3d4ef9a11114c8f69e35e2f1d78e5f7a">mlx::core::fast::ScaledDotProductAttention</a></li>
<li>ScaledDotProductAttention()&#160;:&#160;<a class="el" href="classmlx_1_1core_1_1fast_1_1_scaled_dot_product_attention.html#a6a904c377fffc3c193102a3123f5e706">mlx::core::fast::ScaledDotProductAttention</a></li>
<li>scales&#160;:&#160;<a class="el" href="struct_quantized_block_loader.html#a6123e4a9209d6eacb58b2c2344ed1ecf">QuantizedBlockLoader&lt; T, BROWS, BCOLS, dst_ld, reduction_dim, tgp_size, group_size, bits &gt;</a></li>
<li>Scan()&#160;:&#160;<a class="el" href="classmlx_1_1core_1_1_scan.html#ac93e8f9c6771de825d2186ef34fa7087">mlx::core::Scan</a></li>
@@ -103,7 +104,7 @@ $(function(){ initResizable(false); });
<li>set_bytes()&#160;:&#160;<a class="el" href="structmlx_1_1core_1_1metal_1_1_command_encoder.html#a9c343f791812a45c6c03a5c9f27f74d5">mlx::core::metal::CommandEncoder</a></li>
<li>set_cache_limit()&#160;:&#160;<a class="el" href="classmlx_1_1core_1_1metal_1_1_metal_allocator.html#af392bced29d9e4e3f1a7cc4725d83764">mlx::core::metal::MetalAllocator</a></li>
<li>set_compute_pipeline_state()&#160;:&#160;<a class="el" href="structmlx_1_1core_1_1metal_1_1_command_encoder.html#a6d4c03a6585deedb5ccd1a1057d0c6ef">mlx::core::metal::CommandEncoder</a></li>
<li>set_data()&#160;:&#160;<a class="el" href="classmlx_1_1core_1_1array.html#a631acd8e318189640b8338f9ae1a554d">mlx::core::array</a></li>
<li>set_data()&#160;:&#160;<a class="el" href="classmlx_1_1core_1_1array.html#af9e3a02b4c0023c36248dc75c887214f">mlx::core::array</a></li>
<li>set_default_stream()&#160;:&#160;<a class="el" href="classmlx_1_1core_1_1scheduler_1_1_scheduler.html#a6d15314ac9cf25efc9bd1278de9a66bb">mlx::core::scheduler::Scheduler</a></li>
<li>set_input_array()&#160;:&#160;<a class="el" href="structmlx_1_1core_1_1metal_1_1_command_encoder.html#ab69ff0d7f14b9b59db4df0608193dce4">mlx::core::metal::CommandEncoder</a></li>
<li>set_memory_limit()&#160;:&#160;<a class="el" href="classmlx_1_1core_1_1metal_1_1_metal_allocator.html#a179e3127ef9377ce54295f771c34ba1b">mlx::core::metal::MetalAllocator</a></li>
@@ -115,7 +116,7 @@ $(function(){ initResizable(false); });
<li>set_value()&#160;:&#160;<a class="el" href="classmlx_1_1core_1_1_event.html#a0d077b11f4b28f882b42440b7ac6d40d">mlx::core::Event</a></li>
<li>set_vector_bytes()&#160;:&#160;<a class="el" href="structmlx_1_1core_1_1metal_1_1_command_encoder.html#a68c3c6a036e11ec40211c09811bbed1b">mlx::core::metal::CommandEncoder</a></li>
<li>set_wired_limit()&#160;:&#160;<a class="el" href="classmlx_1_1core_1_1metal_1_1_metal_allocator.html#a84fa0347da18055bc13ba0a5c4b57253">mlx::core::metal::MetalAllocator</a></li>
<li>shape()&#160;:&#160;<a class="el" href="classmlx_1_1core_1_1array.html#a4a2a2c8a4a5beafd723fc13f2055d55d">mlx::core::array</a>, <a class="el" href="structmlx_1_1core_1_1fast_1_1_custom_kernel_shape_info.html#a63db720fe0c2abc4b71e22a58a015f8a">mlx::core::fast::CustomKernelShapeInfo</a>, <a class="el" href="structmlx_1_1core_1_1_reduction_plan.html#a6cfa8771fa9caf6fdcc3d74c9fca83ae">mlx::core::ReductionPlan</a>, <a class="el" href="structmlx_1_1steel_1_1_layout2_d.html#a23183747ab1ddbdd3f1fcac6d0faa2cd">mlx::steel::Layout2D&lt; Shape, Layout &gt;</a>, <a class="el" href="classpocketfft_1_1detail_1_1arr__info.html#accada8146cb8d3ab7facb4c1e3413ec0">pocketfft::detail::arr_info</a></li>
<li>shape()&#160;:&#160;<a class="el" href="classmlx_1_1core_1_1array.html#a1d06c76b0f3010a5c329d0e9e29e0597">mlx::core::array</a>, <a class="el" href="structmlx_1_1core_1_1fast_1_1_custom_kernel_shape_info.html#a63db720fe0c2abc4b71e22a58a015f8a">mlx::core::fast::CustomKernelShapeInfo</a>, <a class="el" href="structmlx_1_1core_1_1_reduction_plan.html#a6cfa8771fa9caf6fdcc3d74c9fca83ae">mlx::core::ReductionPlan</a>, <a class="el" href="structmlx_1_1steel_1_1_layout2_d.html#a23183747ab1ddbdd3f1fcac6d0faa2cd">mlx::steel::Layout2D&lt; Shape, Layout &gt;</a>, <a class="el" href="classpocketfft_1_1detail_1_1arr__info.html#accada8146cb8d3ab7facb4c1e3413ec0">pocketfft::detail::arr_info</a></li>
<li>Shape2D()&#160;:&#160;<a class="el" href="structmlx_1_1steel_1_1_shape2_d.html#a070ce70eb6d84361c7f313159c438a5c">mlx::steel::Shape2D&lt; RInt, CInt &gt;</a></li>
<li>shapes&#160;:&#160;<a class="el" href="struct_indices.html#a5ab170f1a77636180889ddfffd4f7d2f">Indices&lt; IdxT, NIDX &gt;</a></li>
<li>shp&#160;:&#160;<a class="el" href="classpocketfft_1_1detail_1_1arr__info.html#a2467e9e01de1ba4d7cd28c1af783da8d">pocketfft::detail::arr_info</a></li>
@@ -170,7 +171,7 @@ $(function(){ initResizable(false); });
<li>stride_out()&#160;:&#160;<a class="el" href="classpocketfft_1_1detail_1_1multi__iter.html#a81d71a13bf0b85e556fbb9834167ecc7">pocketfft::detail::multi_iter&lt; N &gt;</a></li>
<li>strided_device_idx&#160;:&#160;<a class="el" href="struct_read_writer.html#a4c0b12484aac4fd6759d67c190391989">ReadWriter&lt; in_T, out_T, step, four_step_real &gt;</a></li>
<li>strided_shared_idx&#160;:&#160;<a class="el" href="struct_read_writer.html#ace40adb02cfb33d89c98353327c251fc">ReadWriter&lt; in_T, out_T, step, four_step_real &gt;</a></li>
<li>strides&#160;:&#160;<a class="el" href="struct_indices.html#a7f73d7652f0f751e6a06c2663e329a4a">Indices&lt; IdxT, NIDX &gt;</a>, <a class="el" href="classmlx_1_1core_1_1array.html#a186cf2648da92584d5c1c8b24e69629b">mlx::core::array</a>, <a class="el" href="structmlx_1_1core_1_1fast_1_1_custom_kernel_shape_info.html#a63954de7da62942ec69afcaaa19d46f2">mlx::core::fast::CustomKernelShapeInfo</a>, <a class="el" href="structmlx_1_1core_1_1_reduction_plan.html#a9bf7cae845ab633247c1811613ece8bd">mlx::core::ReductionPlan</a></li>
<li>strides&#160;:&#160;<a class="el" href="struct_indices.html#a7f73d7652f0f751e6a06c2663e329a4a">Indices&lt; IdxT, NIDX &gt;</a>, <a class="el" href="classmlx_1_1core_1_1array.html#a28cf1928f5ec2f972a94ff1c0e71187d">mlx::core::array</a>, <a class="el" href="structmlx_1_1core_1_1fast_1_1_custom_kernel_shape_info.html#a63954de7da62942ec69afcaaa19d46f2">mlx::core::fast::CustomKernelShapeInfo</a>, <a class="el" href="structmlx_1_1core_1_1_reduction_plan.html#a9bf7cae845ab633247c1811613ece8bd">mlx::core::ReductionPlan</a></li>
<li>submit()&#160;:&#160;<a class="el" href="classpocketfft_1_1detail_1_1threading_1_1thread__pool.html#a8698d49e8f406cdb88006aac6a91f9a4">pocketfft::detail::threading::thread_pool</a></li>
<li>Subtract()&#160;:&#160;<a class="el" href="classmlx_1_1core_1_1_subtract.html#a834854757394f8de7082af65bf86ed9c">mlx::core::Subtract</a></li>
<li>Sum&#160;:&#160;<a class="el" href="classmlx_1_1core_1_1distributed_1_1_all_reduce.html#abb4560980e5d01aed14175ce8f6fc924a1fc7c1f09c80650ab0497e2d6781d65f">mlx::core::distributed::AllReduce</a>, <a class="el" href="classmlx_1_1core_1_1_reduce.html#a0848518b16ae6d4043d6be247bdf31c9a8582875544f1d3d396a1a376473ef1dd">mlx::core::Reduce</a>, <a class="el" href="classmlx_1_1core_1_1_scan.html#a47bf2ec54ead4b8f00f9f188518630f1ade23893033e4849f5596e7ce76a5fc36">mlx::core::Scan</a>, <a class="el" href="classmlx_1_1core_1_1_scatter.html#a614d19af11dc30644b2b4941033b613ca14abe2d8818efa71726be4e156813d6f">mlx::core::Scatter</a></li>