[LLVM-14] Recover non-struct pointee types from demangling func args

Generalize getParameterTypes to return Type* instead of StructType* so
that primitive pointee types (e.g. bfloat16) are recovered from
demangling function arguments, not just struct types.

Add StructType* overload wrapper for backward compatibility.

AI-assisted: Claude Sonnet 4.6 (commercial SaaS)

Author: vmaksimo

lib/SPIRV/SPIRVInternal.h

@@ -1024,9 +1024,12 @@ bool containsUnsignedAtomicType(StringRef Name);
 std::string mangleBuiltin(StringRef UniqName, ArrayRef<Type *> ArgTypes,
                           BuiltinFuncMangleInfo *BtnInfo);
 
-/// Extract the pointee types of arguments from a mangled function name. If the
-/// corresponding type is not a pointer to a struct type, its value will be a
-/// nullptr instead.
+/// Extract the pointee types of arguments from a mangled function name,
+/// including non-struct types such as bfloat16. Unknown types are nullptr.
+void getParameterTypes(Function *F, SmallVectorImpl<Type *> &ArgTys);
+
+/// Struct-typed variant of getParameterTypes. Non-struct pointee types (e.g.
+/// bfloat) are returned as nullptr.
 void getParameterTypes(Function *F, SmallVectorImpl<StructType *> &ArgTys);
 inline void getParameterTypes(CallInst *CI,
                               SmallVectorImpl<StructType *> &ArgTys) {

lib/SPIRV/SPIRVUtil.cpp

@@ -771,7 +771,7 @@ static std::string demangleBuiltinOpenCLTypeName(StringRef MangledStructName) {
   return LlvmStructName;
 }
 
-void getParameterTypes(Function *F, SmallVectorImpl<StructType *> &ArgTys) {
+void getParameterTypes(Function *F, SmallVectorImpl<Type *> &ArgTys) {
   // If there's no mangled name, we can't do anything. Also, if there's no
   // parameters, do nothing.
   StringRef Name = F->getName();
@@ -791,7 +791,7 @@ void getParameterTypes(Function *F, SmallVectorImpl<StructType *> &ArgTys) {
       assert(!HasSret && &Arg == F->getArg(0) &&
              "sret parameter should only appear on the first argument");
       HasSret = true;
-      ArgTys.push_back(dyn_cast<StructType>(Ty));
+      ArgTys.push_back(Ty);
     } else {
       ArgTys.push_back(nullptr);
     }
@@ -858,7 +858,7 @@ void getParameterTypes(Function *F, SmallVectorImpl<StructType *> &ArgTys) {
     }
 
     for (StringRef Arg : ArgParams) {
-      StructType *Pointee = nullptr;
+      Type *Pointee = nullptr;
       if (Arg.endswith("*") && !Arg.endswith("**")) {
         // Strip off address space and other qualifiers.
         StringRef MangledStructName = Arg.split(' ').first;
@@ -889,6 +889,13 @@ void getParameterTypes(Function *F, SmallVectorImpl<StructType *> &ArgTys) {
   free(Buf);
 }
 
+void getParameterTypes(Function *F, SmallVectorImpl<StructType *> &ArgTys) {
+  SmallVector<Type *, 8> Tys;
+  getParameterTypes(F, Tys);
+  for (Type *T : Tys)
+    ArgTys.push_back(dyn_cast_or_null<StructType>(T));
+}
+
 CallInst *mutateCallInst(
     Module *M, CallInst *CI,
     std::function<std::string(CallInst *, std::vector<Value *> &)> ArgMutate,