fixing local caching in ndarray shape

apache · Aug 12, 2019 · 7ca3d32 · 7ca3d32
1 parent 24ea3c1
commit 7ca3d32
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/python/mxnet/ndarray/ndarray.py b/python/mxnet/ndarray/ndarray.py
@@ -110,7 +110,7 @@
 # Caching whether MXNet was built with INT64 support or not
 _INT64_TENSOR_SIZE_ENABLED = None
 
-def int64_enabled():
+def _int64_enabled():
     global _INT64_TENSOR_SIZE_ENABLED
     if _INT64_TENSOR_SIZE_ENABLED is None:
         _INT64_TENSOR_SIZE_ENABLED = Features().is_enabled('INT64_TENSOR_SIZE')
@@ -142,7 +142,7 @@ def _new_alloc_handle(shape, ctx, delay_alloc, dtype=mx_real_t):
         A new empty `NDArray` handle.
     """
     hdl = NDArrayHandle()
-    if sys.version_info[0] > 2 and int64_enabled():
+    if sys.version_info[0] > 2 and _int64_enabled():
         check_call(_LIB.MXNDArrayCreateEx64(
             c_array_buf(mx_int64, native_array('q', shape)),
             ctypes.c_int(len(shape)),
@@ -2238,7 +2238,7 @@ def shape(self):
         (2L, 3L, 4L)
         """
         ndim = mx_int()
-        if _INT64_TENSOR_SIZE_ENABLED:
+        if _int64_enabled():
             pdata = ctypes.POINTER(mx_int64)()
             check_call(_LIB.MXNDArrayGetShapeEx64(
                 self.handle, ctypes.byref(ndim), ctypes.byref(pdata)))

diff --git a/python/mxnet/symbol/symbol.py b/python/mxnet/symbol/symbol.py
@@ -39,7 +39,7 @@
 from ..base import check_call, MXNetError, NotImplementedForSymbol
 from ..context import Context, current_context
 from ..ndarray import NDArray, _DTYPE_NP_TO_MX, _DTYPE_MX_TO_NP, _GRAD_REQ_MAP
-from ..ndarray.ndarray import _STORAGE_TYPE_STR_TO_ID, int64_enabled
+from ..ndarray.ndarray import _STORAGE_TYPE_STR_TO_ID, _int64_enabled
 from ..ndarray import _ndarray_cls
 from ..executor import Executor
 from . import _internal