register loadbinary_hip (#1151)

11c7b6cf · eqy · Tianqi Chen · ef58f596 · 11c7b6cf
Commit 11c7b6cf authored May 10, 2018 by eqy Committed by Tianqi Chen May 10, 2018
Show whitespace changes
Inline Side-by-side

Showing with 6 additions and 0 deletions

src/runtime/rocm/rocm_module.cc
+6 -0

No files found.
--- a/src/runtime/rocm/rocm_module.cc
+++ b/src/runtime/rocm/rocm_module.cc
@@ -229,6 +229,12 @@ TVM_REGISTER_GLOBAL("module.loadbinary_hsaco")
 .set_body([](TVMArgs args, TVMRetValue* rv) {
    *rv = ROCMModuleLoadBinary(args[0]);
  });
+
+
+TVM_REGISTER_GLOBAL("module.loadbinary_hip")
+.set_body([](TVMArgs args, TVMRetValue* rv) {
+    *rv = ROCMModuleLoadBinary(args[0]);
+  });
 }  // namespace runtime
 }  // namespace tvm
 #endif  // TVM_ROCM_RUNTIME