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

[XLA:CPU] Remove uses of deprecated CreateAlignedLoad/Store builders

Browse Source 
These are being removed in f515ca8995

PiperOrigin-RevId: 357555615
Change-Id: I0e5858797c0459cc67221fa2fef1ee39cf0b1e12
This commit is contained in:
Benjamin Kramer 2021-02-15 05:01:36 -08:00 committed by TensorFlower Gardener
parent dc3d53c528
commit 011e64e79b
1 changed files with 10 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
tensorflow/compiler/xla/service/cpu/vector_support_library.cc
View File

@ -219,7 +219,8 @@ llvm::Value* VectorSupportLibrary::LoadVector(llvm::Value* pointer) {
pointer = b()->CreateBitCast(pointer, vector_pointer_type(), name());
}
return b()->CreateAlignedLoad(
pointer, ShapeUtil::ByteSizeOfPrimitiveType(primitive_type_), name());
pointer, llvm::Align(ShapeUtil::ByteSizeOfPrimitiveType(primitive_type_)),
name());
}
llvm::Value* VectorSupportLibrary::LoadScalar(llvm::Value* pointer) {
@ -227,7 +228,8 @@ llvm::Value* VectorSupportLibrary::LoadScalar(llvm::Value* pointer) {
pointer = b()->CreateBitCast(pointer, scalar_pointer_type(), name());
}
return b()->CreateAlignedLoad(
pointer, ShapeUtil::ByteSizeOfPrimitiveType(primitive_type_), name());
pointer, llvm::Align(ShapeUtil::ByteSizeOfPrimitiveType(primitive_type_)),
name());
}
void VectorSupportLibrary::StoreVector(llvm::Value* value,
@ -236,8 +238,9 @@ void VectorSupportLibrary::StoreVector(llvm::Value* value,
if (pointer->getType() != vector_pointer_type()) {
pointer = b()->CreateBitCast(pointer, vector_pointer_type());
}
b()->CreateAlignedStore(value, pointer,
ShapeUtil::ByteSizeOfPrimitiveType(primitive_type_));
b()->CreateAlignedStore(
value, pointer,
llvm::Align(ShapeUtil::ByteSizeOfPrimitiveType(primitive_type_)));
}
void VectorSupportLibrary::StoreScalar(llvm::Value* value,
@ -246,8 +249,9 @@ void VectorSupportLibrary::StoreScalar(llvm::Value* value,
if (pointer->getType() != scalar_pointer_type()) {
pointer = b()->CreateBitCast(pointer, scalar_pointer_type(), name());
}
b()->CreateAlignedStore(value, pointer,
ShapeUtil::ByteSizeOfPrimitiveType(primitive_type_));
b()->CreateAlignedStore(
value, pointer,
llvm::Align(ShapeUtil::ByteSizeOfPrimitiveType(primitive_type_)));
}
llvm::Value* VectorSupportLibrary::LoadBroadcast(llvm::Value* pointer) {