From 374b9b103ee54e0cfc72125dfac77cd58b556de7 Mon Sep 17 00:00:00 2001
From: Zhi Chen <chzhi@amazon.com>
Date: Fri, 25 Oct 2019 17:27:45 +0000
Subject: [PATCH] [hotfix] missing include headers

---
 src/arithmetic/const_fold.h   | 1 +
 src/codegen/codegen_cuda.cc   | 1 +
 src/codegen/codegen_opencl.cc | 1 +
 3 files changed, 3 insertions(+)

diff --git a/src/arithmetic/const_fold.h b/src/arithmetic/const_fold.h
index 57f90534fbb4..86f1927f2abe 100644
--- a/src/arithmetic/const_fold.h
+++ b/src/arithmetic/const_fold.h
@@ -28,6 +28,7 @@
 #include <tvm/ir_mutator.h>
 #include <tvm/expr_operator.h>
 #include <algorithm>
+#include <cmath>
 #include "int_operator.h"
 
 namespace tvm {
diff --git a/src/codegen/codegen_cuda.cc b/src/codegen/codegen_cuda.cc
index 55b4810ed4d8..39a3ab7df0cc 100644
--- a/src/codegen/codegen_cuda.cc
+++ b/src/codegen/codegen_cuda.cc
@@ -24,6 +24,7 @@
 #include <tvm/base.h>
 #include <tvm/runtime/registry.h>
 #include <tvm/packed_func_ext.h>
+#include <cmath>
 #include <vector>
 #include <string>
 #include "codegen_cuda.h"
diff --git a/src/codegen/codegen_opencl.cc b/src/codegen/codegen_opencl.cc
index 0b33bf43c151..3120bb543aea 100644
--- a/src/codegen/codegen_opencl.cc
+++ b/src/codegen/codegen_opencl.cc
@@ -22,6 +22,7 @@
  * \file codegen_opencl.cc
  */
 #include <tvm/packed_func_ext.h>
+#include <cmath>
 #include <vector>
 #include <string>
 #include "codegen_opencl.h"