Brecht Van Lommel
a84a8a528d
While keeping SSE2, SSE4.1 and AVX2. This does not affect hardware support, it only slightly reduces performance for some older CPUs. To reduce maintenance cost and improve compile times. Differential Revision: https://developer.blender.org/D16978
93 lines
2.5 KiB
C++
93 lines
2.5 KiB
C++
/* SPDX-License-Identifier: Apache-2.0
|
|
* Copyright 2011-2022 Blender Foundation */
|
|
|
|
#pragma once
|
|
|
|
#include "util/debug.h"
|
|
#include "util/system.h"
|
|
|
|
CCL_NAMESPACE_BEGIN
|
|
|
|
/* A wrapper around per-microarchitecture variant of a kernel function.
|
|
*
|
|
* Provides a function-call-like API which gets routed to the most suitable implementation.
|
|
*
|
|
* For example, on a computer which only has SSE4.1 the kernel_sse41 will be used. */
|
|
template<typename FunctionType> class CPUKernelFunction {
|
|
public:
|
|
CPUKernelFunction(FunctionType kernel_default,
|
|
FunctionType kernel_sse2,
|
|
FunctionType kernel_sse41,
|
|
FunctionType kernel_avx2)
|
|
{
|
|
kernel_info_ = get_best_kernel_info(kernel_default, kernel_sse2, kernel_sse41, kernel_avx2);
|
|
}
|
|
|
|
template<typename... Args> inline auto operator()(Args... args) const
|
|
{
|
|
assert(kernel_info_.kernel);
|
|
|
|
return kernel_info_.kernel(args...);
|
|
}
|
|
|
|
const char *get_uarch_name() const
|
|
{
|
|
return kernel_info_.uarch_name;
|
|
}
|
|
|
|
protected:
|
|
/* Helper class which allows to pass human-readable microarchitecture name together with function
|
|
* pointer. */
|
|
class KernelInfo {
|
|
public:
|
|
KernelInfo() : KernelInfo("", nullptr)
|
|
{
|
|
}
|
|
|
|
/* TODO(sergey): Use string view, to have higher-level functionality (i.e. comparison) without
|
|
* memory allocation. */
|
|
KernelInfo(const char *uarch_name, FunctionType kernel)
|
|
: uarch_name(uarch_name), kernel(kernel)
|
|
{
|
|
}
|
|
|
|
const char *uarch_name;
|
|
FunctionType kernel;
|
|
};
|
|
|
|
KernelInfo get_best_kernel_info(FunctionType kernel_default,
|
|
FunctionType kernel_sse2,
|
|
FunctionType kernel_sse41,
|
|
FunctionType kernel_avx2)
|
|
{
|
|
/* Silence warnings about unused variables when compiling without some architectures. */
|
|
(void)kernel_sse2;
|
|
(void)kernel_sse41;
|
|
(void)kernel_avx2;
|
|
|
|
#ifdef WITH_CYCLES_OPTIMIZED_KERNEL_AVX2
|
|
if (DebugFlags().cpu.has_avx2() && system_cpu_support_avx2()) {
|
|
return KernelInfo("AVX2", kernel_avx2);
|
|
}
|
|
#endif
|
|
|
|
#ifdef WITH_CYCLES_OPTIMIZED_KERNEL_SSE41
|
|
if (DebugFlags().cpu.has_sse41() && system_cpu_support_sse41()) {
|
|
return KernelInfo("SSE4.1", kernel_sse41);
|
|
}
|
|
#endif
|
|
|
|
#ifdef WITH_CYCLES_OPTIMIZED_KERNEL_SSE2
|
|
if (DebugFlags().cpu.has_sse2() && system_cpu_support_sse2()) {
|
|
return KernelInfo("SSE2", kernel_sse2);
|
|
}
|
|
#endif
|
|
|
|
return KernelInfo("default", kernel_default);
|
|
}
|
|
|
|
KernelInfo kernel_info_;
|
|
};
|
|
|
|
CCL_NAMESPACE_END
|