mirror of
https://github.com/llvm/llvm-project.git
synced 2025-04-26 09:46:07 +00:00
eb7d535199
Another upstreaming of C API extensions we have in Julia/LLVM.jl. Although [we went](https://github.com/maleadt/LLVM.jl/pull/431) with a string-based API there, here I'm proposing something that's similar to existing metadata/attribute APIs: - explicit functions to map syncscope names to IDs, and back - `LLVM*SyncScope` versions of builder APIs that already take a `SingleThread` argument: atomic rmw, atomic xchg, fence - `LLVMGetAtomicSyncScopeID` and `LLVMSetAtomicSyncScopeID` for other atomic instructions - testing through `llvm-c-test`'s `--echo` functionality
87 lines
2.9 KiB
C
87 lines
2.9 KiB
C
/*===-- attributes.c - tool for testing libLLVM and llvm-c API ------------===*\
|
|
|* *|
|
|
|* 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 *|
|
|
|* *|
|
|
|*===----------------------------------------------------------------------===*|
|
|
|* *|
|
|
|* This file implements the --test-attributes and --test-callsite-attributes *|
|
|
|* commands in llvm-c-test. *|
|
|
|* *|
|
|
\*===----------------------------------------------------------------------===*/
|
|
|
|
#include "llvm-c-test.h"
|
|
|
|
#include <assert.h>
|
|
#include <stdlib.h>
|
|
|
|
int llvm_test_function_attributes(void) {
|
|
LLVMEnablePrettyStackTrace();
|
|
|
|
LLVMModuleRef M = llvm_load_module(LLVMGetGlobalContext(), false, true);
|
|
|
|
LLVMValueRef F = LLVMGetFirstFunction(M);
|
|
while (F) {
|
|
// Read attributes
|
|
int Idx, ParamCount;
|
|
for (Idx = LLVMAttributeFunctionIndex, ParamCount = LLVMCountParams(F);
|
|
Idx <= ParamCount; ++Idx) {
|
|
int AttrCount = LLVMGetAttributeCountAtIndex(F, Idx);
|
|
LLVMAttributeRef *Attrs = 0;
|
|
if (AttrCount) {
|
|
Attrs =
|
|
(LLVMAttributeRef *)malloc(AttrCount * sizeof(LLVMAttributeRef));
|
|
assert(Attrs);
|
|
}
|
|
LLVMGetAttributesAtIndex(F, Idx, Attrs);
|
|
free(Attrs);
|
|
}
|
|
F = LLVMGetNextFunction(F);
|
|
}
|
|
|
|
LLVMDisposeModule(M);
|
|
|
|
return 0;
|
|
}
|
|
|
|
int llvm_test_callsite_attributes(void) {
|
|
LLVMEnablePrettyStackTrace();
|
|
|
|
LLVMModuleRef M = llvm_load_module(LLVMGetGlobalContext(), false, true);
|
|
|
|
LLVMValueRef F = LLVMGetFirstFunction(M);
|
|
while (F) {
|
|
LLVMBasicBlockRef BB;
|
|
for (BB = LLVMGetFirstBasicBlock(F); BB; BB = LLVMGetNextBasicBlock(BB)) {
|
|
LLVMValueRef I;
|
|
for (I = LLVMGetFirstInstruction(BB); I; I = LLVMGetNextInstruction(I)) {
|
|
if (LLVMIsACallInst(I)) {
|
|
// Read attributes
|
|
int Idx, ParamCount;
|
|
for (Idx = LLVMAttributeFunctionIndex,
|
|
ParamCount = LLVMCountParams(F);
|
|
Idx <= ParamCount; ++Idx) {
|
|
int AttrCount = LLVMGetCallSiteAttributeCount(I, Idx);
|
|
LLVMAttributeRef *Attrs = 0;
|
|
if (AttrCount) {
|
|
Attrs = (LLVMAttributeRef *)malloc(
|
|
AttrCount * sizeof(LLVMAttributeRef));
|
|
assert(Attrs);
|
|
}
|
|
LLVMGetCallSiteAttributes(I, Idx, Attrs);
|
|
free(Attrs);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
F = LLVMGetNextFunction(F);
|
|
}
|
|
|
|
LLVMDisposeModule(M);
|
|
|
|
return 0;
|
|
}
|