Skip to content

Commit

Permalink
formatting
Browse files Browse the repository at this point in the history
  • Loading branch information
masa committed Jul 14, 2021
1 parent 9e19ba2 commit 4769406
Showing 1 changed file with 4 additions and 5 deletions.
9 changes: 4 additions & 5 deletions src/target/llvm/codegen_llvm.cc
Original file line number Diff line number Diff line change
Expand Up @@ -524,15 +524,14 @@ void CodeGenLLVM::GetAlignment(DataType t, const VarNode* buf_var, const PrimExp
*p_alignment = align_bits / 8;
}

llvm::GlobalVariable* CodeGenLLVM::AllocateSharedMemory(DataType dtype,
size_t size,
llvm::GlobalVariable* CodeGenLLVM::AllocateSharedMemory(DataType dtype, size_t size,
unsigned int shared_address_space,
int alignment,
llvm::GlobalValue::LinkageTypes linkage) {
llvm::Type* type = llvm::ArrayType::get(DTypeToLLVMType(dtype), size);
llvm::GlobalVariable* global = new llvm::GlobalVariable(
*module_, type, false, linkage, nullptr, "shmem", nullptr,
llvm::GlobalValue::NotThreadLocal, shared_address_space);
llvm::GlobalVariable* global =
new llvm::GlobalVariable(*module_, type, false, linkage, nullptr, "shmem", nullptr,
llvm::GlobalValue::NotThreadLocal, shared_address_space);
#if TVM_LLVM_VERSION >= 100
global->setAlignment(llvm::Align(alignment));
#else
Expand Down

0 comments on commit 4769406

Please sign in to comment.