Skip to content

GitLab

Commit 04058bf6 authored by Jing Zhang's avatar Jing Zhang
Browse files

fixed comment

parent 573bd067
Show whitespace changes
Inline Side-by-side
Showing with 4 additions and 0 deletions
library/include/ck/library/utility/host_tensor_generator.hpp
View file @ 04058bf6
...@@ -95,6 +95,7 @@ struct GeneratorTensor_2<int8_t> ...@@ -95,6 +95,7 @@ struct GeneratorTensor_2<int8_t>
} }
}; };
#if defined CK_ENABLE_FP8 || defined CK_ENABLE_BF8
template <> template <>
struct GeneratorTensor_2<ck::f8_t> struct GeneratorTensor_2<ck::f8_t>
{ {
...@@ -108,6 +109,7 @@ struct GeneratorTensor_2<ck::f8_t> ...@@ -108,6 +109,7 @@ struct GeneratorTensor_2<ck::f8_t>
return ck::type_convert<ck::f8_t>(tmp); return ck::type_convert<ck::f8_t>(tmp);
} }
}; };
#endif
template <typename T> template <typename T>
struct GeneratorTensor_3 struct GeneratorTensor_3
...@@ -141,6 +143,7 @@ struct GeneratorTensor_3<ck::bhalf_t> ...@@ -141,6 +143,7 @@ struct GeneratorTensor_3<ck::bhalf_t>
} }
}; };
#if defined CK_ENABLE_FP8 || defined CK_ENABLE_BF8
template <> template <>
struct GeneratorTensor_3<ck::f8_t> struct GeneratorTensor_3<ck::f8_t>
{ {
...@@ -157,6 +160,7 @@ 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); return ck::type_convert<ck::f8_t>(fp32_tmp);
} }
}; };
#endif
template <typename T> template <typename T>
struct GeneratorTensor_4 struct GeneratorTensor_4
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment