18e70a4d50
Previously this took a reference to a map and returned a bool to say whether it succeeded. We can return a StringMap instead, as all callers but 1 simply iterated the map if the bool was true, and passed in empty maps as the starting point. lldb's lit-cpuid did specifically check whether the call failed, but due to the way the x86 routines work this works out the same as checking if the returned map is empty.
37 lines
1.1 KiB
C++
37 lines
1.1 KiB
C++
//===- lit-cpuid.cpp - Get CPU feature flags for lit exported features ----===//
|
|
//
|
|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
// See https://llvm.org/LICENSE.txt for license information.
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// lit-cpuid obtains the feature list for the currently running CPU, and outputs
|
|
// those flags that are interesting for LLDB lit tests.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "llvm/ADT/StringMap.h"
|
|
#include "llvm/Support/raw_ostream.h"
|
|
#include "llvm/TargetParser/Host.h"
|
|
|
|
using namespace llvm;
|
|
|
|
int main(int argc, char **argv) {
|
|
#if defined(__i386__) || defined(_M_IX86) || \
|
|
defined(__x86_64__) || defined(_M_X64)
|
|
const StringMap<bool> features = sys::getHostCPUFeatures();
|
|
if (features.empty())
|
|
return 1;
|
|
|
|
if (features.lookup("sse"))
|
|
outs() << "sse\n";
|
|
if (features.lookup("avx"))
|
|
outs() << "avx\n";
|
|
if (features.lookup("avx512f"))
|
|
outs() << "avx512f\n";
|
|
#endif
|
|
|
|
return 0;
|
|
}
|