Commit a866b43f authored by Umang Yadav's avatar Umang Yadav
Browse files

works without vectorization

parent f8761f3f
...@@ -142,6 +142,11 @@ MIGRAPHX_BUILTIN_TYPE_TRAITN(is_constructible); ...@@ -142,6 +142,11 @@ MIGRAPHX_BUILTIN_TYPE_TRAITN(is_constructible);
MIGRAPHX_BUILTIN_TYPE_TRAITN(is_nothrow_constructible); MIGRAPHX_BUILTIN_TYPE_TRAITN(is_nothrow_constructible);
MIGRAPHX_BUILTIN_TYPE_TRAITN(is_trivially_constructible); MIGRAPHX_BUILTIN_TYPE_TRAITN(is_trivially_constructible);
template <>
struct is_fundamental<migraphx::fp8e4m3fnuz> : false_type
{
};
template <class T> template <class T>
struct remove_cv struct remove_cv
{ {
......
...@@ -254,6 +254,10 @@ __device__ __host__ auto vectorize() ...@@ -254,6 +254,10 @@ __device__ __host__ auto vectorize()
{ {
f(xs...); f(xs...);
} }
else if constexpr(is_same<decltype(any_of(xs...)), migraphx::fp8e4m3fnuz>{})
{
return f(xs...);
}
else else
{ {
f(vectorize_tensor<N, Axis>(xs)...); f(vectorize_tensor<N, Axis>(xs)...);
......
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