matteosal commented on a change in pull request #20755:
URL: https://github.com/apache/incubator-mxnet/pull/20755#discussion_r758464130
##########
File path: python/mxnet/base.py
##########
@@ -957,3 +957,15 @@ def _init_np_op_module(root_module_name, np_module_name,
mx_module_name, make_op
function.__doc__ = getattr(_np_op_doc, name).__doc__
else:
function.__doc__ = re.sub('NDArray', 'ndarray', function.__doc__)
+
+
+def get_branch():
+ out = ctypes.c_char_p()
+ check_call(_LIB.MXGetBranch(ctypes.byref(out)))
+ return py_str(out.value)
+
+def get_commit_hash():
+ out = ctypes.c_char_p()
+ check_call(_LIB.MXGetCommitHash(ctypes.byref(out)))
+ return py_str(out.value)
Review comment:
Done
##########
File path: tools/diagnose.py
##########
@@ -96,13 +96,8 @@ def check_mxnet():
print('Version :', mxnet.__version__)
mx_dir = os.path.dirname(mxnet.__file__)
print('Directory :', mx_dir)
- commit_hash = os.path.join(mx_dir, 'COMMIT_HASH')
- if os.path.exists(commit_hash):
- with open(commit_hash, 'r') as f:
- ch = f.read().strip()
- print('Commit Hash :', ch)
- else:
- print('Commit hash file "{}" not found. Not installed from
pre-built package or built from source.'.format(commit_hash))
+ print('Branch :', mxnet.base.get_branch())
+ print('Commit Hash :', mxnet.base.get_commit_hash())
Review comment:
Done
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
