Commit 6d45d740 authored by Umang Yadav's avatar Umang Yadav
Browse files

disable vectorization for fp8

parent b2bb524f
......@@ -92,6 +92,10 @@ constexpr auto tensor_step(T x, Axis axis)
{
return x;
}
else if(is_same<T, migraphx::fp8e4m3fnuz>{})
{
return x;
}
else
{
constexpr auto s = decltype(x.get_shape()){};
......@@ -254,10 +258,6 @@ __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