Skip to content

Commit

Permalink
[GLUTEN-7681][CH] [ARM]fix compile issue for SparkFunctionFloor (#7682)
Browse files Browse the repository at this point in the history
  • Loading branch information
loudongfeng authored Oct 28, 2024
1 parent 932d8a2 commit d8b0ee4
Showing 1 changed file with 10 additions and 5 deletions.
15 changes: 10 additions & 5 deletions cpp-ch/local-engine/Functions/SparkFunctionFloor.h
Original file line number Diff line number Diff line change
Expand Up @@ -168,18 +168,23 @@ struct SparkFloatFloorImpl
memcpy(p_out, &tmp_dst, tail_size_bytes);
}

#if USE_MULTITARGET_CODE
if (isArchSupported(TargetArch::AVX2))
{
if constexpr (std::is_same_v<T, Float32>)
{
TargetSpecific::AVX2::checkFloat32AndSetNullables(out.data(), null_map.data(), out.size());
return;
}
else if constexpr (std::is_same_v<T, Float64>)
{
TargetSpecific::AVX2::checkFloat64AndSetNullables(out.data(), null_map.data(), out.size());
return;
}
}
else
{
for (size_t i = 0; i < out.size(); ++i)
checkAndSetNullable(out[i], null_map[i]);
}
#endif
for (size_t i = 0; i < out.size(); ++i)
checkAndSetNullable(out[i], null_map[i]);
}
};

Expand Down

0 comments on commit d8b0ee4

Please sign in to comment.