mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-05-18 20:09:25 +00:00
d49eb1d431
* 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
[ROCm/composable_kernel commit: f2cfc6b94e]
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
|