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