Run AutoDiff closure specialization for all VJPs and not only single BB ones

kovdan01 · kovdan01 · commit fec443a795e6 · 2025-05-19T02:39:57.000+03:00
diff --git a/SwiftCompilerSources/Sources/Optimizer/FunctionPasses/ClosureSpecialization.swift b/SwiftCompilerSources/Sources/Optimizer/FunctionPasses/ClosureSpecialization.swift
@@ -122,8 +122,7 @@ let autodiffClosureSpecialization = FunctionPass(name: "autodiff-closure-special
   (function: Function, context: FunctionPassContext) in
 
   guard !function.isDefinedExternally,
-        function.isAutodiffVJP,
-        function.blocks.singleElement != nil else {
+        function.isAutodiffVJP else {
     return
   }
   

Original file line number	Diff line number	Diff line change
`@@ -122,8 +122,7 @@ let autodiffClosureSpecialization = FunctionPass(name: "autodiff-closure-special`
`122`	`122`	`(function: Function, context: FunctionPassContext) in`
`123`	`123`
`124`	`124`	`guard !function.isDefinedExternally,`
`125`		`- function.isAutodiffVJP,`
`126`		`- function.blocks.singleElement != nil else {`
	`125`	`+ function.isAutodiffVJP else {`
`127`	`126`	`return`
`128`	`127`	`}`
`129`	`128`