From 095df2a68c8d7c2aa53c876cd6fa0bf49740b68b Mon Sep 17 00:00:00 2001
From: ashjeong <ashjeong@umich.edu>
Date: Thu, 17 Oct 2024 18:21:36 +0900
Subject: [PATCH] style(benchmark): fix lint errors

Headers are moved out of the `TACHYON_CUDA` guard to ensure cpplint passes.
---
 benchmark/fft/fft_benchmark.cc             | 2 ++
 benchmark/fft/fft_benchmark_gpu.cc         | 2 ++
 benchmark/fft_batch/fft_batch_benchmark.cc | 2 ++
 benchmark/msm/msm_benchmark.cc             | 2 ++
 benchmark/msm/msm_benchmark_gpu.cc         | 3 ++-
 5 files changed, 10 insertions(+), 1 deletion(-)

diff --git a/benchmark/fft/fft_benchmark.cc b/benchmark/fft/fft_benchmark.cc
index 01eab98ed..ccf96f1dd 100644
--- a/benchmark/fft/fft_benchmark.cc
+++ b/benchmark/fft/fft_benchmark.cc
@@ -2,6 +2,8 @@
 #include <stdint.h>
 
 #include <iostream>
+#include <memory>
+#include <vector>
 
 // clang-format off
 #include "benchmark/fft/fft_config.h"
diff --git a/benchmark/fft/fft_benchmark_gpu.cc b/benchmark/fft/fft_benchmark_gpu.cc
index 094bd8594..ea523db1d 100644
--- a/benchmark/fft/fft_benchmark_gpu.cc
+++ b/benchmark/fft/fft_benchmark_gpu.cc
@@ -1,3 +1,5 @@
+#include <memory>
+#include <vector>
 #if TACHYON_CUDA
 #include <iostream>
 
diff --git a/benchmark/fft_batch/fft_batch_benchmark.cc b/benchmark/fft_batch/fft_batch_benchmark.cc
index 73606f9d7..b76375861 100644
--- a/benchmark/fft_batch/fft_batch_benchmark.cc
+++ b/benchmark/fft_batch/fft_batch_benchmark.cc
@@ -1,4 +1,6 @@
 #include <iostream>
+#include <string>
+#include <vector>
 
 #include "absl/strings/substitute.h"
 
diff --git a/benchmark/msm/msm_benchmark.cc b/benchmark/msm/msm_benchmark.cc
index b64855336..62c4e0f40 100644
--- a/benchmark/msm/msm_benchmark.cc
+++ b/benchmark/msm/msm_benchmark.cc
@@ -1,4 +1,6 @@
 #include <iostream>
+#include <memory>
+#include <vector>
 
 // clang-format off
 #include "benchmark/msm/msm_config.h"
diff --git a/benchmark/msm/msm_benchmark_gpu.cc b/benchmark/msm/msm_benchmark_gpu.cc
index 71d87863f..328a025c1 100644
--- a/benchmark/msm/msm_benchmark_gpu.cc
+++ b/benchmark/msm/msm_benchmark_gpu.cc
@@ -1,5 +1,6 @@
-#if TACHYON_CUDA
 #include <iostream>
+#include <vector>
+#if TACHYON_CUDA
 
 // clang-format off
 #include "benchmark/msm/msm_config.h"