1 files changed, 14 insertions, 14 deletions
diff --git a/llvm/lib/Target/AVR/AVRInstrumentFunctions.cpp b/llvm/lib/Target/AVR/AVRInstrumentFunctions.cpp
index a2f20e02be1..5553dc2da31 100644
--- a/llvm/lib/Target/AVR/AVRInstrumentFunctions.cpp
+++ b/llvm/lib/Target/AVR/AVRInstrumentFunctions.cpp
@@ -72,6 +72,18 @@ static Value *CreateStringPtr(BasicBlock &BB, StringRef Str) {
     {ConstantInt::get(I8, 0), ConstantInt::get(I8, 0)}, "", &BB);
 }
 
+static std::string GetTypeName(Type &Ty) {
+  if (auto *IntTy = dyn_cast<IntegerType>(&Ty)) {
+    return std::string("i") + std::to_string(IntTy->getBitWidth());
+  }
+
+  if (Ty.isFloatingPointTy()) {
+    return std::string("f") + std::to_string(Ty.getPrimitiveSizeInBits());
+  }
+
+  llvm_unreachable("unknown return type");
+}
+
 /// Builds a call to one of the signature begin/end hooks.
 static void BuildSignatureCall(StringRef SymName, BasicBlock &BB, Function &F) {
   LLVMContext &Ctx = F.getContext();
@@ -103,13 +115,7 @@ static void BuildEndSignature(BasicBlock &BB, Function &F) {
 /// Get the name of the external symbol that we need to call
 /// to notify about this argument.
 static std::string GetArgumentSymbolName(Argument &Arg) {
-  Type *Ty = Arg.getType();
-
-  if (auto *IntTy = dyn_cast<IntegerType>(Ty)) {
-    return (symbols::PREFIX + "_argument_i" + std::to_string(IntTy->getBitWidth())).str();
-  }
-
-  llvm_unreachable("unknown argument type");
+  return (symbols::PREFIX + "_argument_" + GetTypeName(*Arg.getType())).str();
 }
 
 /// Builds a call to one of the argument hooks.
@@ -153,13 +159,7 @@ static void BuildEntryBlock(Function &F) {
 }
 
 static std::string GetReturnSymbolName(Value &Val) {
-  Type *Ty = Val.getType();
-
-  if (auto *IntTy = dyn_cast<IntegerType>(Ty)) {
-    return (symbols::PREFIX + "_result_u" + std::to_string(IntTy->getBitWidth())).str();
-  }
-
-  llvm_unreachable("unknown return type");
+  return (symbols::PREFIX + "_result_" + GetTypeName(*Val.getType())).str();
 }
 
 static void BuildExitHook(Instruction &I) {