Skip to content

Commit 2212f1b

Browse files
committed
[CuTe] [Xe] Allow size-1 fragments in block 2D copies
1 parent 65b1a3f commit 2212f1b

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

include/cute/atom/copy_traits_xe_2d.hpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -56,7 +56,7 @@ namespace cute {
5656

5757
// Utility to check if a layout belongs to a coordinate tensor.
5858
template <typename Layout>
59-
static constexpr bool is_counting_layout_v = is_arithmetic_tuple_like<decltype(Layout{}(0))>::value;
59+
static constexpr bool is_counting_layout_v = is_arithmetic_tuple_like<decltype(Layout{}(0))>::value || is_constant_v<1, decltype(size(Layout{}))>;
6060

6161

6262

0 commit comments

Comments
 (0)