mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-05-19 12:30:16 +00:00
Allow building CK for specific data types and split off last remaining DL instances. (#830)
* properly split conv_nd_bwd_data instances
* split conv2d_fwd instance data types
* split the gemm, conv2d_fwd and batched_gemm_softamx_gemm
* split the tests by data types where possible
* filter examples by DTYPES
* split few remaining examples by DTYPES
* filter most instances by DTYPES
* add new lines at end of headers, fix grouped_gemm profiler
* fix syntax
* split the ckprofiler instances by DTYPES
* split the conv2d and quantization DL and XDL instances
* fix the splitting of conv2d DL instances
* split softmax and pool_fwd tests for fp16 and fp32 types
* fix syntax
* fix the dl_int8 quantization instances isolation
[ROCm/composable_kernel commit: 08eb176929]
This commit is contained in:
@@ -10,8 +10,9 @@
|
||||
|
||||
template <ck::index_t N>
|
||||
using I = ck::Number<N>;
|
||||
|
||||
#ifdef __fp16__
|
||||
using F16 = ck::half_t;
|
||||
#endif
|
||||
using F32 = float;
|
||||
|
||||
template <typename Tuple>
|
||||
@@ -22,7 +23,9 @@ class TestSoftmax : public ck::TestSoftmax<Tuple>
|
||||
// clang-format off
|
||||
using KernelTypes = ::testing::Types<
|
||||
// InDataType, AccDataType, OutDataType, Rank
|
||||
#ifdef __fp16__
|
||||
std::tuple< F16, F32, F16, I<4>>,
|
||||
#endif
|
||||
std::tuple< F32, F32, F32, I<4>>
|
||||
>;
|
||||
// clang-format on
|
||||
|
||||
Reference in New Issue
Block a user