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

60510a47 · Tianqi Chen · GitHub · 621337d5 · 60510a47
Commit 60510a47 authored Oct 16, 2017 by Tianqi Chen Committed by GitHub Oct 16, 2017
Hide whitespace changes
Inline Side-by-side

Showing with 29 additions and 0 deletions

src/codegen/intrin_rule_metal.cc
+29 -0

No files found.
--- a/src/codegen/intrin_rule_metal.cc
+++ b/src/codegen/intrin_rule_metal.cc
+/*!
+ *  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