Performance tweak

Author: liufengyun

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

@@ -20,29 +20,28 @@ class SpecializeFunctions extends MiniPhase with InfoTransformer {
     !ctx.settings.scalajs.value
 
   override def infoMayChange(sym: Symbol)(using Context): Boolean =
-    !sym.is(Flags.Package) && sym.isDefinedInCurrentRun
+    !sym.is(Flags.Package) && sym.isClass && sym.isDefinedInCurrentRun
 
   /** Transforms the type to include decls for specialized applys  */
   override def transformInfo(tp: Type, sym: Symbol)(using Context) = tp match {
     case tp: ClassInfo =>
       val newApplys = new mutable.ListBuffer[Symbol]
+      val apply = tp.decls.lookup(nme.apply)
+
+      if (!apply.exists) return tp
 
       var arity = 0
       while (arity < 3) {
         val func = defn.FunctionClass(arity)
         if (tp.derivesFrom(func)) {
-          var specializedMethodName: Name = null
-          def isSpecializable =
-            val paramTypes = tp.cls.typeRef.baseType(func).argInfos
-            val argTypes = paramTypes.init
-            val retType  = paramTypes.last
-            defn.isSpecializableFunction(sym.asClass, argTypes, retType) && {
-              specializedMethodName = nme.apply.specializedFunction(retType, argTypes)
-              true
-            }
+          val paramTypes = tp.cls.typeRef.baseType(func).argInfos
+          val argTypes = paramTypes.init
+          val retType  = paramTypes.last
+
+          val isSpecializable = defn.isSpecializableFunction(sym.asClass, argTypes, retType)
 
-          val apply = tp.decls.lookup(nme.apply)
-          if (apply.exists && isSpecializable) {
+          if (isSpecializable) {
+            val specializedMethodName: Name = nme.apply.specializedFunction(retType, argTypes)
             val applySpecialized = newSymbol(
               sym,
               specializedMethodName,
@@ -71,7 +70,7 @@ class SpecializeFunctions extends MiniPhase with InfoTransformer {
    *  in the template body.
    */
   override def transformTemplate(tree: Template)(using Context) = {
-    val cls = tree.symbol.enclosingClass.asClass
+    val cls = tree.symbol.owner.asClass
 
     if (!derivesFromFn012(cls)) return tree