[ATTR/SYMBOL] Expose op_name attr to python (dmlc#132)

* [ATTR/SYMBOL] Expose op_name attr to python * fix xcode
tony32769 · Jul 28, 2017 · e9c2672 · e9c2672
1 parent cbe2070
commit e9c2672
Show file tree

Hide file tree

Showing 4 changed files with 16 additions and 1 deletion.
diff --git a/.travis.yml b/.travis.yml
@@ -6,6 +6,8 @@ os:
   - linux
   - osx
 
+osx_image: xcode8
+
 env:
   # code analysis
   - TASK=lint

diff --git a/dmlc-core b/dmlc-core
diff --git a/src/core/symbolic.cc b/src/core/symbolic.cc
@@ -473,6 +473,9 @@ bool Symbol::GetAttr(const std::string& key, std::string* out) const {
   if (key == "name") {
     *out = node->attrs.name;
     return true;
+  } else if (key == "op_name") {
+    *out = node->attrs.op->name;
+    return true;
   }
   auto it = node->attrs.dict.find(key);
   if (it == node->attrs.dict.end()) return false;

diff --git a/tests/python/test_symbol.py b/tests/python/test_symbol.py
@@ -45,6 +45,15 @@ def test_copy():
                 name='exp', gpu=1, attr={"kk": "1"})
     assert y.__copy__().debug_str() == y.debug_str()
 
+
+def test_op_name():
+    x = sym.Variable('x')
+    y = sym.exp(x)
+    op_name = y.attr("op_name")
+    op_func = sym.__dict__[op_name]
+    z = op_func(x)
+
+
 def test_control_dep():
     x = sym.Variable('x')
     y = sym.conv2d(data=x, name='conv')
@@ -53,6 +62,7 @@ def test_control_dep():
     t._add_control_deps([z, y])
 
 if __name__ == "__main__":
+    test_op_name()
     test_copy()
     test_default_input()
     test_compose()
-Original file line number
+Diff line change
@@ Expand Up / @@ -6,6 +6,8 @@ os: @@
       - linux
       - osx
+    osx_image: xcode8
     env:
       # code analysis
       - TASK=lint
@@ Expand Down @@
+26 −24		CMakeLists.txt
+3 −3		cmake/Modules/FindHDFS.cmake
+0 −2		cmake/lint.cmake
+27 −2		include/dmlc/any.h
+12 −0		include/dmlc/array_view.h
+27 −5		include/dmlc/base.h
+38 −5		include/dmlc/data.h
+17 −0		include/dmlc/endian.h
+7 −3		include/dmlc/json.h
+11 −3		include/dmlc/logging.h
+14 −5		include/dmlc/omp.h
+50 −0		include/dmlc/optional.h
+50 −6		include/dmlc/parameter.h
+2 −1		include/dmlc/registry.h
+9 −11		include/dmlc/serializer.h
+12 −6		include/dmlc/thread_local.h
+1 −0		include/dmlc/threadediter.h
+1 −1		include/dmlc/type_traits.h
+3 −0		make/config.mk
+7 −1		make/dmlc.mk
+46 −14		scripts/lint.py
+1 −1		scripts/travis/travis_setup_env.sh
+19 −0		src/data.cc
+97 −0		src/data/libfm_parser.h
+34 −3		src/data/row_block.h
+50 −0		src/data/strtonum.h
+2 −1		src/io.cc
+1 −0		src/io/hdfs_filesys.cc
+111 −26		src/io/s3_filesys.cc
+5 −0		src/io/s3_filesys.h
+10 −1		src/io/threaded_input_split.h
+1 −1		src/recordio.cc
+2 −1		test/dmlc_test.mk
+1 −0		test/filesys_test.cc
+36 −0		test/libfm_parser_test.cc
+130 −7		test/unittest/unittest_optional.cc
+1 −1		tracker/README.md
+8 −19		tracker/dmlc_tracker/local.py
+104 −0		tracker/dmlc_tracker/mesos.py
+16 −2		tracker/dmlc_tracker/mpi.py
+3 −1		tracker/dmlc_tracker/opts.py
+1 −1		tracker/dmlc_tracker/ssh.py
+3 −0		tracker/dmlc_tracker/submit.py
+0 −2		tracker/dmlc_tracker/yarn.py