mirror of
https://github.com/c64scene-ar/llvm-6502.git
synced 2024-12-17 18:31:04 +00:00
416a0e993f
This is a more thorough fix for the issue than r203483. An IR pass will run before NVPTX codegen to make sure there are no invalid symbol names that can't be consumed by the ptxas assembler. git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@205212 91177308-0d34-0410-b5e6-96231b3b80d8
85 lines
2.5 KiB
C++
85 lines
2.5 KiB
C++
//===-- NVPTXAssignValidGlobalNames.cpp - Assign valid names to globals ---===//
|
|
//
|
|
// The LLVM Compiler Infrastructure
|
|
//
|
|
// This file is distributed under the University of Illinois Open Source
|
|
// License. See LICENSE.TXT for details.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// Clean up the names of global variables in the module to not contain symbols
|
|
// that are invalid in PTX.
|
|
//
|
|
// Currently NVPTX, like other backends, relies on generic symbol name
|
|
// sanitizing done by MC. However, the ptxas assembler is more stringent and
|
|
// disallows some additional characters in symbol names. This pass makes sure
|
|
// such names do not reach MC at all.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "NVPTX.h"
|
|
#include "llvm/IR/GlobalVariable.h"
|
|
#include "llvm/IR/Module.h"
|
|
#include "llvm/PassManager.h"
|
|
#include "llvm/Support/raw_ostream.h"
|
|
#include <string>
|
|
|
|
using namespace llvm;
|
|
|
|
namespace {
|
|
/// \brief NVPTXAssignValidGlobalNames
|
|
class NVPTXAssignValidGlobalNames : public ModulePass {
|
|
public:
|
|
static char ID;
|
|
NVPTXAssignValidGlobalNames() : ModulePass(ID) {}
|
|
|
|
virtual bool runOnModule(Module &M);
|
|
|
|
/// \brief Clean up the name to remove symbols invalid in PTX.
|
|
std::string cleanUpName(StringRef Name);
|
|
};
|
|
}
|
|
|
|
char NVPTXAssignValidGlobalNames::ID = 0;
|
|
|
|
namespace llvm {
|
|
void initializeNVPTXAssignValidGlobalNamesPass(PassRegistry &);
|
|
}
|
|
|
|
INITIALIZE_PASS(NVPTXAssignValidGlobalNames, "nvptx-assign-valid-global-names",
|
|
"Assign valid PTX names to globals", false, false)
|
|
|
|
bool NVPTXAssignValidGlobalNames::runOnModule(Module &M) {
|
|
for (GlobalVariable &GV : M.globals()) {
|
|
// We are only allowed to rename local symbols.
|
|
if (GV.hasLocalLinkage()) {
|
|
// setName doesn't do extra work if the name does not change.
|
|
// Note: this does not create collisions - if setName is asked to set the
|
|
// name to something that already exists, it adds a proper postfix to
|
|
// avoid collisions.
|
|
GV.setName(cleanUpName(GV.getName()));
|
|
}
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
std::string NVPTXAssignValidGlobalNames::cleanUpName(StringRef Name) {
|
|
std::string ValidName;
|
|
raw_string_ostream ValidNameStream(ValidName);
|
|
for (unsigned I = 0, E = Name.size(); I != E; ++I) {
|
|
char C = Name[I];
|
|
if (C == '.' || C == '@') {
|
|
ValidNameStream << "_$_";
|
|
} else {
|
|
ValidNameStream << C;
|
|
}
|
|
}
|
|
|
|
return ValidNameStream.str();
|
|
}
|
|
|
|
ModulePass *llvm::createNVPTXAssignValidGlobalNamesPass() {
|
|
return new NVPTXAssignValidGlobalNames();
|
|
}
|