add tanh dispatch (#619)

c7101537 · masahi · Tianqi Chen · c1008ec4 · c7101537
Commit c7101537 authored Nov 07, 2017 by masahi Committed by Tianqi Chen Nov 06, 2017
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 0 deletions

src/codegen/llvm/intrin_rule_rocm.cc
+2 -0

No files found.
--- a/src/codegen/llvm/intrin_rule_rocm.cc
+++ b/src/codegen/llvm/intrin_rule_rocm.cc
@@ -43,6 +43,8 @@ TVM_REGISTER_GLOBAL("tvm.intrin.rule.rocm.sqrt")
 TVM_REGISTER_GLOBAL("tvm.intrin.rule.rocm.pow")
 .set_body(DispatchExternOCML);

+TVM_REGISTER_GLOBAL("tvm.intrin.rule.rocm.tanh")
+.set_body(DispatchExternOCML);
 }  // namespace llvm
 }  // namespace codegen
 }  // namespace tvm