Tweak performance

Author: liufengyun

compiler/src/dotty/tools/dotc/transform/SpecializeFunctions.scala

@@ -105,10 +105,7 @@ class SpecializeFunctions extends MiniPhase with InfoTransformer {
   /** Dispatch to specialized `apply`s in user code when available */
   override def transformApply(tree: Apply)(using Context) =
     tree match {
-      case Apply(fun, args)
-        if fun.symbol.name == nme.apply &&
-        derivesFromFn012(fun.symbol.owner)
-      =>
+      case Apply(fun, args) if fun.symbol.name == nme.apply =>
         val paramTypes = (fun.tpe.widen.firstParamTypes :+ tree.tpe).map(_.widenSingleton.dealias)
         val isSpecializable =
           defn.isSpecializableFunction(
@@ -117,23 +114,22 @@ class SpecializeFunctions extends MiniPhase with InfoTransformer {
             paramTypes.last
           )
 
-        if (isSpecializable && !paramTypes.exists(_.isInstanceOf[ExprType])) {
-          val specializedApply = specializedName(nme.apply, paramTypes)
-          val newSel = fun match {
-            case Select(qual, _) =>
-              qual.select(specializedApply)
-            case _ =>
-              (fun.tpe: @unchecked) match {
-                case TermRef(prefix: ThisType, name) =>
-                  tpd.This(prefix.cls).select(specializedApply)
-                case TermRef(prefix: NamedType, name) =>
-                  tpd.ref(prefix).select(specializedApply)
-              }
-          }
-
-          newSel.appliedToArgs(args)
+        if (!isSpecializable || paramTypes.exists(_.isInstanceOf[ExprType])) return tree
+
+        val specializedApply = specializedName(nme.apply, paramTypes)
+        val newSel = fun match {
+          case Select(qual, _) =>
+            qual.select(specializedApply)
+          case _ =>
+            (fun.tpe: @unchecked) match {
+              case TermRef(prefix: ThisType, name) =>
+                tpd.This(prefix.cls).select(specializedApply)
+              case TermRef(prefix: NamedType, name) =>
+                tpd.ref(prefix).select(specializedApply)
+            }
         }
-        else tree
+
+        newSel.appliedToArgs(args)
 
       case _ => tree
     }