mirror of
https://github.com/ml-explore/mlx.git
synced 2025-09-18 10:26:56 +08:00
docs update
This commit is contained in:
Awni Hannun
committed by
CircleCI Docs
CircleCI Docs
parent
ba4eff9520
commit
0e688cbd0f
94
docs/build/html/jit_2indexing_8h_source.html
vendored
94
docs/build/html/jit_2indexing_8h_source.html
vendored
@@ -123,47 +123,59 @@ $(function() { codefold.init(0); });
|
||||
<div class="line"><a id="l00038" name="l00038"></a><span class="lineno"> 38</span><span class="stringliteral"> device mlx_atomic<{1}>* out [[buffer(2)]],</span></div>
|
||||
<div class="line"><a id="l00039" name="l00039"></a><span class="lineno"> 39</span><span class="stringliteral"> const constant int* out_shape [[buffer(3)]],</span></div>
|
||||
<div class="line"><a id="l00040" name="l00040"></a><span class="lineno"> 40</span><span class="stringliteral"> const constant size_t* out_strides [[buffer(4)]],</span></div>
|
||||
<div class="line"><a id="l00041" name="l00041"></a><span class="lineno"> 41</span><span class="stringliteral"> const constant size_t& upd_size [[buffer(5)]],</span></div>
|
||||
<div class="line"><a id="l00042" name="l00042"></a><span class="lineno"> 42</span><span class="stringliteral"> {5}</span></div>
|
||||
<div class="line"><a id="l00043" name="l00043"></a><span class="lineno"> 43</span><span class="stringliteral"> uint2 gid [[thread_position_in_grid]]) {{</span></div>
|
||||
<div class="line"><a id="l00044" name="l00044"></a><span class="lineno"> 44</span><span class="stringliteral"> const array<const device {2}*, {4}> idx_buffers = {{ {6} }};</span></div>
|
||||
<div class="line"><a id="l00045" name="l00045"></a><span class="lineno"> 45</span><span class="stringliteral"> return scatter_1d_index_impl<{1}, {2}, {3}, {4}>(</span></div>
|
||||
<div class="line"><a id="l00046" name="l00046"></a><span class="lineno"> 46</span><span class="stringliteral"> updates, out, out_shape, out_strides, upd_size, idx_buffers, gid);</span></div>
|
||||
<div class="line"><a id="l00047" name="l00047"></a><span class="lineno"> 47</span><span class="stringliteral">}}</span></div>
|
||||
<div class="line"><a id="l00048" name="l00048"></a><span class="lineno"> 48</span><span class="stringliteral"></span> </div>
|
||||
<div class="line"><a id="l00049" name="l00049"></a><span class="lineno"> 49</span><span class="stringliteral">[[kernel]] void scatter{0}_{4}(</span></div>
|
||||
<div class="line"><a id="l00050" name="l00050"></a><span class="lineno"> 50</span><span class="stringliteral"> const device {1}* updates [[buffer(1)]],</span></div>
|
||||
<div class="line"><a id="l00051" name="l00051"></a><span class="lineno"> 51</span><span class="stringliteral"> device mlx_atomic<{1}>* out [[buffer(2)]],</span></div>
|
||||
<div class="line"><a id="l00052" name="l00052"></a><span class="lineno"> 52</span><span class="stringliteral"> const constant int* upd_shape [[buffer(3)]],</span></div>
|
||||
<div class="line"><a id="l00053" name="l00053"></a><span class="lineno"> 53</span><span class="stringliteral"> const constant size_t* upd_strides [[buffer(4)]],</span></div>
|
||||
<div class="line"><a id="l00054" name="l00054"></a><span class="lineno"> 54</span><span class="stringliteral"> const constant size_t& upd_ndim [[buffer(5)]],</span></div>
|
||||
<div class="line"><a id="l00055" name="l00055"></a><span class="lineno"> 55</span><span class="stringliteral"> const constant size_t& upd_size [[buffer(6)]],</span></div>
|
||||
<div class="line"><a id="l00056" name="l00056"></a><span class="lineno"> 56</span><span class="stringliteral"> const constant int* out_shape [[buffer(7)]],</span></div>
|
||||
<div class="line"><a id="l00057" name="l00057"></a><span class="lineno"> 57</span><span class="stringliteral"> const constant size_t* out_strides [[buffer(8)]],</span></div>
|
||||
<div class="line"><a id="l00058" name="l00058"></a><span class="lineno"> 58</span><span class="stringliteral"> const constant size_t& out_ndim [[buffer(9)]],</span></div>
|
||||
<div class="line"><a id="l00059" name="l00059"></a><span class="lineno"> 59</span><span class="stringliteral"> const constant int* axes [[buffer(10)]],</span></div>
|
||||
<div class="line"><a id="l00060" name="l00060"></a><span class="lineno"> 60</span><span class="stringliteral"> const constant int* idx_shapes [[buffer(11)]],</span></div>
|
||||
<div class="line"><a id="l00061" name="l00061"></a><span class="lineno"> 61</span><span class="stringliteral"> const constant size_t* idx_strides [[buffer(12)]],</span></div>
|
||||
<div class="line"><a id="l00062" name="l00062"></a><span class="lineno"> 62</span><span class="stringliteral"> const constant int& idx_ndim [[buffer(13)]],</span></div>
|
||||
<div class="line"><a id="l00063" name="l00063"></a><span class="lineno"> 63</span><span class="stringliteral"> {5}</span></div>
|
||||
<div class="line"><a id="l00064" name="l00064"></a><span class="lineno"> 64</span><span class="stringliteral"> uint2 gid [[thread_position_in_grid]]) {{</span></div>
|
||||
<div class="line"><a id="l00065" name="l00065"></a><span class="lineno"> 65</span><span class="stringliteral"> Indices<{2}, {4}> idxs{{ {{ {6} }}, idx_shapes, idx_strides, idx_ndim}};</span></div>
|
||||
<div class="line"><a id="l00066" name="l00066"></a><span class="lineno"> 66</span><span class="stringliteral"></span> </div>
|
||||
<div class="line"><a id="l00067" name="l00067"></a><span class="lineno"> 67</span><span class="stringliteral"> return scatter_impl<{1}, {2}, {3}, {4}>(</span></div>
|
||||
<div class="line"><a id="l00068" name="l00068"></a><span class="lineno"> 68</span><span class="stringliteral"> updates,</span></div>
|
||||
<div class="line"><a id="l00069" name="l00069"></a><span class="lineno"> 69</span><span class="stringliteral"> out,</span></div>
|
||||
<div class="line"><a id="l00070" name="l00070"></a><span class="lineno"> 70</span><span class="stringliteral"> upd_shape,</span></div>
|
||||
<div class="line"><a id="l00071" name="l00071"></a><span class="lineno"> 71</span><span class="stringliteral"> upd_strides,</span></div>
|
||||
<div class="line"><a id="l00072" name="l00072"></a><span class="lineno"> 72</span><span class="stringliteral"> upd_ndim,</span></div>
|
||||
<div class="line"><a id="l00073" name="l00073"></a><span class="lineno"> 73</span><span class="stringliteral"> upd_size,</span></div>
|
||||
<div class="line"><a id="l00074" name="l00074"></a><span class="lineno"> 74</span><span class="stringliteral"> out_shape,</span></div>
|
||||
<div class="line"><a id="l00075" name="l00075"></a><span class="lineno"> 75</span><span class="stringliteral"> out_strides,</span></div>
|
||||
<div class="line"><a id="l00076" name="l00076"></a><span class="lineno"> 76</span><span class="stringliteral"> out_ndim,</span></div>
|
||||
<div class="line"><a id="l00077" name="l00077"></a><span class="lineno"> 77</span><span class="stringliteral"> axes,</span></div>
|
||||
<div class="line"><a id="l00078" name="l00078"></a><span class="lineno"> 78</span><span class="stringliteral"> idxs,</span></div>
|
||||
<div class="line"><a id="l00079" name="l00079"></a><span class="lineno"> 79</span><span class="stringliteral"> gid);</span></div>
|
||||
<div class="line"><a id="l00080" name="l00080"></a><span class="lineno"> 80</span><span class="stringliteral">}}</span></div>
|
||||
<div class="line"><a id="l00081" name="l00081"></a><span class="lineno"> 81</span><span class="stringliteral">)";</span></div>
|
||||
<div class="line"><a id="l00041" name="l00041"></a><span class="lineno"> 41</span><span class="stringliteral"> const constant size_t& out_ndim [[buffer(5)]],</span></div>
|
||||
<div class="line"><a id="l00042" name="l00042"></a><span class="lineno"> 42</span><span class="stringliteral"> const constant int* upd_shape [[buffer(6)]],</span></div>
|
||||
<div class="line"><a id="l00043" name="l00043"></a><span class="lineno"> 43</span><span class="stringliteral"> const constant size_t& upd_ndim [[buffer(7)]],</span></div>
|
||||
<div class="line"><a id="l00044" name="l00044"></a><span class="lineno"> 44</span><span class="stringliteral"> const constant size_t& upd_size [[buffer(8)]],</span></div>
|
||||
<div class="line"><a id="l00045" name="l00045"></a><span class="lineno"> 45</span><span class="stringliteral"> {5}</span></div>
|
||||
<div class="line"><a id="l00046" name="l00046"></a><span class="lineno"> 46</span><span class="stringliteral"> uint2 gid [[thread_position_in_grid]]) {{</span></div>
|
||||
<div class="line"><a id="l00047" name="l00047"></a><span class="lineno"> 47</span><span class="stringliteral"> const array<const device {2}*, {4}> idx_buffers = {{ {6} }};</span></div>
|
||||
<div class="line"><a id="l00048" name="l00048"></a><span class="lineno"> 48</span><span class="stringliteral"> return scatter_1d_index_impl<{1}, {2}, {3}, {4}>(</span></div>
|
||||
<div class="line"><a id="l00049" name="l00049"></a><span class="lineno"> 49</span><span class="stringliteral"> updates,</span></div>
|
||||
<div class="line"><a id="l00050" name="l00050"></a><span class="lineno"> 50</span><span class="stringliteral"> out,</span></div>
|
||||
<div class="line"><a id="l00051" name="l00051"></a><span class="lineno"> 51</span><span class="stringliteral"> out_shape,</span></div>
|
||||
<div class="line"><a id="l00052" name="l00052"></a><span class="lineno"> 52</span><span class="stringliteral"> out_strides,</span></div>
|
||||
<div class="line"><a id="l00053" name="l00053"></a><span class="lineno"> 53</span><span class="stringliteral"> out_ndim,</span></div>
|
||||
<div class="line"><a id="l00054" name="l00054"></a><span class="lineno"> 54</span><span class="stringliteral"> upd_shape,</span></div>
|
||||
<div class="line"><a id="l00055" name="l00055"></a><span class="lineno"> 55</span><span class="stringliteral"> upd_ndim,</span></div>
|
||||
<div class="line"><a id="l00056" name="l00056"></a><span class="lineno"> 56</span><span class="stringliteral"> upd_size,</span></div>
|
||||
<div class="line"><a id="l00057" name="l00057"></a><span class="lineno"> 57</span><span class="stringliteral"> idx_buffers,</span></div>
|
||||
<div class="line"><a id="l00058" name="l00058"></a><span class="lineno"> 58</span><span class="stringliteral"> gid);</span></div>
|
||||
<div class="line"><a id="l00059" name="l00059"></a><span class="lineno"> 59</span><span class="stringliteral">}}</span></div>
|
||||
<div class="line"><a id="l00060" name="l00060"></a><span class="lineno"> 60</span><span class="stringliteral"></span> </div>
|
||||
<div class="line"><a id="l00061" name="l00061"></a><span class="lineno"> 61</span><span class="stringliteral">[[kernel]] void scatter{0}_{4}(</span></div>
|
||||
<div class="line"><a id="l00062" name="l00062"></a><span class="lineno"> 62</span><span class="stringliteral"> const device {1}* updates [[buffer(1)]],</span></div>
|
||||
<div class="line"><a id="l00063" name="l00063"></a><span class="lineno"> 63</span><span class="stringliteral"> device mlx_atomic<{1}>* out [[buffer(2)]],</span></div>
|
||||
<div class="line"><a id="l00064" name="l00064"></a><span class="lineno"> 64</span><span class="stringliteral"> const constant int* upd_shape [[buffer(3)]],</span></div>
|
||||
<div class="line"><a id="l00065" name="l00065"></a><span class="lineno"> 65</span><span class="stringliteral"> const constant size_t* upd_strides [[buffer(4)]],</span></div>
|
||||
<div class="line"><a id="l00066" name="l00066"></a><span class="lineno"> 66</span><span class="stringliteral"> const constant size_t& upd_ndim [[buffer(5)]],</span></div>
|
||||
<div class="line"><a id="l00067" name="l00067"></a><span class="lineno"> 67</span><span class="stringliteral"> const constant size_t& upd_size [[buffer(6)]],</span></div>
|
||||
<div class="line"><a id="l00068" name="l00068"></a><span class="lineno"> 68</span><span class="stringliteral"> const constant int* out_shape [[buffer(7)]],</span></div>
|
||||
<div class="line"><a id="l00069" name="l00069"></a><span class="lineno"> 69</span><span class="stringliteral"> const constant size_t* out_strides [[buffer(8)]],</span></div>
|
||||
<div class="line"><a id="l00070" name="l00070"></a><span class="lineno"> 70</span><span class="stringliteral"> const constant size_t& out_ndim [[buffer(9)]],</span></div>
|
||||
<div class="line"><a id="l00071" name="l00071"></a><span class="lineno"> 71</span><span class="stringliteral"> const constant int* axes [[buffer(10)]],</span></div>
|
||||
<div class="line"><a id="l00072" name="l00072"></a><span class="lineno"> 72</span><span class="stringliteral"> const constant int* idx_shapes [[buffer(11)]],</span></div>
|
||||
<div class="line"><a id="l00073" name="l00073"></a><span class="lineno"> 73</span><span class="stringliteral"> const constant size_t* idx_strides [[buffer(12)]],</span></div>
|
||||
<div class="line"><a id="l00074" name="l00074"></a><span class="lineno"> 74</span><span class="stringliteral"> const constant int& idx_ndim [[buffer(13)]],</span></div>
|
||||
<div class="line"><a id="l00075" name="l00075"></a><span class="lineno"> 75</span><span class="stringliteral"> {5}</span></div>
|
||||
<div class="line"><a id="l00076" name="l00076"></a><span class="lineno"> 76</span><span class="stringliteral"> uint2 gid [[thread_position_in_grid]]) {{</span></div>
|
||||
<div class="line"><a id="l00077" name="l00077"></a><span class="lineno"> 77</span><span class="stringliteral"> Indices<{2}, {4}> idxs{{ {{ {6} }}, idx_shapes, idx_strides, idx_ndim}};</span></div>
|
||||
<div class="line"><a id="l00078" name="l00078"></a><span class="lineno"> 78</span><span class="stringliteral"></span> </div>
|
||||
<div class="line"><a id="l00079" name="l00079"></a><span class="lineno"> 79</span><span class="stringliteral"> return scatter_impl<{1}, {2}, {3}, {4}>(</span></div>
|
||||
<div class="line"><a id="l00080" name="l00080"></a><span class="lineno"> 80</span><span class="stringliteral"> updates,</span></div>
|
||||
<div class="line"><a id="l00081" name="l00081"></a><span class="lineno"> 81</span><span class="stringliteral"> out,</span></div>
|
||||
<div class="line"><a id="l00082" name="l00082"></a><span class="lineno"> 82</span><span class="stringliteral"> upd_shape,</span></div>
|
||||
<div class="line"><a id="l00083" name="l00083"></a><span class="lineno"> 83</span><span class="stringliteral"> upd_strides,</span></div>
|
||||
<div class="line"><a id="l00084" name="l00084"></a><span class="lineno"> 84</span><span class="stringliteral"> upd_ndim,</span></div>
|
||||
<div class="line"><a id="l00085" name="l00085"></a><span class="lineno"> 85</span><span class="stringliteral"> upd_size,</span></div>
|
||||
<div class="line"><a id="l00086" name="l00086"></a><span class="lineno"> 86</span><span class="stringliteral"> out_shape,</span></div>
|
||||
<div class="line"><a id="l00087" name="l00087"></a><span class="lineno"> 87</span><span class="stringliteral"> out_strides,</span></div>
|
||||
<div class="line"><a id="l00088" name="l00088"></a><span class="lineno"> 88</span><span class="stringliteral"> out_ndim,</span></div>
|
||||
<div class="line"><a id="l00089" name="l00089"></a><span class="lineno"> 89</span><span class="stringliteral"> axes,</span></div>
|
||||
<div class="line"><a id="l00090" name="l00090"></a><span class="lineno"> 90</span><span class="stringliteral"> idxs,</span></div>
|
||||
<div class="line"><a id="l00091" name="l00091"></a><span class="lineno"> 91</span><span class="stringliteral"> gid);</span></div>
|
||||
<div class="line"><a id="l00092" name="l00092"></a><span class="lineno"> 92</span><span class="stringliteral">}}</span></div>
|
||||
<div class="line"><a id="l00093" name="l00093"></a><span class="lineno"> 93</span><span class="stringliteral">)";</span></div>
|
||||
<div class="ttc" id="ajit_2indexing_8h_html_a1a03318128191891a84707602b57b3cf"><div class="ttname"><a href="jit_2indexing_8h.html#a1a03318128191891a84707602b57b3cf">gather_kernels</a></div><div class="ttdeci">constexpr std::string_view gather_kernels</div><div class="ttdef"><b>Definition</b> indexing.h:3</div></div>
|
||||
<div class="ttc" id="ajit_2indexing_8h_html_a768c949cd650a44c6b402fc1440c1a56"><div class="ttname"><a href="jit_2indexing_8h.html#a768c949cd650a44c6b402fc1440c1a56">scatter_kernels</a></div><div class="ttdeci">constexpr std::string_view scatter_kernels</div><div class="ttdef"><b>Definition</b> indexing.h:35</div></div>
|
||||
</div><!-- fragment --></div><!-- contents -->
|
||||
|
||||
Reference in New Issue
Block a user