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

Use align and aliasing attribute name getters form LLVMDialect now that they are available.

Browse Source 
PiperOrigin-RevId: 342586230
Change-Id: I4ff8313baeeaa506f97fa78ebc3fb09d36a17936
This commit is contained in:
Stephan Herhut 2020-11-16 01:27:35 -08:00 committed by TensorFlower Gardener
parent 6b682a13d4
commit 34ef6a6805
1 changed files with 3 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
tensorflow/compiler/mlir/tools/kernel_gen/transforms/tensorflow_abi_knowledge_propagation.cc
View File

@ -37,10 +37,6 @@ namespace {
#define GEN_PASS_CLASSES
#include "tensorflow/compiler/mlir/tools/kernel_gen/transforms/kernel_gen_passes.h.inc"
// TODO(herhut): Move upstream to MLIR.
static constexpr StringRef kLLVMAlignAttrName = "llvm.align";
static constexpr StringRef kLLVMNoAliasAttrName = "llvm.noalias";
struct PropagateTfAbiKnowledgeToKernelsPass
: public PropagateTfAbiKnowledgeToKernelsBase<
PropagateTfAbiKnowledgeToKernelsPass> {
@ -116,7 +112,7 @@ struct PropagateTfAbiKnowledgeToKernelsPass
}
inner_stride.replaceAllUsesWith(one);
kernel.setArgAttr(
kernel_p + 1, kLLVMAlignAttrName,
kernel_p + 1, LLVM::LLVMDialect::getAlignAttrName(),
b.getIndexAttr(
tf_framework::TFFrameworkDialect::kAllocationAlignment));
}
@ -126,7 +122,8 @@ struct PropagateTfAbiKnowledgeToKernelsPass
// but we could use the alias analysis from buffer placement here
// to make sure.
// Add the no_alias attribute to the correspondign pointer.
kernel.setArgAttr(kernel_p + 1, kLLVMNoAliasAttrName,
kernel.setArgAttr(kernel_p + 1,
LLVM::LLVMDialect::getNoAliasAttrName(),
b.getBoolAttr(true));
}
// Advance base, aligned, offset, strides and sizes many arguments.