Merge branch 'develop' into lwpck-756

8c4897d1 · Rostyslav Geyyer · GitHub · 9ba9ebec · 9e86ebd6 · 8c4897d1
Unverified Commit 8c4897d1 authored Aug 30, 2023 by Rostyslav Geyyer Committed by GitHub Aug 30, 2023
Show whitespace changes
Inline Side-by-side

Showing with 8 additions and 2 deletions

test/softmax/test_softmax_rank3.cpp test/softmax/test_softmax_rank3.cpp +4 -1

test/softmax/test_softmax_rank4.cpp test/softmax/test_softmax_rank4.cpp +4 -1

No files found.
--- a/test/softmax/test_softmax_rank3.cpp
+++ b/test/softmax/test_softmax_rank3.cpp
@@ -10,8 +10,9 @@

 template <ck::index_t N>
 using I = ck::Number<N>;
-
+#ifdef CK_ENABLE_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 CK_ENABLE_FP16
    std::tuple<       F16,         F32,         F16,    I<3>>,
+#endif
    std::tuple<       F32,         F32,         F32,    I<3>>
    >;
 // clang-format on

--- a/test/softmax/test_softmax_rank4.cpp
+++ b/test/softmax/test_softmax_rank4.cpp
@@ -10,8 +10,9 @@

 template <ck::index_t N>
 using I = ck::Number<N>;
-
+#ifdef CK_ENABLE_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 CK_ENABLE_FP16
    std::tuple<       F16,         F32,         F16,    I<4>>,
+#endif
    std::tuple<       F32,         F32,         F32,    I<4>>
    >;
 // clang-format on