This commit is contained in:
Srinath Kailasa
2025-07-30 22:14:54 -04:00
committed by GitHub
parent 6fb5e667c1
commit 3b054767b3
3 changed files with 12 additions and 12 deletions
+4 -4
View File
@@ -72,8 +72,8 @@ namespace layout {
// test capacity
auto capacity = tensor_nhwc.capacity(cutlass::Tensor4DCoord(n_size, h_size, w_size, c_size));
decltype(capacity) referece_capacity = ldh * n_size;
EXPECT_EQ(capacity, referece_capacity);
decltype(capacity) reference_capacity = ldh * n_size;
EXPECT_EQ(capacity, reference_capacity);
// test packed
auto packed_tensor_layout = tensor_nhwc.packed(cutlass::Tensor4DCoord(n_size, h_size, w_size, c_size));
@@ -115,8 +115,8 @@ namespace layout {
// test capacity
auto capacity = tensor_nchw.capacity(cutlass::Tensor4DCoord(n_size, h_size, w_size, c_size));
decltype(capacity) referece_capacity = ldc * n_size;
EXPECT_EQ(capacity, referece_capacity);
decltype(capacity) reference_capacity = ldc * n_size;
EXPECT_EQ(capacity, reference_capacity);
// test packed
auto packed_tensor_layout = tensor_nchw.packed(cutlass::Tensor4DCoord(n_size, h_size, w_size, c_size));