From 06068e7f857fb5697a21875ecc07b4ecdb443404 Mon Sep 17 00:00:00 2001
From: "Steven S. Lyubomirsky" <sslyu@cs.washington.edu>
Date: Mon, 23 Dec 2019 14:54:36 -0800
Subject: [PATCH] Update datatype check to use new TVM APIs

---
 aot/to_source.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/aot/to_source.py b/aot/to_source.py
index 77e7f4e..abc9273 100644
--- a/aot/to_source.py
+++ b/aot/to_source.py
@@ -405,7 +405,7 @@ def mk_file(body, ctx):
       cpu_ctx.device_type = kDLCPU;
       cpu_ctx.device_id = 0;
       NDArray cpu_array = nd.CopyTo(cpu_ctx);
-      CHECK_EQ(TVMType2Type(cpu_array->dtype), Bool());
+      CHECK_EQ(DataType(cpu_array->dtype), DataType::Bool());
       return reinterpret_cast<uint8_t*>(cpu_array->data)[0];
     }}