Merge pull request #31 from slyubomirsky/dtype-refactor

[bugfix] Update datatype check to use new TVM APIs
uwsampl · Dec 24, 2019 · d48e8d5 · d48e8d5
2 parents 43c83d7 + 06068e7
commit d48e8d5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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];
     }}