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

Updates for 3.4 release. (#1305)

Browse Source
This commit is contained in:
ANIKET SHIVAM
2024-01-16 10:42:51 -08:00
committed by GitHub
parent acba5beee5
commit 2f589ffa76
166 changed files with 5996 additions and 4702 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
test/unit/cute/core/logical_divide.cpp
View File

@@ -45,10 +45,10 @@ test_logical_divide(LayoutA const& layoutA,
auto layoutR = logical_divide(layoutA, layoutB);
CUTLASS_TRACE_HOST("test_logical_divide()");
CUTLASS_TRACE_HOST(shape(layoutA) << " / " << shape(layoutB) << " => " << shape(layoutR) );
CUTLASS_TRACE_HOST( shape(layoutA) << " / " << shape(layoutB) << " => " << shape(layoutR));
CUTLASS_TRACE_HOST(stride(layoutA) << " " << stride(layoutB) << " => " << stride(layoutR));
// Test that layout R is compatible with layout B
// Test that layout B is compatible with layout R_0
ASSERT_EQ(rank(layoutR), 2);
ASSERT_TRUE(compatible(layoutB, layout<0>(layoutR)));
}
@@ -186,10 +186,10 @@ TEST(CuTe_core, Logical_divide)
// Enforcement for dynamic cases
auto result = logical_divide(layout, tile);
static_assert(decltype(shape<0>(result) == Int<32>{})::value);
static_assert(decltype(stride<0>(result) == Int<1>{})::value);
assert(shape<1>(result) == 1);
static_assert(decltype(stride<1>(result) == Int<32>{})::value);
ASSERT_TRUE(decltype(shape<0>(result) == Int<32>{})::value);
ASSERT_TRUE(decltype(stride<0>(result) == Int<1>{})::value);
ASSERT_TRUE(shape<1>(result) == 1);
ASSERT_TRUE(decltype(stride<1>(result) == Int<32>{})::value);
}
{
@@ -200,10 +200,10 @@ TEST(CuTe_core, Logical_divide)
// Enforcement for dynamic cases
auto result = logical_divide(layout, tile);
static_assert(decltype(shape<0>(result) == Int<32>{})::value);
static_assert(decltype(stride<0>(result) == Int<1>{})::value);
assert(shape<1>(result) == 2);
static_assert(decltype(stride<1>(result) == Int<32>{})::value);
ASSERT_TRUE(decltype(shape<0>(result) == Int<32>{})::value);
ASSERT_TRUE(decltype(stride<0>(result) == Int<1>{})::value);
ASSERT_TRUE(shape<1>(result) == 2);
ASSERT_TRUE(decltype(stride<1>(result) == Int<32>{})::value);
}
{
@@ -221,10 +221,10 @@ TEST(CuTe_core, Logical_divide)
// Enforcement for dynamic cases
auto result = logical_divide(layout, tile);
static_assert(decltype(shape<0>(result) == Int<48>{})::value);
static_assert(decltype(stride<0>(result) == Int<1>{})::value);
assert(shape<1>(result) == 1);
static_assert(decltype(stride<1>(result) == Int<48>{})::value);
ASSERT_TRUE(decltype(shape<0>(result) == Int<48>{})::value);
ASSERT_TRUE(decltype(stride<0>(result) == Int<1>{})::value);
ASSERT_TRUE(shape<1>(result) == 1);
ASSERT_TRUE(decltype(stride<1>(result) == Int<48>{})::value);
}
// DISALLOWED