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