[hotfix] missing include headers (#4204)

diff --git a/src/arithmetic/const_fold.h b/src/arithmetic/const_fold.h
index 57f9053..86f1927 100644 (file)
--- 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 55b4810..39a3ab7 100644 (file)
--- 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 0b33bf4..3120bb5 100644 (file)
--- 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"