[hotfix] missing include headers (apache#4204)

kevinthesun · Oct 30, 2019 · e216d28 · e216d28
1 parent 3d0af15
commit e216d28
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 0 deletions.
diff --git 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
@@ -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
@@ -22,6 +22,7 @@
  * \file codegen_opencl.cc
  */
 #include <tvm/packed_func_ext.h>
+#include <cmath>
 #include <vector>
 #include <string>
 #include "codegen_opencl.h"