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

Use mlir::OpState::operator->() to get to methods of mlir::Operation.

Browse Source 
This is a preparation step to remove those methods from OpState.

PiperOrigin-RevId: 348434761
Change-Id: I975647b30b42722b9af870f024efe8d05199dd1e
This commit is contained in:
Christian Sigg 2020-12-21 02:11:17 -08:00 committed by TensorFlower Gardener
parent 3324e3d6be
commit 5ce6ac8cb9
2 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tensorflow/compiler/xla/service/mlir_gpu/mlir_compiler_impl.cc
View File

@ -300,9 +300,9 @@ Status InsertBufferLoadPreduleIntoKernel(
auto num_original_args = kernel.getNumArguments();
std::vector<LLVMType> new_arg_types(buffers.size(), ptr_type);
kernel.setAttr(kernel.getTypeAttrName(),
mlir::TypeAttr::get(LLVMType::getFunctionTy(
void_type, new_arg_types, /*isVarArg=*/false)));
kernel->setAttr(kernel.getTypeAttrName(),
mlir::TypeAttr::get(LLVMType::getFunctionTy(
void_type, new_arg_types, /*isVarArg=*/false)));
std::vector<Value> original_args(kernel.args_begin(), kernel.args_end());
std::vector<mlir::Type> as_mlir_types(new_arg_types.begin(),

4
tensorflow/compiler/xla/service/mlir_gpu/passes.cc
View File

@ -257,8 +257,8 @@ struct RewriteKernelSignaturePass
auto new_kernel = kernel_builder.create<mlir::gpu::GPUFuncOp>(
kernel.getLoc(), kernel.getName(),
kernel_builder.getFunctionType(operand_types, {}));
new_kernel.setAttr(mlir::gpu::GPUDialect::getKernelFuncAttrName(),
kernel_builder.getUnitAttr());
new_kernel->setAttr(mlir::gpu::GPUDialect::getKernelFuncAttrName(),
kernel_builder.getUnitAttr());
// Create a map from old kernel argument to new one.
mlir::BlockAndValueMapping old_kernel_to_new;