Commit 2f188c92 by MORITA Kazutaka Committed by Tianqi Chen

Remove incorrect extension registration of tvm::Target (#1272)

parent 04268ed9
......@@ -47,22 +47,11 @@
#include <topi/rocm/dense.h>
#include <topi/rocm/vision.h>
namespace tvm {
namespace runtime {
template<>
struct extension_class_info<tvm::Target> {
static const int code = 28;
};
} // namespace tvm
} // namespace runtime
namespace topi {
using namespace tvm;
using namespace tvm::runtime;
TVM_REGISTER_EXT_TYPE(tvm::Target);
/*! \brief Canonicalize an argument that may be Array<Expr> or int to Array<Expr> */
Array<Expr> ArrayOrInt(TVMArgValue arg) {
if (arg.type_code() == kDLInt || arg.type_code() == kDLUInt) {
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment