ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-06-30 11:47:48 +00:00
Code Issues Packages Projects Releases Wiki Activity

Format file and enable gmock.

Browse Source
This commit is contained in:
John Shumway
2025-09-18 22:38:40 +00:00
parent 9f65631f00
commit b4bb2bf317
2 changed files with 9 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
cmake/gtest.cmake
View File

@@ -24,7 +24,7 @@ if(WIN32)
set(gtest_force_shared_crt ON CACHE_INTERNAL "")
endif()
set(BUILD_GMOCK OFF CACHE INTERNAL "")
set(BUILD_GMOCK ON CACHE INTERNAL "")
set(INSTALL_GTEST OFF CACHE INTERNAL "")
# Store the current value of BUILD_SHARED_LIBS
@@ -38,7 +38,7 @@ set(ROCM_DISABLE_CHECKS FALSE)
# Restore the old value of BUILD_SHARED_LIBS
set(BUILD_SHARED_LIBS ${__build_shared_libs} CACHE BOOL "Type of libraries to build" FORCE)
set(BUILD_GMOCK OFF CACHE INTERNAL "")
set(BUILD_GMOCK ON CACHE INTERNAL "")
set(INSTALL_GTEST OFF CACHE INTERNAL "")
set(GTEST_CXX_FLAGS

13
experimental/builder/test/test_conv_grp_bwd_2d.cpp
View File

@@ -44,17 +44,18 @@ TEST(ConvBuilderGrpBwd2d, TestFirstExample)
// include/ck/library/tensor_operation_instance/gpu/grouped_conv_bwd_data/device_grouped_conv_bwd_data_xdl_instance.hpp
static constexpr const ConvSignature SIGNATURE;
static constexpr const ConvAlgorithm ALGORITHM{
.thread_block = {.block_size = 64, .submatrix = {.m = 16, .n = 64, .k = 32}},
.thread_block = {.block_size = 64, .submatrix = {.m = 16, .n = 64, .k = 32}},
// .tuning_params = {.ak1 = 8, .bk1 = 8, .m_xdl_per_wave = 4, .n_xdl_per_wave = 1},
// .block_transfer = {
// .thread_cluster_dims_a = {.k0 = 4, .m = 16, .k1 = 1},
// .thread_cluster_dims_b = {.k0 = 4, .n = 8, .k1 = 1}}
};
};
using Builder = ckb::ConvBuilder<SIGNATURE, ALGORITHM>;
EXPECT_EQ(
Builder::Instance::TypeString(),
"DeviceGroupedConvBwdDataMultipleD_Xdl_CShuffle_v1<64, 16, 64, 32, 8, 8, Default, 16, 16, 1, 4, 8, 8, 1, 1, "
"TransposeTransferInScalarPerVectorAligned: 1, TransposeTransferOutScalarPerVectorAligned: 1>");
EXPECT_EQ(Builder::Instance::TypeString(),
"DeviceGroupedConvBwdDataMultipleD_Xdl_CShuffle_v1<64, 16, 64, 32, 8, 8, Default, "
"16, 16, 1, 4, 8, 8, 1, 1, "
"TransposeTransferInScalarPerVectorAligned: 1, "
"TransposeTransferOutScalarPerVectorAligned: 1>");
}
} // namespace