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

rebase

Browse Source
This commit is contained in:
CircleCI Docs
2024-11-22 20:24:16 +00:00
parent 6433a22733
commit 8ba8a39e43
905 changed files with 30035 additions and 16934 deletions
Download Patch File Download Diff File Expand all files Collapse all files

72
docs/build/html/metal_2kernels_2copy_8h.html vendored
View File

@@ -113,36 +113,36 @@ Functions</h2></td></tr>
<tr class="memitem:aba4530a7db6a61ca36f50e4f5e58fb77" id="r_aba4530a7db6a61ca36f50e4f5e58fb77"><td class="memTemplParams" colspan="2">template&lt;typename T , typename U &gt; </td></tr>
<tr class="memitem:aba4530a7db6a61ca36f50e4f5e58fb77"><td class="memTemplItemLeft" align="right" valign="top">void&#160;</td><td class="memTemplItemRight" valign="bottom"><a class="el" href="#aba4530a7db6a61ca36f50e4f5e58fb77">copy_g_nd1</a> (device const T *src, device U *dst, constant const int64_t &amp;src_stride, uint index)</td></tr>
<tr class="separator:aba4530a7db6a61ca36f50e4f5e58fb77"><td class="memSeparator" colspan="2">&#160;</td></tr>
<tr class="memitem:aee678c7c31119f3e609685589f37490c" id="r_aee678c7c31119f3e609685589f37490c"><td class="memTemplParams" colspan="2">template&lt;typename T , typename U &gt; </td></tr>
<tr class="memitem:aee678c7c31119f3e609685589f37490c"><td class="memTemplItemLeft" align="right" valign="top">void&#160;</td><td class="memTemplItemRight" valign="bottom"><a class="el" href="#aee678c7c31119f3e609685589f37490c">copy_g_nd2</a> (device const T *src, device U *dst, constant const int64_t *src_strides, uint2 index, uint2 grid_dim)</td></tr>
<tr class="separator:aee678c7c31119f3e609685589f37490c"><td class="memSeparator" colspan="2">&#160;</td></tr>
<tr class="memitem:a821f8f3f3891159a295c66fc25aed1ff" id="r_a821f8f3f3891159a295c66fc25aed1ff"><td class="memTemplParams" colspan="2">template&lt;typename T , typename U &gt; </td></tr>
<tr class="memitem:a821f8f3f3891159a295c66fc25aed1ff"><td class="memTemplItemLeft" align="right" valign="top">void&#160;</td><td class="memTemplItemRight" valign="bottom"><a class="el" href="#a821f8f3f3891159a295c66fc25aed1ff">copy_g_nd3</a> (device const T *src, device U *dst, constant const int64_t *src_strides, uint3 index, uint3 grid_dim)</td></tr>
<tr class="separator:a821f8f3f3891159a295c66fc25aed1ff"><td class="memSeparator" colspan="2">&#160;</td></tr>
<tr class="memitem:a778ce2dbfbaa23b24bd5efbe68448c36" id="r_a778ce2dbfbaa23b24bd5efbe68448c36"><td class="memTemplParams" colspan="2">template&lt;typename T , typename U , int N = 1&gt; </td></tr>
<tr class="memitem:a778ce2dbfbaa23b24bd5efbe68448c36"><td class="memTemplItemLeft" align="right" valign="top">void&#160;</td><td class="memTemplItemRight" valign="bottom"><a class="el" href="#a778ce2dbfbaa23b24bd5efbe68448c36">copy_g</a> (device const T *src, device U *dst, constant const int *src_shape, constant const int64_t *src_strides, constant const int &amp;ndim, uint3 index, uint3 grid_dim)</td></tr>
<tr class="separator:a778ce2dbfbaa23b24bd5efbe68448c36"><td class="memSeparator" colspan="2">&#160;</td></tr>
<tr class="memitem:a39ec5b7b8351e4332b842982a2ee6260" id="r_a39ec5b7b8351e4332b842982a2ee6260"><td class="memTemplParams" colspan="2">template&lt;typename T , typename U , typename IdxT = int64_t&gt; </td></tr>
<tr class="memitem:a39ec5b7b8351e4332b842982a2ee6260"><td class="memTemplItemLeft" align="right" valign="top">void&#160;</td><td class="memTemplItemRight" valign="bottom"><a class="el" href="#a39ec5b7b8351e4332b842982a2ee6260">copy_g_nd2</a> (device const T *src, device U *dst, constant const int64_t *src_strides, uint2 index, uint2 grid_dim)</td></tr>
<tr class="separator:a39ec5b7b8351e4332b842982a2ee6260"><td class="memSeparator" colspan="2">&#160;</td></tr>
<tr class="memitem:aab82689380897ff4716b5eafd6ef3ecc" id="r_aab82689380897ff4716b5eafd6ef3ecc"><td class="memTemplParams" colspan="2">template&lt;typename T , typename U , typename IdxT = int64_t&gt; </td></tr>
<tr class="memitem:aab82689380897ff4716b5eafd6ef3ecc"><td class="memTemplItemLeft" align="right" valign="top">void&#160;</td><td class="memTemplItemRight" valign="bottom"><a class="el" href="#aab82689380897ff4716b5eafd6ef3ecc">copy_g_nd3</a> (device const T *src, device U *dst, constant const int64_t *src_strides, uint3 index, uint3 grid_dim)</td></tr>
<tr class="separator:aab82689380897ff4716b5eafd6ef3ecc"><td class="memSeparator" colspan="2">&#160;</td></tr>
<tr class="memitem:a71e4103db4689d90ef6f9d5ba93604cf" id="r_a71e4103db4689d90ef6f9d5ba93604cf"><td class="memTemplParams" colspan="2">template&lt;typename T , typename U , int N = 1, typename IdxT = int64_t&gt; </td></tr>
<tr class="memitem:a71e4103db4689d90ef6f9d5ba93604cf"><td class="memTemplItemLeft" align="right" valign="top">void&#160;</td><td class="memTemplItemRight" valign="bottom"><a class="el" href="#a71e4103db4689d90ef6f9d5ba93604cf">copy_g</a> (device const T *src, device U *dst, constant const int *src_shape, constant const int64_t *src_strides, constant const int &amp;ndim, uint3 index, uint3 grid_dim)</td></tr>
<tr class="separator:a71e4103db4689d90ef6f9d5ba93604cf"><td class="memSeparator" colspan="2">&#160;</td></tr>
<tr class="memitem:a3278d9c999718bee3ccbe2922f501bf1" id="r_a3278d9c999718bee3ccbe2922f501bf1"><td class="memTemplParams" colspan="2">template&lt;typename T , typename U &gt; </td></tr>
<tr class="memitem:a3278d9c999718bee3ccbe2922f501bf1"><td class="memTemplItemLeft" align="right" valign="top">void&#160;</td><td class="memTemplItemRight" valign="bottom"><a class="el" href="#a3278d9c999718bee3ccbe2922f501bf1">copy_gg_nd1</a> (device const T *src, device U *dst, constant const int64_t &amp;src_stride, constant const int64_t &amp;dst_stride, uint index)</td></tr>
<tr class="separator:a3278d9c999718bee3ccbe2922f501bf1"><td class="memSeparator" colspan="2">&#160;</td></tr>
<tr class="memitem:a3e2d3cc7f34f56170409b6735f51a950" id="r_a3e2d3cc7f34f56170409b6735f51a950"><td class="memTemplParams" colspan="2">template&lt;typename T , typename U &gt; </td></tr>
<tr class="memitem:a3e2d3cc7f34f56170409b6735f51a950"><td class="memTemplItemLeft" align="right" valign="top">void&#160;</td><td class="memTemplItemRight" valign="bottom"><a class="el" href="#a3e2d3cc7f34f56170409b6735f51a950">copy_gg_nd2</a> (device const T *src, device U *dst, constant const int64_t *src_strides, constant const int64_t *dst_strides, uint2 index)</td></tr>
<tr class="separator:a3e2d3cc7f34f56170409b6735f51a950"><td class="memSeparator" colspan="2">&#160;</td></tr>
<tr class="memitem:a59f43b5bffed936d7559ceb06a10aabd" id="r_a59f43b5bffed936d7559ceb06a10aabd"><td class="memTemplParams" colspan="2">template&lt;typename T , typename U &gt; </td></tr>
<tr class="memitem:a59f43b5bffed936d7559ceb06a10aabd"><td class="memTemplItemLeft" align="right" valign="top">void&#160;</td><td class="memTemplItemRight" valign="bottom"><a class="el" href="#a59f43b5bffed936d7559ceb06a10aabd">copy_gg_nd3</a> (device const T *src, device U *dst, constant const int64_t *src_strides, constant const int64_t *dst_strides, uint3 index)</td></tr>
<tr class="separator:a59f43b5bffed936d7559ceb06a10aabd"><td class="memSeparator" colspan="2">&#160;</td></tr>
<tr class="memitem:a1e39c2683eeaf05955e7619fbd34aea5" id="r_a1e39c2683eeaf05955e7619fbd34aea5"><td class="memTemplParams" colspan="2">template&lt;typename T , typename U , int N = 1&gt; </td></tr>
<tr class="memitem:a1e39c2683eeaf05955e7619fbd34aea5"><td class="memTemplItemLeft" align="right" valign="top">void&#160;</td><td class="memTemplItemRight" valign="bottom"><a class="el" href="#a1e39c2683eeaf05955e7619fbd34aea5">copy_gg</a> (device const T *src, device U *dst, constant const int *src_shape, constant const int64_t *src_strides, constant const int64_t *dst_strides, constant const int &amp;ndim, uint3 index)</td></tr>
<tr class="separator:a1e39c2683eeaf05955e7619fbd34aea5"><td class="memSeparator" colspan="2">&#160;</td></tr>
<tr class="memitem:af0b06ac3a96852a64fa4274a94b58301" id="r_af0b06ac3a96852a64fa4274a94b58301"><td class="memTemplParams" colspan="2">template&lt;typename T , typename U , typename IdxT = int64_t&gt; </td></tr>
<tr class="memitem:af0b06ac3a96852a64fa4274a94b58301"><td class="memTemplItemLeft" align="right" valign="top">void&#160;</td><td class="memTemplItemRight" valign="bottom"><a class="el" href="#af0b06ac3a96852a64fa4274a94b58301">copy_gg_nd2</a> (device const T *src, device U *dst, constant const int64_t *src_strides, constant const int64_t *dst_strides, uint2 index)</td></tr>
<tr class="separator:af0b06ac3a96852a64fa4274a94b58301"><td class="memSeparator" colspan="2">&#160;</td></tr>
<tr class="memitem:a3f3836ad0b6545ec9b9e1864224f7a13" id="r_a3f3836ad0b6545ec9b9e1864224f7a13"><td class="memTemplParams" colspan="2">template&lt;typename T , typename U , typename IdxT = int64_t&gt; </td></tr>
<tr class="memitem:a3f3836ad0b6545ec9b9e1864224f7a13"><td class="memTemplItemLeft" align="right" valign="top">void&#160;</td><td class="memTemplItemRight" valign="bottom"><a class="el" href="#a3f3836ad0b6545ec9b9e1864224f7a13">copy_gg_nd3</a> (device const T *src, device U *dst, constant const int64_t *src_strides, constant const int64_t *dst_strides, uint3 index)</td></tr>
<tr class="separator:a3f3836ad0b6545ec9b9e1864224f7a13"><td class="memSeparator" colspan="2">&#160;</td></tr>
<tr class="memitem:ade9a9eea9b8262a854a11721fe2bb9fa" id="r_ade9a9eea9b8262a854a11721fe2bb9fa"><td class="memTemplParams" colspan="2">template&lt;typename T , typename U , int N = 1, typename IdxT = int64_t&gt; </td></tr>
<tr class="memitem:ade9a9eea9b8262a854a11721fe2bb9fa"><td class="memTemplItemLeft" align="right" valign="top">void&#160;</td><td class="memTemplItemRight" valign="bottom"><a class="el" href="#ade9a9eea9b8262a854a11721fe2bb9fa">copy_gg</a> (device const T *src, device U *dst, constant const int *src_shape, constant const int64_t *src_strides, constant const int64_t *dst_strides, constant const int &amp;ndim, uint3 index)</td></tr>
<tr class="separator:ade9a9eea9b8262a854a11721fe2bb9fa"><td class="memSeparator" colspan="2">&#160;</td></tr>
</table>
<h2 class="groupheader">Function Documentation</h2>
<a id="a778ce2dbfbaa23b24bd5efbe68448c36" name="a778ce2dbfbaa23b24bd5efbe68448c36"></a>
<h2 class="memtitle"><span class="permalink"><a href="#a778ce2dbfbaa23b24bd5efbe68448c36">&#9670;&#160;</a></span>copy_g()</h2>
<a id="a71e4103db4689d90ef6f9d5ba93604cf" name="a71e4103db4689d90ef6f9d5ba93604cf"></a>
<h2 class="memtitle"><span class="permalink"><a href="#a71e4103db4689d90ef6f9d5ba93604cf">&#9670;&#160;</a></span>copy_g()</h2>
<div class="memitem">
<div class="memproto">
<div class="memtemplate">
template&lt;typename T , typename U , int N = 1&gt; </div>
template&lt;typename T , typename U , int N = 1, typename IdxT = int64_t&gt; </div>
<table class="memname">
<tr>
<td class="memname">void copy_g </td>
@@ -217,13 +217,13 @@ template&lt;typename T , typename U &gt; </div>
</div>
</div>
<a id="aee678c7c31119f3e609685589f37490c" name="aee678c7c31119f3e609685589f37490c"></a>
<h2 class="memtitle"><span class="permalink"><a href="#aee678c7c31119f3e609685589f37490c">&#9670;&#160;</a></span>copy_g_nd2()</h2>
<a id="a39ec5b7b8351e4332b842982a2ee6260" name="a39ec5b7b8351e4332b842982a2ee6260"></a>
<h2 class="memtitle"><span class="permalink"><a href="#a39ec5b7b8351e4332b842982a2ee6260">&#9670;&#160;</a></span>copy_g_nd2()</h2>
<div class="memitem">
<div class="memproto">
<div class="memtemplate">
template&lt;typename T , typename U &gt; </div>
template&lt;typename T , typename U , typename IdxT = int64_t&gt; </div>
<table class="memname">
<tr>
<td class="memname">void copy_g_nd2 </td>
@@ -255,13 +255,13 @@ template&lt;typename T , typename U &gt; </div>
</div>
</div>
<a id="a821f8f3f3891159a295c66fc25aed1ff" name="a821f8f3f3891159a295c66fc25aed1ff"></a>
<h2 class="memtitle"><span class="permalink"><a href="#a821f8f3f3891159a295c66fc25aed1ff">&#9670;&#160;</a></span>copy_g_nd3()</h2>
<a id="aab82689380897ff4716b5eafd6ef3ecc" name="aab82689380897ff4716b5eafd6ef3ecc"></a>
<h2 class="memtitle"><span class="permalink"><a href="#aab82689380897ff4716b5eafd6ef3ecc">&#9670;&#160;</a></span>copy_g_nd3()</h2>
<div class="memitem">
<div class="memproto">
<div class="memtemplate">
template&lt;typename T , typename U &gt; </div>
template&lt;typename T , typename U , typename IdxT = int64_t&gt; </div>
<table class="memname">
<tr>
<td class="memname">void copy_g_nd3 </td>
@@ -293,13 +293,13 @@ template&lt;typename T , typename U &gt; </div>
</div>
</div>
<a id="a1e39c2683eeaf05955e7619fbd34aea5" name="a1e39c2683eeaf05955e7619fbd34aea5"></a>
<h2 class="memtitle"><span class="permalink"><a href="#a1e39c2683eeaf05955e7619fbd34aea5">&#9670;&#160;</a></span>copy_gg()</h2>
<a id="ade9a9eea9b8262a854a11721fe2bb9fa" name="ade9a9eea9b8262a854a11721fe2bb9fa"></a>
<h2 class="memtitle"><span class="permalink"><a href="#ade9a9eea9b8262a854a11721fe2bb9fa">&#9670;&#160;</a></span>copy_gg()</h2>
<div class="memitem">
<div class="memproto">
<div class="memtemplate">
template&lt;typename T , typename U , int N = 1&gt; </div>
template&lt;typename T , typename U , int N = 1, typename IdxT = int64_t&gt; </div>
<table class="memname">
<tr>
<td class="memname">void copy_gg </td>
@@ -379,13 +379,13 @@ template&lt;typename T , typename U &gt; </div>
</div>
</div>
<a id="a3e2d3cc7f34f56170409b6735f51a950" name="a3e2d3cc7f34f56170409b6735f51a950"></a>
<h2 class="memtitle"><span class="permalink"><a href="#a3e2d3cc7f34f56170409b6735f51a950">&#9670;&#160;</a></span>copy_gg_nd2()</h2>
<a id="af0b06ac3a96852a64fa4274a94b58301" name="af0b06ac3a96852a64fa4274a94b58301"></a>
<h2 class="memtitle"><span class="permalink"><a href="#af0b06ac3a96852a64fa4274a94b58301">&#9670;&#160;</a></span>copy_gg_nd2()</h2>
<div class="memitem">
<div class="memproto">
<div class="memtemplate">
template&lt;typename T , typename U &gt; </div>
template&lt;typename T , typename U , typename IdxT = int64_t&gt; </div>
<table class="memname">
<tr>
<td class="memname">void copy_gg_nd2 </td>
@@ -417,13 +417,13 @@ template&lt;typename T , typename U &gt; </div>
</div>
</div>
<a id="a59f43b5bffed936d7559ceb06a10aabd" name="a59f43b5bffed936d7559ceb06a10aabd"></a>
<h2 class="memtitle"><span class="permalink"><a href="#a59f43b5bffed936d7559ceb06a10aabd">&#9670;&#160;</a></span>copy_gg_nd3()</h2>
<a id="a3f3836ad0b6545ec9b9e1864224f7a13" name="a3f3836ad0b6545ec9b9e1864224f7a13"></a>
<h2 class="memtitle"><span class="permalink"><a href="#a3f3836ad0b6545ec9b9e1864224f7a13">&#9670;&#160;</a></span>copy_gg_nd3()</h2>
<div class="memitem">
<div class="memproto">
<div class="memtemplate">
template&lt;typename T , typename U &gt; </div>
template&lt;typename T , typename U , typename IdxT = int64_t&gt; </div>
<table class="memname">
<tr>
<td class="memname">void copy_gg_nd3 </td>