[SYCL] Update aspect propagation for virtual functions

Author: jzc

llvm/lib/SYCLLowerIR/SYCLPropagateAspectsUsage.cpp

@@ -647,6 +647,37 @@ void setSyclFixedTargetsMD(const std::vector<Function *> &EntryPoints,
     F->setMetadata("sycl_fixed_targets", MDN);
 }
 
+void collectVirtualFunctionSetInfo(
+    Function &F, StringMap<std::vector<Function *>> &VirtualFunctionSets) {
+  if (!F.hasFnAttribute("indirectly-callable"))
+    return;
+  Attribute IndirectlyCallableAttr = F.getFnAttribute("indirectly-callable");
+  StringRef SetName = IndirectlyCallableAttr.getValueAsString();
+  VirtualFunctionSets[SetName].push_back(&F);
+}
+
+// For each set S of virtual functions that F declares,
+// propagate S through the CG and then
+void processDeclaredVirtualFunctionSets(
+    Function *F, CallGraphTy &CG, FunctionToAspectsMapTy &AspectsMap,
+    SmallPtrSet<const Function *, 16> &Visited,
+    StringMap<std::vector<Function *>> &VirtualFunctionSets) {
+  if (!F->hasFnAttribute("calls-indirectly"))
+    return;
+  Attribute CallsIndirectlyAttr = F->getFnAttribute("calls-indirectly");
+  SmallVector<StringRef, 4> DeclaredVirtualFunctionSetNames;
+  CallsIndirectlyAttr.getValueAsString().split(DeclaredVirtualFunctionSetNames,
+                                               ",");
+  auto &AspectsF = AspectsMap[F];
+  for (auto Name : DeclaredVirtualFunctionSetNames) {
+    for (auto VFn : VirtualFunctionSets[Name]) {
+      propagateAspectsThroughCG(VFn, CG, AspectsMap, Visited);
+      for (auto Aspect : AspectsMap[VFn])
+        AspectsF.insert(Aspect);
+    }
+  }
+}
+
 /// Returns a map of functions with corresponding used aspects.
 std::pair<FunctionToAspectsMapTy, FunctionToAspectsMapTy>
 buildFunctionsToAspectsMap(Module &M, TypeToAspectsMapTy &TypesWithAspects,
@@ -655,16 +686,21 @@ buildFunctionsToAspectsMap(Module &M, TypeToAspectsMapTy &TypesWithAspects,
                            bool ValidateAspects, bool FP64ConvEmu) {
   FunctionToAspectsMapTy FunctionToUsedAspects;
   FunctionToAspectsMapTy FunctionToDeclaredAspects;
+  StringMap<std::vector<Function *>> VirtualFunctionSets;
   CallGraphTy CG;
 
   for (Function &F : M.functions()) {
     processFunction(F, FunctionToUsedAspects, FunctionToDeclaredAspects,
                     TypesWithAspects, CG, AspectValues, FP64ConvEmu);
+    collectVirtualFunctionSetInfo(F, VirtualFunctionSets);
   }
 
   SmallPtrSet<const Function *, 16> Visited;
-  for (Function *F : EntryPoints)
+  for (Function *F : EntryPoints) {
     propagateAspectsThroughCG(F, CG, FunctionToUsedAspects, Visited);
+    processDeclaredVirtualFunctionSets(F, CG, FunctionToUsedAspects, Visited,
+                                       VirtualFunctionSets);
+  }
 
   if (ValidateAspects)
     validateUsedAspectsForFunctions(FunctionToUsedAspects, AspectValues,

llvm/test/SYCLLowerIR/PropagateAspectsUsage/virtual-functions-1.ll

@@ -0,0 +1,20 @@
+; RUN: opt -passes=sycl-propagate-aspects-usage < %s -S | FileCheck %s
+
+; CHECK: @vfn() #0 !sycl_used_aspects ![[#aspects:]]
+define spir_func void @vfn() #0 {
+  %tmp = alloca double
+  ret void
+}
+
+; CHECK: @foo() #1 !sycl_used_aspects ![[#aspects]]
+define spir_kernel void @foo() #1 {
+  ret void
+}
+
+; CHECK: ![[#aspects]] = !{i32 6}
+
+attributes #0 = { "indirectly-callable"="_ZTSv" }
+attributes #1 = { "calls-indirectly"="_ZTSv" }
+
+!sycl_aspects = !{!0}
+!0 = !{!"fp64", i32 6}

llvm/test/SYCLLowerIR/PropagateAspectsUsage/virtual-functions-2.ll

@@ -0,0 +1,36 @@
+; RUN: opt -passes=sycl-propagate-aspects-usage < %s -S | FileCheck %s
+
+%Foo = type { i32 }
+%Bar = type { i32 }
+
+; CHECK: @vfnFoo() #0 !sycl_used_aspects ![[#aspectsFoo:]]
+define spir_func void @vfnFoo() #0 {
+  %tmp = alloca %Foo
+  ret void
+}
+
+; CHECK: @vfnBar() #1 !sycl_used_aspects ![[#aspectsBar:]]
+define spir_func void @vfnBar() #1 {
+  %tmp = alloca %Bar
+  ret void
+}
+
+; CHECK: @kernel() #2 !sycl_used_aspects ![[#aspectsKernel:]]
+define spir_kernel void @kernel() #2 {
+  ret void
+}
+
+; CHECK: ![[#aspectsFoo]] = !{i32 1}
+; CHECK: ![[#aspectsBar]] = !{i32 2}
+; CHECK: ![[#aspectsKernel]] = !{i32 1, i32 2}
+
+attributes #0 = { "indirectly-callable"="setFoo" }
+attributes #1 = { "indirectly-callable"="setBar" }
+attributes #2 = { "calls-indirectly"="setFoo,setBar" }
+
+!sycl_aspects = !{!0}
+!0 = !{!"fp64", i32 6}
+
+!sycl_types_that_use_aspects = !{!1, !2}
+!1 = !{!"Foo", i32 1}
+!2 = !{!"Bar", i32 2}

llvm/test/SYCLLowerIR/PropagateAspectsUsage/virtual-functions-3.ll

@@ -0,0 +1,51 @@
+; RUN: opt -passes=sycl-propagate-aspects-usage < %s -S | FileCheck %s
+
+%Foo = type { i32 }
+%Bar = type { i32 }
+
+; CHECK: @vfnFoo() #0 !sycl_used_aspects ![[#aspectsFoo:]]
+define spir_func void @vfnFoo() #0 {
+  call void @subFoo()
+  ret void
+}
+
+define spir_func void @subFoo() {
+  %tmp = alloca %Foo
+  ret void
+}
+
+; CHECK: @vfnBar() #1 !sycl_used_aspects ![[#aspectsBar:]]
+define spir_func void @vfnBar() #1 {
+  call void @subBar()
+  ret void
+}
+
+define spir_func void @subBar() {
+  %tmp = alloca %Bar
+  ret void
+}
+
+; CHECK: @kernelA() #2 !sycl_used_aspects ![[#aspectsFoo]]
+define spir_kernel void @kernelA() #2 {
+  ret void
+}
+
+; CHECK: @kernelB() #3 !sycl_used_aspects ![[#aspectsBar]]
+define spir_kernel void @kernelB() #3 {
+  ret void
+}
+
+; CHECK: ![[#aspectsFoo]] = !{i32 1}
+; CHECK: ![[#aspectsBar]] = !{i32 2}
+
+attributes #0 = { "indirectly-callable"="setFoo" }
+attributes #1 = { "indirectly-callable"="setBar" }
+attributes #2 = { "calls-indirectly"="setFoo" }
+attributes #3 = { "calls-indirectly"="setBar" }
+
+!sycl_aspects = !{!0}
+!0 = !{!"fp64", i32 6}
+
+!sycl_types_that_use_aspects = !{!1, !2}
+!1 = !{!"Foo", i32 1}
+!2 = !{!"Bar", i32 2}