experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Migrate off deprecated llvm::VectorType::getNumElements

Browse Source 
Cast to FixedVectorType or use getElementCount where appropriate.

PiperOrigin-RevId: 354900671
Change-Id: I546953b169c5b4b7e5bf9d25d71a5d9051c2fa35
This commit is contained in:
Benjamin Kramer 2021-02-01 03:02:18 -08:00 committed by TensorFlower Gardener
parent 0f8a271836
commit b1952612df
2 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/compiler/xla/service/cpu/ir_emitter.cc
View File

@ -1547,7 +1547,7 @@ IrEmitter::EmitInnerLoopForVectorizedReduction(
auto shard_type = accumulator_shard->getType()->getPointerElementType(); auto shard_type = accumulator_shard->getType()->getPointerElementType();
if (auto vector_type = llvm::dyn_cast<llvm::VectorType>(shard_type)) { if (auto vector_type = llvm::dyn_cast<llvm::VectorType>(shard_type)) {
initial_value = initial_value =
VectorSplat(vector_type->getNumElements(), init_value_ssa); VectorSplat(vector_type->getElementCount(), init_value_ssa);
} else { } else {
initial_value = init_value_ssa; initial_value = init_value_ssa;
} }

5
tensorflow/compiler/xla/service/cpu/llvm_ir_runtime.cc
View File

@ -121,8 +121,9 @@ void RewriteCalls(
} }
// Generate the vectorized code. // Generate the vectorized code.
CHECK_EQ(vector_width, CHECK_EQ(
llvm::cast<llvm::VectorType>(input->getType())->getNumElements()); vector_width,
llvm::cast<llvm::FixedVectorType>(input->getType())->getNumElements());
llvm::Value* result = fn_body_generator(&b, input, vector_width); llvm::Value* result = fn_body_generator(&b, input, vector_width);
// Downcast result to scalar type if necessary. // Downcast result to scalar type if necessary.