Skip to content

Commit 0d5d355

Browse files
Update legalizations for LowerGpuOpsToROCDLOps (llvm#108266)
LLVM::FAbsOp and LLVM::SqrtOp are legal after llvm#102971
1 parent 650d852 commit 0d5d355

File tree

1 file changed

+3
-4
lines changed

1 file changed

+3
-4
lines changed

mlir/lib/Conversion/GPUToROCDL/LowerGpuOpsToROCDLOps.cpp

+3-4
Original file line numberDiff line numberDiff line change
@@ -329,10 +329,9 @@ void mlir::configureGpuToROCDLConversionLegality(ConversionTarget &target) {
329329
target.addLegalDialect<::mlir::LLVM::LLVMDialect>();
330330
target.addLegalDialect<ROCDL::ROCDLDialect>();
331331
target.addIllegalDialect<gpu::GPUDialect>();
332-
target.addIllegalOp<LLVM::CosOp, LLVM::ExpOp, LLVM::Exp2Op, LLVM::FAbsOp,
333-
LLVM::FCeilOp, LLVM::FFloorOp, LLVM::FRemOp, LLVM::LogOp,
334-
LLVM::Log10Op, LLVM::Log2Op, LLVM::PowOp, LLVM::SinOp,
335-
LLVM::SqrtOp>();
332+
target.addIllegalOp<LLVM::CosOp, LLVM::ExpOp, LLVM::Exp2Op, LLVM::FCeilOp,
333+
LLVM::FFloorOp, LLVM::FRemOp, LLVM::LogOp, LLVM::Log10Op,
334+
LLVM::Log2Op, LLVM::PowOp, LLVM::SinOp>();
336335

337336
// TODO: Remove once we support replacing non-root ops.
338337
target.addLegalOp<gpu::YieldOp, gpu::GPUModuleOp>();

0 commit comments

Comments
 (0)