add better error message for tensor.extract/insert_slice op in layout system

Part of #2028

cc @kragall

PiperOrigin-RevId: 797461798

Author: asraa

lib/Transforms/LayoutPropagation/NewLayoutPropagation.cpp

@@ -234,6 +234,12 @@ LogicalResult NewLayoutPropagation::visitOperation(Operation* op) {
       .Case<affine::AffineForOp>([&](auto op) { return visitOperation(op); })
       // tensor ops
       .Case<tensor::ExtractOp>([&](auto op) { return visitOperation(op); })
+      .Case<tensor::ExtractSliceOp, tensor::InsertSliceOp>([&](auto op) {
+        // TODO(#2028): Support tensor.extract_slice and tensor.insert_slice in
+        // layout.
+        return op->emitError()
+               << "Layout propagation not supported for this op";
+      })
       // tensor ops
       .Case<CollapseShapeOp, ExpandShapeOp>(
           [&](auto op) { return visitOperation(op); })

tests/Transforms/new_layout_propagation/slice.mlir

@@ -0,0 +1,15 @@
+// RUN: heir-opt --new-layout-propagation %s -verify-diagnostics
+
+module {
+  func.func @test(%arg0: !secret.secret<tensor<16xi8>>) -> !secret.secret<tensor<11x16xi8>> {
+    %0 = tensor.empty() : tensor<11x16xi8>
+    %1 = secret.generic(%arg0: !secret.secret<tensor<16xi8>>) {
+    ^body(%input0: tensor<16xi8>):
+    // expected-error @+1 {{Layout propagation not supported for this op}}
+      %extracted_slice = tensor.extract_slice %input0[0] [4] [1] : tensor<16xi8> to tensor<4xi8>
+      %inserted_slice = tensor.insert_slice %extracted_slice into %0[0, 0] [1, 4] [1, 1] : tensor<4xi8> into tensor<11x16xi8>
+      secret.yield %inserted_slice : tensor<11x16xi8>
+    } -> !secret.secret<tensor<11x16xi8>>
+    return %1 : !secret.secret<tensor<11x16xi8>>
+  }
+}