From 2c94367a70a8c1bcc4ad1d7fd7e065b3fe777783 Mon Sep 17 00:00:00 2001 From: chinakook Date: Fri, 29 Nov 2019 18:38:43 +0800 Subject: [PATCH 1/2] update symbol to json add remove_amp_cast argument to keep same with symbol.save --- python/mxnet/symbol/symbol.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/python/mxnet/symbol/symbol.py b/python/mxnet/symbol/symbol.py index c7c45a4be909..a6acee079783 100644 --- a/python/mxnet/symbol/symbol.py +++ b/python/mxnet/symbol/symbol.py @@ -1366,7 +1366,7 @@ def save(self, fname, remove_amp_cast=True): else: check_call(_LIB.MXSymbolSaveToFile(self.handle, c_str(fname))) - def tojson(self): + def tojson(self, remove_amp_cast=True): """Saves symbol to a JSON string. See Also @@ -1374,7 +1374,12 @@ def tojson(self): symbol.load_json : Used to load symbol from JSON string. """ json_str = ctypes.c_char_p() - check_call(_LIB.MXSymbolSaveToJSON(self.handle, ctypes.byref(json_str))) + if remove_amp_cast: + handle = SymbolHandle() + check_call(_LIB.MXSymbolRemoveAmpCast(self.handle, ctypes.byref(handle))) + check_call(_LIB.MXSymbolSaveToJSON(handle, ctypes.byref(json_str))) + else: + check_call(_LIB.MXSymbolSaveToJSON(self.handle, ctypes.byref(json_str))) return py_str(json_str.value) @staticmethod From 732e1a4fa55ddf0b47494fdcf6c3f695c1eb1b2f Mon Sep 17 00:00:00 2001 From: JackieWu Date: Sun, 16 Feb 2020 10:49:14 +0800 Subject: [PATCH 2/2] retrigger CI