[CODEGEN] Use correct math intrin for metal (#562)

apache · Oct 17, 2017 · 60510a4 · 60510a4
1 parent 621337d
commit 60510a4
Showing 1 changed file with 29 additions and 0 deletions.
diff --git a/src/codegen/intrin_rule_metal.cc b/src/codegen/intrin_rule_metal.cc
@@ -0,0 +1,29 @@
+/*!
+ *  Copyright (c) 2017 by Contributors
+ * \file intrin_rule_metal.cc
+ * \brief Metal intrinsic rules.
+ */
+#include "./intrin_rule.h"
+
+namespace tvm {
+namespace codegen {
+namespace intrin {
+
+TVM_REGISTER_GLOBAL("tvm.intrin.rule.metal.exp")
+.set_body(DispatchExtern<FloatDirect>);
+
+TVM_REGISTER_GLOBAL("tvm.intrin.rule.metal.log")
+.set_body(DispatchExtern<FloatDirect>);
+
+TVM_REGISTER_GLOBAL("tvm.intrin.rule.metal.tanh")
+.set_body(DispatchExtern<FloatDirect>);
+
+TVM_REGISTER_GLOBAL("tvm.intrin.rule.metal.sqrt")
+.set_body(DispatchExtern<FloatDirect>);
+
+TVM_REGISTER_GLOBAL("tvm.intrin.rule.metal.pow")
+.set_body(DispatchExtern<FloatDirect>);
+
+}  // namespace intrin
+}  // namespace codegen
+}  // namespace tvm