From 483ffb6fe849facbd4b95eebb2c623a4b9f5bacb Mon Sep 17 00:00:00 2001
From: Zhaoyu Dong <zhdo@microsoft.com>
Date: Thu, 22 Oct 2020 00:47:35 +0800
Subject: [PATCH] use ElementCount constructor in latest llvm

---
 src/target/llvm/codegen_llvm.cc | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/src/target/llvm/codegen_llvm.cc b/src/target/llvm/codegen_llvm.cc
index cb04e6b8055b..dcb105c90a0e 100644
--- a/src/target/llvm/codegen_llvm.cc
+++ b/src/target/llvm/codegen_llvm.cc
@@ -475,7 +475,10 @@ llvm::Value* CodeGenLLVM::CreateBroadcast(llvm::Value* value, int lanes) {
   llvm::Constant* undef = llvm::UndefValue::get(type);
   llvm::Constant* zero = ConstInt32(0);
   value = builder_->CreateInsertElement(undef, value, zero);
-#if TVM_LLVM_VERSION >= 110
+#if TVM_LLVM_VERSION >= 120
+  llvm::Constant* mask = llvm::ConstantVector::getSplat(
+      llvm::ElementCount(llvm::PolySize<unsigned>::getFixed(static_cast<unsigned>(lanes))), zero);
+#elif TVM_LLVM_VERSION >= 110
   llvm::Constant* mask =
       llvm::ConstantVector::getSplat(llvm::ElementCount(lanes, /*Scalable=*/false), zero);
 #else