From 941e3a69f92628188cb4d7c2da3ef6d14a1d94cc Mon Sep 17 00:00:00 2001
From: Ryan Hill <38674843+RyanUnderhill@users.noreply.github.com>
Date: Wed, 4 Nov 2020 21:15:38 -0800
Subject: [PATCH] Test a build break fix (#5706)

---
 onnxruntime/core/providers/cuda/cuda_common.h | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/onnxruntime/core/providers/cuda/cuda_common.h b/onnxruntime/core/providers/cuda/cuda_common.h
index a1b1b64fe5..c46d1ec696 100644
--- a/onnxruntime/core/providers/cuda/cuda_common.h
+++ b/onnxruntime/core/providers/cuda/cuda_common.h
@@ -4,6 +4,8 @@
 #pragma once
 #include "cuda_pch.h"
 #include "core/common/status.h"
+#include "core/framework/arena.h"
+#include "core/framework/bfc_arena.h"
 #include "core/framework/data_transfer_manager.h"
 #include "core/framework/op_kernel.h"
 #include "core/graph/graph_viewer.h"
@@ -221,7 +223,7 @@ inline bool CalculateFdmStrides(gsl::span<fast_divmod> p, const std::vector<int6
 
 class CublasMathModeSetter {
  public:
-  CublasMathModeSetter(const cudaDeviceProp& prop,cublasHandle_t handle, cublasMath_t mode) : prop_(prop), handle_(handle) {
+  CublasMathModeSetter(const cudaDeviceProp& prop, cublasHandle_t handle, cublasMath_t mode) : prop_(prop), handle_(handle) {
     cublasGetMathMode(handle, &mode_);
 #if defined(CUDA_VERSION) && CUDA_VERSION < 11000
     if (prop.major >= 7 && mode == CUBLAS_TENSOR_OP_MATH) {