fixed comment

04058bf6 · Jing Zhang · 573bd067 · 04058bf6
Commit 04058bf6 authored Sep 01, 2023 by Jing Zhang
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 0 deletions

library/include/ck/library/utility/host_tensor_generator.hpp library/include/ck/library/utility/host_tensor_generator.hpp +4 -0

No files found.
--- a/library/include/ck/library/utility/host_tensor_generator.hpp
+++ b/library/include/ck/library/utility/host_tensor_generator.hpp
@@ -95,6 +95,7 @@ struct GeneratorTensor_2<int8_t>
    }
 };

+#if defined CK_ENABLE_FP8 || defined CK_ENABLE_BF8
 template <>
 struct GeneratorTensor_2<ck::f8_t>
 {
@@ -108,6 +109,7 @@ struct GeneratorTensor_2<ck::f8_t>
        return ck::type_convert<ck::f8_t>(tmp);
    }
 };
+#endif

 template <typename T>
 struct GeneratorTensor_3
@@ -141,6 +143,7 @@ struct GeneratorTensor_3<ck::bhalf_t>
    }
 };

+#if defined CK_ENABLE_FP8 || defined CK_ENABLE_BF8
 template <>
 struct GeneratorTensor_3<ck::f8_t>
 {
@@ -157,6 +160,7 @@ struct GeneratorTensor_3<ck::f8_t>
        return ck::type_convert<ck::f8_t>(fp32_tmp);
    }
 };
+#endif

 template <typename T>
 struct GeneratorTensor_4