diff --git a/mlir/lib/Conversion/SCFToGPU/SCFToGPUPass.cpp b/mlir/lib/Conversion/SCFToGPU/SCFToGPUPass.cpp index 66fbacf69d5b..15075b53ab90 100644 --- a/mlir/lib/Conversion/SCFToGPU/SCFToGPUPass.cpp +++ b/mlir/lib/Conversion/SCFToGPU/SCFToGPUPass.cpp @@ -50,9 +50,7 @@ struct ParallelLoopToGpuPass OwningRewritePatternList patterns; populateParallelLoopToGPUPatterns(patterns, &getContext()); ConversionTarget target(getContext()); - target.addLegalDialect<AffineDialect, complex::ComplexDialect, - gpu::GPUDialect, scf::SCFDialect, - StandardOpsDialect>(); + target.markUnknownOpDynamicallyLegal([](Operation *) { return true; }); configureParallelLoopToGPULegality(target); if (failed(applyPartialConversion(getOperation(), target, std::move(patterns))))