The LLVMTarget detection is too strict

wudu98 · February 13, 2023, 1:48pm

When I update TVM from v0.8.0 to v0.10.0 and run autoTVM on M2, TVM report an error.

[20:46:40] /Users/xxx/tvm_v10/src/target/llvm/llvm_module.cc:396: Architecture mismatch: module=arm64-apple-darwin host=arm-apple-darwin21.5.0

If ignore the error, the code can run correctly.

github.com

apache/tvm/blob/main/src/target/llvm/llvm_module.cc#L366


  return;
}
With<LLVMTarget> llvm_target(*llvm_instance_, LLVMTarget::GetTargetMetadata(*module_));
llvm::EngineBuilder builder(std::move(module_owning_ptr_));
builder.setEngineKind(llvm::EngineKind::JIT);
builder.setOptLevel(llvm::CodeGenOpt::Aggressive);
builder.setMCPU(llvm_target->GetCPU());
builder.setMAttrs(llvm_target->GetTargetFeatures());
builder.setTargetOptions(llvm_target->GetTargetOptions());
auto tm = std::unique_ptr<llvm::TargetMachine>(builder.selectTarget());
if (!IsCompatibleWithHost(tm.get())) {
  LOG(FATAL) << "Cannot run module, architecture mismatch";
}
llvm::DataLayout layout(tm->createDataLayout());
ICHECK(layout == module_->getDataLayout())
    << "Data layout mismatch between module("
    << module_->getDataLayout().getStringRepresentation() << ")"
    << " and ExecutionEngine (" << layout.getStringRepresentation() << ")";
ee_ = builder.create(tm.release());
ICHECK(ee_ != nullptr) << "Failed to initialize jit engine for " << module_->getTargetTriple();
ee_->runStaticConstructorsDestructors(false);