File tree Expand file tree Collapse file tree 1 file changed +29
-0
lines changed Expand file tree Collapse file tree 1 file changed +29
-0
lines changed Original file line number Diff line number Diff line change 1+ /* !
2+ * Copyright (c) 2017 by Contributors
3+ * \file intrin_rule_metal.cc
4+ * \brief Metal intrinsic rules.
5+ */
6+ #include " ./intrin_rule.h"
7+
8+ namespace tvm {
9+ namespace codegen {
10+ namespace intrin {
11+
12+ TVM_REGISTER_GLOBAL (" tvm.intrin.rule.metal.exp" )
13+ .set_body(DispatchExtern<FloatDirect>);
14+
15+ TVM_REGISTER_GLOBAL (" tvm.intrin.rule.metal.log" )
16+ .set_body(DispatchExtern<FloatDirect>);
17+
18+ TVM_REGISTER_GLOBAL (" tvm.intrin.rule.metal.tanh" )
19+ .set_body(DispatchExtern<FloatDirect>);
20+
21+ TVM_REGISTER_GLOBAL (" tvm.intrin.rule.metal.sqrt" )
22+ .set_body(DispatchExtern<FloatDirect>);
23+
24+ TVM_REGISTER_GLOBAL (" tvm.intrin.rule.metal.pow" )
25+ .set_body(DispatchExtern<FloatDirect>);
26+
27+ } // namespace intrin
28+ } // namespace codegen
29+ } // namespace tvm
You can’t perform that action at this time.
0 commit comments