Fix compilation issue: undefined template

Mingzhe Li · Mingzhe Li · commit 616599d77028 · 2018-03-20T12:00:45.000-07:00
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,6 +13,8 @@
  * See the License for the specific language governing permissions and
  * limitations under the License.
  */
+
+#include <sstream>
 #include "tc/core/polyhedral/codegen.h"
 
 namespace tc {
