Skip to content

Commit 60510a4

Browse files
authored
[CODEGEN] Use correct math intrin for metal (#562)
1 parent 621337d commit 60510a4

File tree

1 file changed

+29
-0
lines changed

1 file changed

+29
-0
lines changed

src/codegen/intrin_rule_metal.cc

Lines changed: 29 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,29 @@
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

0 commit comments

Comments
 (0)