mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-05-04 21:51:28 +00:00
f2cfc6b94e
* Add missing copyright statements * Use ck_tile::host_tensor_descriptor instead of a custom lambda * Refactor use of check_data_type in test classes * Use TEST_SUITE_NAME with TYPED_TEST_SUITE * Remove an unused namespace * Make dim3 const * Add BF8 x BF8 tests for CompV3 in test_gemm_pipeline_kernel_types.hpp * Add F8 x BF8 tests for CompV3 in test_gemm_pipeline_kernel_types.hpp * Add BF16 x I4 tests for CompV3 in test_gemm_pipeline_kernel_types.hpp * Add BF16 x BF16 tests for CompV3 in test_gemm_pipeline_kernel_types.hpp * Add BF8 x I4 tests for CompV3 in test_gemm_pipeline_kernel_types.hpp * Add F8 x I4 tests for CompV3 in test_gemm_pipeline_kernel_types.hpp * Add F16 x I4 tests for CompV3 in test_gemm_pipeline_kernel_types.hpp * Skip failing tests of F16 x I4 for CompV3 with K == 2 * K_Tile * Add missing precision type combinations to CompV4 from CompV3 * Move the INT8 tests around for consistency with KernelTypesCompV3Wmma * Add missing precision type combinations to CompV3Wmma from CompV3 * Remove the basic and universal tests and their dependencies * On __gfx950__, avoid using transposed loading of A with datatype pk_int4_t of B * Use ADataType and BDataType instead of ComputeDataType for WarpGemm * Explicitly set some return types to void * Use more general typenames in InterleavedPKTypeLoader * Add load_interleaved_pk_type.hpp to common.hpp * Use std::is_same_v in load_int4_tile * Add handling of LoadTranspose to load_int4_tile * Factor out common code in several places using load_int4_tile * Add support for pk_int4_t using load_int4_tile * Fix formatting
31 lines
812 B
C++
31 lines
812 B
C++
// SPDX-License-Identifier: MIT
|
|
// Copyright (c) 2025, Advanced Micro Devices, Inc. All rights reserved.
|
|
|
|
#include "test_gemm_pipeline_kernel_types.hpp"
|
|
#include "test_gemm_pipeline_util.hpp"
|
|
#include "gtest/gtest.h"
|
|
|
|
template <typename T>
|
|
class TestCkTileGemmPipelineCompV4
|
|
: public TestCkTileGemmPipeline<T, TestCkTileGemmPipelineCompV4<T>>
|
|
{
|
|
public:
|
|
static constexpr bool check_data_type()
|
|
{
|
|
using Base = TestCkTileGemmPipeline<T, TestCkTileGemmPipelineCompV4<T>>;
|
|
if constexpr(std::is_same_v<typename Base::BDataType, I4>)
|
|
{
|
|
return false;
|
|
}
|
|
return true;
|
|
}
|
|
};
|
|
|
|
#define TEST_SUITE_NAME TestCkTileGemmPipelineCompV4
|
|
|
|
TYPED_TEST_SUITE(TEST_SUITE_NAME, KernelTypesCompV4);
|
|
|
|
#include "test_gemm_pipeline_ut_cases.inc"
|
|
|
|
#undef TEST_SUITE_NAME
|