renames

llo/opt/derive.hpp

@@ -1,4 +1,4 @@
-#include "llo/opt/plugin_opt.hpp"
+#include "llo/opt/multi_opt.hpp"
 
 #include "llo/generated/grader.hpp"
 

llo/opt/multi_opt.hpp

@@ -0,0 +1,26 @@
+#include "opt/graph_edit.hpp"
+
+#include "llo/opt/const_merge.hpp"
+#include "llo/opt/ops_merge.hpp"
+#include "llo/opt/one_prune.hpp"
+#include "llo/opt/zero_prune.hpp"
+
+#include "llo/variable.hpp"
+
+#ifndef LLO_MULTI_OPT_HPP
+#define LLO_MULTI_OPT_HPP
+
+namespace llo
+{
+
+ade::TensptrT multi_optimize (ade::TensptrT root,
+    std::vector<opt::EditFuncT> edits = {
+        const_merge_edit,
+        zero_prune_edit,
+        one_prune_edit,
+        ops_merge_edit,
+    });
+
+}
+
+#endif // LLO_MULTI_OPT_HPP

llo/python/llo.cpp

@@ -6,9 +6,11 @@
 
 #include "ade/ade.hpp"
 
+#include "llo/opt/derive.hpp"
+
+#include "llo/constant.hpp"
 #include "llo/variable.hpp"
 #include "llo/eval.hpp"
-#include "llo/opt/derive.hpp"
 
 namespace py = pybind11;
 

llo/src/derive.cpp

@@ -11,7 +11,7 @@ ade::TensptrT derive (ade::TensptrT root, ade::iTensor* target)
 	root->accept(grader);
 	auto it = grader.derivatives_.find(root.get());
 	assert(grader.derivatives_.end() != it);
-	return plugin_optimize(it->second);
+	return multi_optimize(it->second);
 }
 
 }

llo/src/multi_opt.cpp

@@ -0,0 +1,26 @@
+#include "llo/opt/multi_opt.hpp"
+
+#ifdef LLO_MULTI_OPT_HPP
+
+namespace llo
+{
+
+ade::TensptrT multi_optimize (ade::TensptrT root,
+    std::vector<opt::EditFuncT> edits)
+{
+	return opt::graph_edit(root,
+        [&edits](ade::Opcode opcode, ade::ArgsT args)
+        {
+            ade::TensptrT out;
+            for (auto it = edits.begin(), et = edits.end();
+                it != et && nullptr == out; ++it)
+            {
+                out = (*it)(opcode, args);
+            }
+            return out;
+        });
+}
+
+}
+
+#endif

llo/test/common.hpp

@@ -14,4 +14,5 @@
 
 #define EXPECT_FATAL(EVENT, MSG) try { EVENT; FAIL() << \
 	"did not expect " << #EVENT << " to succeed"; } \
-	catch (std::runtime_error& e) { EXPECT_STREQ(MSG, e.what()); }
+	catch (std::runtime_error& e) { EXPECT_STREQ(MSG, e.what()); }\
+	catch (std::exception& e) { FAIL() << "unexpected throw " << e.what(); }