mirror of
https://gitee.com/milvus-io/milvus.git
synced 2026-01-03 17:31:58 +08:00
102 lines
3.1 KiB
C++
102 lines
3.1 KiB
C++
// Licensed to the Apache Software Foundation (ASF) under one
|
|
// or more contributor license agreements. See the NOTICE file
|
|
// distributed with this work for additional information
|
|
// regarding copyright ownership. The ASF licenses this file
|
|
// to you under the Apache License, Version 2.0 (the
|
|
// "License"); you may not use this file except in compliance
|
|
// with the License. You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing,
|
|
// software distributed under the License is distributed on an
|
|
// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
|
|
// KIND, either express or implied. See the License for the
|
|
// specific language governing permissions and limitations
|
|
// under the License.
|
|
|
|
// From Apache Impala (incubating) as of 2016-01-29. Pared down to a minimal
|
|
// set of functions needed for Apache Arrow / Apache parquet-cpp
|
|
|
|
#ifndef ARROW_UTIL_CPU_INFO_H
|
|
#define ARROW_UTIL_CPU_INFO_H
|
|
|
|
#include <cstdint>
|
|
#include <string>
|
|
|
|
#include "arrow/util/visibility.h"
|
|
|
|
namespace arrow {
|
|
namespace internal {
|
|
|
|
/// CpuInfo is an interface to query for cpu information at runtime. The caller can
|
|
/// ask for the sizes of the caches and what hardware features are supported.
|
|
/// On Linux, this information is pulled from a couple of sys files (/proc/cpuinfo and
|
|
/// /sys/devices)
|
|
class ARROW_EXPORT CpuInfo {
|
|
public:
|
|
static constexpr int64_t SSSE3 = (1 << 1);
|
|
static constexpr int64_t SSE4_1 = (1 << 2);
|
|
static constexpr int64_t SSE4_2 = (1 << 3);
|
|
static constexpr int64_t POPCNT = (1 << 4);
|
|
|
|
/// Cache enums for L1 (data), L2 and L3
|
|
enum CacheLevel {
|
|
L1_CACHE = 0,
|
|
L2_CACHE = 1,
|
|
L3_CACHE = 2,
|
|
};
|
|
|
|
static CpuInfo* GetInstance();
|
|
|
|
/// Determine if the CPU meets the minimum CPU requirements and if not, issue an error
|
|
/// and terminate.
|
|
void VerifyCpuRequirements();
|
|
|
|
/// Returns all the flags for this cpu
|
|
int64_t hardware_flags();
|
|
|
|
/// Returns whether of not the cpu supports this flag
|
|
bool IsSupported(int64_t flag) const { return (hardware_flags_ & flag) != 0; }
|
|
|
|
/// \brief The processor supports SSE4.2 and the Arrow libraries are built
|
|
/// with support for it
|
|
bool CanUseSSE4_2() const;
|
|
|
|
/// Toggle a hardware feature on and off. It is not valid to turn on a feature
|
|
/// that the underlying hardware cannot support. This is useful for testing.
|
|
void EnableFeature(int64_t flag, bool enable);
|
|
|
|
/// Returns the size of the cache in KB at this cache level
|
|
int64_t CacheSize(CacheLevel level);
|
|
|
|
/// Returns the number of cpu cycles per millisecond
|
|
int64_t cycles_per_ms();
|
|
|
|
/// Returns the number of cores (including hyper-threaded) on this machine.
|
|
int num_cores();
|
|
|
|
/// Returns the model name of the cpu (e.g. Intel i7-2600)
|
|
std::string model_name();
|
|
|
|
private:
|
|
CpuInfo();
|
|
|
|
void Init();
|
|
|
|
/// Inits CPU cache size variables with default values
|
|
void SetDefaultCacheSize();
|
|
|
|
int64_t hardware_flags_;
|
|
int64_t original_hardware_flags_;
|
|
int64_t cache_sizes_[L3_CACHE + 1];
|
|
int64_t cycles_per_ms_;
|
|
int num_cores_;
|
|
std::string model_name_;
|
|
};
|
|
|
|
} // namespace internal
|
|
} // namespace arrow
|
|
|
|
#endif // ARROW_UTIL_CPU_INFO_H
|