Keep original time units in perFile macro expansions counting

danicheg · Jul 6, 2024 · 860e1f4 · 860e1f4
1 parent 616b83b
commit 860e1f4
Showing 1 changed file with 2 additions and 13 deletions.
diff --git a/plugin/src/main/scala/ch/epfl/scala/profilers/ProfilingImpl.scala b/plugin/src/main/scala/ch/epfl/scala/profilers/ProfilingImpl.scala
@@ -52,7 +52,7 @@ final class ProfilingImpl[G <: Global](
   }
 
   object MacroInfo {
-    final val Empty = MacroInfo(0, 0, 0.millis)
+    final val Empty = MacroInfo(0, 0, 0.nanos)
     implicit val macroInfoOrdering: Ordering[MacroInfo] = Ordering.by(_.expansionTime)
     def aggregate(infos: Iterator[MacroInfo]): MacroInfo = {
       infos.foldLeft(MacroInfo.Empty)(_ + _)
@@ -79,25 +79,14 @@ final class ProfilingImpl[G <: Global](
     import ProfilingMacroPlugin.macroInfos // , repeatedTrees}
     val perCallSite = macroInfos.toMap
     val perFile = groupPerFile(perCallSite)(MacroInfo.Empty, _ + _)
-      .map {
-        case (sf, mi) =>
-          sf -> mi.copy(expansionTime =
-            FiniteDuration(mi.expansionTime.toMillis, TimeUnit.MILLISECONDS)
-          )
-      }
     val inTotal = MacroInfo.aggregate(perFile.valuesIterator)
 
     /*    val repeated = repeatedTrees.toMap.valuesIterator
       .filter(_.count > 1)
       .map(v => v.original -> v.count)
       .toMap*/
 
-    val callSiteMicros = perCallSite.map {
-      case (k, v) =>
-        k -> v.copy(expansionTime = FiniteDuration(v.expansionTime.toMicros, TimeUnit.MICROSECONDS))
-    }
-
-    MacroProfiler(callSiteMicros, perFile, inTotal, Map.empty)
+    MacroProfiler(perCallSite, perFile, inTotal, Map.empty)
   }
 
   case class ImplicitInfo(count: Int) {