support fp64 scale to improve accuracy

apache · Apr 27, 2020 · 833c70d · 833c70d
1 parent 7a38125
commit 833c70d
Show file tree

Hide file tree

Showing 24 changed files with 224 additions and 1,426 deletions.
diff --git a/3rdparty/dmlc-core b/3rdparty/dmlc-core
diff --git a/include/tvm/ir/module.h b/include/tvm/ir/module.h
@@ -365,7 +365,5 @@ TVM_DLL std::string PrettyPrint(const ObjectRef& node);
 TVM_DLL std::string AsText(const ObjectRef& node,
                            bool show_meta_data = true,
                            runtime::TypedPackedFunc<std::string(ObjectRef)> annotate = nullptr);
-
-TVM_DLL std::string AsGraph(const ObjectRef& node);
 }  // namespace tvm
 #endif  // TVM_IR_MODULE_H_
diff --git a/include/tvm/tir/ir_pass.h b/include/tvm/tir/ir_pass.h
diff --git a/python/tvm/driver/build_module.py b/python/tvm/driver/build_module.py
@@ -176,7 +176,7 @@ def lower(sch,
         tvm.tir.transform.InjectPrefetch(),
         tvm.tir.transform.StorageFlatten(64, cfg.instrument_bound_checkers),
         tvm.tir.transform.NarrowDataType(32),
-        tvm.tir.transform.Simplify(),
+        # tvm.tir.transform.Simplify(),
     ]
     pass_list += lower_phase1
 
@@ -199,7 +199,7 @@ def lower(sch,
 
     # Phase 3
     pass_list += [
-        tvm.tir.transform.Simplify(),
+        #tvm.tir.transform.Simplify(),
         tvm.tir.transform.RemoveNoOp(),
     ]
 

diff --git a/python/tvm/ir/base.py b/python/tvm/ir/base.py
@@ -52,10 +52,6 @@ def astext(self, show_meta_data=True, annotate=None):
         """
         return _ffi_api.AsText(self, show_meta_data, annotate)
 
-    def asgraph(self):
-        """Get .dot graph format of the expression"""
-        return _ffi_api.AsGraph(self)
-
     def __str__(self):
         return _ffi_api.PrettyPrint(self)
 

diff --git a/python/tvm/relay/data_dep_optimization/utils.py b/python/tvm/relay/data_dep_optimization/utils.py
@@ -38,15 +38,3 @@ def _run_opt_pass(expr, opt_pass):
     mod = tvm.IRModule.from_expr(expr)
     mod = opt_pass(mod)
     return mod["main"]
-
-def get_scalar_from_constant(expr):
-    """ Returns scalar value from Relay constant scalar. """
-    assert isinstance(expr, _expr.Constant) and not expr.data.shape, \
-            "Expr is not a constant scalar."
-    value = expr.data.asnumpy()
-    if value.dtype == np.dtype(np.int32):
-        return int(value)
-    if value.dtype == np.dtype(np.float32) or value.dtype == np.dtype(np.float64):
-        return float(value)
-    assert False, "Constant expr must be float32/int32"
-    return None  # To suppress pylint