From 3693fc87eb6db308362e1eebd577d2d9ef7a0b47 Mon Sep 17 00:00:00 2001
From: Liqiang Lu <liqiangxl@gmail.com>
Date: Thu, 30 Mar 2023 10:36:11 -0700
Subject: [PATCH] fix benchmark GeluBackward

---
 benchmark/gelu_backward.cpp | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

diff --git a/benchmark/gelu_backward.cpp b/benchmark/gelu_backward.cpp
index ebd318c6a15..4e271fb3f54 100644
--- a/benchmark/gelu_backward.cpp
+++ b/benchmark/gelu_backward.cpp
@@ -146,11 +146,11 @@ static void GeluBackward_Compile(benchmark::State& benchmark_state) {
   // inputs
   std::vector<c10::IValue> inputs = setupInputs();
 
-  schedulePointwise(&fusion, c10::ArrayRef<c10::IValue>(inputs));
+  auto lparams = schedulePointwise(&fusion, c10::ArrayRef<c10::IValue>(inputs));
 
   for (auto _ : benchmark_state) {
     FusionExecutor executor;
-    executor.compileFusion(&fusion, inputs);
+    executor.compileFusion(&fusion, inputs, lparams);
   }
 }
 
@@ -173,7 +173,7 @@ static void GeluBackward_RunFusion(benchmark::State& benchmark_state) {
   auto lparams = schedulePointwise(&fusion, c10::ArrayRef<c10::IValue>(inputs));
 
   FusionExecutor executor;
-  executor.compileFusion(&fusion, inputs);
+  executor.compileFusion(&fusion, inputs, lparams);
 
   C10_CUDA_CHECK(cudaDeviceSynchronize());
 
@@ -204,7 +204,7 @@ static void GeluBackward_RunFusion_GpuOnly(benchmark::State& benchmark_state) {
 
   FusionExecutor executor;
   executor.setMeasureKernelTimeFlag(true);
-  executor.compileFusion(&fusion, inputs);
+  executor.compileFusion(&fusion, inputs, lparams);
 
   C10_CUDA_CHECK(cudaDeviceSynchronize());
 
@@ -237,7 +237,7 @@ static void GeluBackward_RunFusion_CpuOnly(benchmark::State& benchmark_state) {
 
   FusionExecutor executor;
   executor.setExecuteKernelFlag(false);
-  executor.compileFusion(&fusion, inputs);
+  executor.compileFusion(&fusion, inputs, lparams);
 
   for (auto _ : benchmark_state) {
     outputs = executor.runFusion(c10::ArrayRef<c10::IValue>(inputs), lparams);