From 8b04052069f65dfcb9905e54380bfc8312abef4f Mon Sep 17 00:00:00 2001 From: Hongze Zhang Date: Wed, 15 Jan 2025 17:17:12 +0800 Subject: [PATCH] fixup --- .../gluten/extension/columnar/transition/Transition.scala | 2 +- .../gluten/extension/columnar/transition/TransitionGraph.scala | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/gluten-core/src/main/scala/org/apache/gluten/extension/columnar/transition/Transition.scala b/gluten-core/src/main/scala/org/apache/gluten/extension/columnar/transition/Transition.scala index 833592294532..0d9488d827c2 100644 --- a/gluten-core/src/main/scala/org/apache/gluten/extension/columnar/transition/Transition.scala +++ b/gluten-core/src/main/scala/org/apache/gluten/extension/columnar/transition/Transition.scala @@ -99,7 +99,7 @@ object Transition { graphCache.getOrElseUpdate( aliasOrClass, { val base = GlutenCostModel.find(aliasOrClass) - graphBuilder.build(TransitionGraph.newCostModel(base)) + graphBuilder.build(TransitionGraph.asTransitionCostModel(base)) }) } diff --git a/gluten-core/src/main/scala/org/apache/gluten/extension/columnar/transition/TransitionGraph.scala b/gluten-core/src/main/scala/org/apache/gluten/extension/columnar/transition/TransitionGraph.scala index 2a71646f1605..a5b4612cbf16 100644 --- a/gluten-core/src/main/scala/org/apache/gluten/extension/columnar/transition/TransitionGraph.scala +++ b/gluten-core/src/main/scala/org/apache/gluten/extension/columnar/transition/TransitionGraph.scala @@ -54,7 +54,7 @@ object TransitionGraph { FloydWarshallGraph.builder() } - private[transition] def newCostModel( + private[transition] def asTransitionCostModel( base: GlutenCostModel): FloydWarshallGraph.CostModel[Transition] = { new TransitionCostModel(base) }