leavelet/cutlass
1
0
Fork 0
Code Issues Pull Requests Actions 26 Packages Projects Releases Wiki Activity

add constexpr specifier to make_tiled_copy (#2875)

Browse Source
This commit is contained in:
tsu-bin
2026-01-03 15:39:43 -05:00
committed by GitHub
parent 853ad93d60
commit 3d9de19bb7
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
include/cute/atom/copy_atom.hpp
+4 -4
View File
@@ -406,7 +406,7 @@ template <class... Args,
class LayoutCopy_TV, class LayoutCopy_TV,
class Tiler> class Tiler>
CUTE_HOST_DEVICE CUTE_HOST_DEVICE
auto auto constexpr
make_tiled_copy_impl(Copy_Atom<Args...> const& atom, make_tiled_copy_impl(Copy_Atom<Args...> const& atom,
LayoutCopy_TV const&, LayoutCopy_TV const&,
Tiler const&) Tiler const&)
@@ -420,7 +420,7 @@ make_tiled_copy_impl(Copy_Atom<Args...> const& atom,
template <class... CArgs, class... MArgs> template <class... CArgs, class... MArgs>
CUTE_HOST_DEVICE CUTE_HOST_DEVICE
auto auto constexpr
make_tiled_copy_A(Copy_Atom<CArgs...> const& copy_atom, make_tiled_copy_A(Copy_Atom<CArgs...> const& copy_atom,
TiledMMA<MArgs...> const& mma) TiledMMA<MArgs...> const& mma)
{ {
@@ -429,7 +429,7 @@ make_tiled_copy_A(Copy_Atom<CArgs...> const& copy_atom,
template <class... CArgs, class... MArgs> template <class... CArgs, class... MArgs>
CUTE_HOST_DEVICE CUTE_HOST_DEVICE
auto auto constexpr
make_tiled_copy_B(Copy_Atom<CArgs...> const& copy_atom, make_tiled_copy_B(Copy_Atom<CArgs...> const& copy_atom,
TiledMMA<MArgs...> const& mma) TiledMMA<MArgs...> const& mma)
{ {
@@ -490,7 +490,7 @@ make_tiled_copy_C_atom(Copy_Atom<CArgs...> const& copy_atom,
template <class... Args, template <class... Args,
class ThrLayout, class ThrLayout,
class ValLayout = Layout<_1>> class ValLayout = Layout<_1>>
CUTE_HOST_DEVICE CUTE_HOST_DEVICE constexpr
auto auto
make_tiled_copy(Copy_Atom<Args...> const& copy_atom, make_tiled_copy(Copy_Atom<Args...> const& copy_atom,
ThrLayout const& thr_layout = {}, // (m,n) -> thr_idx ThrLayout const& thr_layout = {}, // (m,n) -> thr_idx