diff options
Diffstat (limited to 'mlir/lib/Dialect/NVGPU/TransformOps/NVGPUTransformOps.cpp')
| -rw-r--r-- | mlir/lib/Dialect/NVGPU/TransformOps/NVGPUTransformOps.cpp | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/mlir/lib/Dialect/NVGPU/TransformOps/NVGPUTransformOps.cpp b/mlir/lib/Dialect/NVGPU/TransformOps/NVGPUTransformOps.cpp index 75dbe0becf80..1046f5798ecd 100644 --- a/mlir/lib/Dialect/NVGPU/TransformOps/NVGPUTransformOps.cpp +++ b/mlir/lib/Dialect/NVGPU/TransformOps/NVGPUTransformOps.cpp @@ -290,8 +290,10 @@ static void getPipelineStages( }); options.inclusive = true; for (Operation &op : forOp.getBody()->getOperations()) { - if (stage0Ops.contains(&op)) - getBackwardSlice(&op, &dependencies, options); + if (stage0Ops.contains(&op)) { + LogicalResult result = getBackwardSlice(&op, &dependencies, options); + assert(result.succeeded() && "expected a backward slice"); + } } for (Operation &op : forOp.getBody()->getOperations()) { |