thevar1able/ClickHouse
1
0
Fork 0
mirror of https://github.com/ClickHouse/ClickHouse.git synced 2024-09-19 16:20:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

linux

Browse Source
This commit is contained in:
Roman Glinskikh 2024-05-29 22:22:17 +03:00
parent 20daaf11da
commit 4cfa85c77c
1 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
src/Functions/PartitionByHyperplanes.cpp
View File

@ -67,6 +67,9 @@ DECLARE_DEFAULT_CODE(
}
) // DECLARE_DEFAULT_CODE
#if defined(OS_LINUX)
DECLARE_AMXBF16_SPECIFIC_CODE(
static std::vector<uint16_t> bufA(2 * tile_size * tile_size), bufB(2 * tile_size * tile_size);
@ -173,6 +176,8 @@ DECLARE_AMXBF16_SPECIFIC_CODE(
}
) // DECLARE_AMX_SPECIFIC_CODE
#endif
DECLARE_AVX2_SPECIFIC_CODE(
void doMultiplyTile(
const ColumnFloat32 & nested_vectors_data,
@ -245,6 +250,8 @@ private:
size_t coordinate_index)
{
#if USE_MULTITARGET_CODE
#if defined(OS_LINUX)
if (isArchSupported(TargetArch::AMXBF16))
{
TargetSpecific::AMXBF16::doMultiplyTile(
@ -259,6 +266,7 @@ private:
coordinate_index);
return;
}
#endif
if (isArchSupported(TargetArch::AVX512BW))
{