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

Lower all StandardOps to LLVMIR in kernel_gen.

Browse Source 
PiperOrigin-RevId: 340702811
Change-Id: I525313971f23516454273f944c82f2a74ff01289
This commit is contained in:
Tres Popp 2020-11-04 12:13:34 -08:00 committed by TensorFlower Gardener
parent 5adba3f435
commit cd0ef5f0a2
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tensorflow/compiler/mlir/tools/kernel_gen/transforms/tf_kernel_to_llvm_pass.cc
View File

@ -56,8 +56,8 @@ class TFKernelToLLVMPass : public TFKernelToLLVMPassBase<TFKernelToLLVMPass> {
// Set target.
ConversionTarget target(getContext());
target.addLegalDialect<LLVM::LLVMDialect>();
target
.addIllegalDialect<gpu::GPUDialect, tf_framework::TFFrameworkDialect>();
target.addIllegalDialect<gpu::GPUDialect, StandardOpsDialect,
tf_framework::TFFrameworkDialect>();
target.addIllegalOp<LLVM::DialectCastOp>();
if (failed(applyPartialConversion(m, target, std::move(patterns)))) {