mirror of
https://github.com/RPCS3/llvm-mirror.git
synced 2024-11-23 11:13:28 +01:00
308e2db83d
Summary: This patch makes sure that the field VFShape.VF is greater than zero when demangling the vector function name of scalable vector functions encoded in the "vector-function-abi-variant" attribute. This change is required to be able to provide instances of VFShape that can be used to query the VFDatabase for the vectorization passes, as such passes always require a positive value for the Vectorization Factor (VF) needed by the vectorization process. It is not possible to extract the value of VFShape.VF from the mangled name of scalable vector functions, because it is encoded as `x`. Therefore, the VFABI demangling function has been modified to extract such information from the IR declaration of the vector function, under the assumption that _all_ vectors in the signature of the vector function have the same number of lanes. Such assumption is valid because it is also assumed by the Vector Function ABI specifications supported by the demangling function (x86, AArch64, and LLVM internal one). The unit tests that demangle scalable names have been modified by adding the IR module that carries the declaration of the vector function name being demangled. In particular, the demangling function fails in the following cases: 1. When the declaration of the scalable vector function is not present in the module. 2. When the value of VFSHape.VF is not greater than 0. Reviewers: jdoerfert, sdesmalen, andwar Reviewed By: jdoerfert Subscribers: mgorny, kristof.beyls, hiraditya, llvm-commits Tags: #llvm Differential Revision: https://reviews.llvm.org/D73286
42 lines
1.7 KiB
C++
42 lines
1.7 KiB
C++
//===-- vfabi-demangler-fuzzer.cpp - Fuzzer VFABI using lib/Fuzzer ------===//
|
|
//
|
|
// 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
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// Build tool to fuzz the demangler for the vector function ABI names.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "llvm/Analysis/VectorUtils.h"
|
|
#include "llvm/AsmParser/Parser.h"
|
|
|
|
using namespace llvm;
|
|
|
|
extern "C" int LLVMFuzzerTestOneInput(const uint8_t *Data, size_t Size) {
|
|
LLVMContext Ctx;
|
|
SMDiagnostic Err;
|
|
const std::unique_ptr<Module> M =
|
|
parseAssemblyString("declare i32 @foo(i32 )\n", Err, Ctx);
|
|
const StringRef MangledName((const char *)Data, Size);
|
|
// Make sure that whatever symbol the demangler is operating on is
|
|
// present in the module (the signature is not important). This is
|
|
// because `tryDemangleForVFABI` fails if the function is not
|
|
// present. We need to make sure we can even invoke
|
|
// `getOrInsertFunction` because such method asserts on strings with
|
|
// zeroes.
|
|
if (!MangledName.empty() && MangledName.find_first_of(0) == StringRef::npos)
|
|
M->getOrInsertFunction(
|
|
MangledName,
|
|
FunctionType::get(Type::getVoidTy(M->getContext()), false));
|
|
const auto Info = VFABI::tryDemangleForVFABI(MangledName, *M);
|
|
|
|
// Do not optimize away the return value. Inspired by
|
|
// https://github.com/google/benchmark/blob/master/include/benchmark/benchmark.h#L307-L345
|
|
asm volatile("" : : "r,m"(Info) : "memory");
|
|
|
|
return 0;
|
|
}
|