Fix typo (#2514)
This commit is contained in:
@@ -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));
|
||||
|
||||
Reference in New Issue
Block a user