Merge pull request #181 from mingzhe09088/fix-undefined

prigoyal · web-flow · commit aad29763e0eb · 2018-03-21T08:38:48.000-04:00
Fix compilation issue: undefined template
diff --git a/include/tc/core/polyhedral/schedule_tree_elem.h b/include/tc/core/polyhedral/schedule_tree_elem.h
@@ -16,6 +16,7 @@
 #pragma once
 
 #include <memory>
+#include <sstream>
 #include <unordered_set>
 #include <vector>
 
diff --git a/src/core/polyhedral/codegen.cc b/src/core/polyhedral/codegen.cc
@@ -13,7 +13,9 @@
  * See the License for the specific language governing permissions and
  * limitations under the License.
  */
+
 #include "tc/core/polyhedral/codegen.h"
+#include <sstream>
 
 namespace tc {
 namespace polyhedral {
