Update Slinky in XNNPACK

PiperOrigin-RevId: 891935408

Author: dsharletg

MODULE.bazel

@@ -115,9 +115,9 @@ http_archive(
 
 http_archive(
     name = "slinky",
-    sha256 = "28d311d7aa3c7f432e3fc14a09f11bff363a1611896c198d1c4a41ed5aab0d48",
-    strip_prefix = "slinky-4de79eb693dfa2791fa469586c8052287cb3110d",
+    sha256 = "3be8c5eb2e0540479ff87f5b0d771a8dddf2056d938aafeab0e47b6f4edaca9d",
+    strip_prefix = "slinky-42f643e511fc7310281055c270f229aaa87bf1f8",
     urls = [
-        "https://github.com/dsharlet/slinky/archive/4de79eb693dfa2791fa469586c8052287cb3110d.zip",
+        "https://github.com/dsharlet/slinky/archive/42f643e511fc7310281055c270f229aaa87bf1f8.zip",
     ],
 )

ynnpack/subgraph/copy.cc

@@ -149,7 +149,8 @@ slinky::func make_reshape(ynn_runtime& runtime,
     slinky::buffer<const void, YNN_MAX_TENSOR_RANK> input_as_output = output;
     input_as_output.raw_buffer::base = input.raw_buffer::base;
     for (size_t d = 0; d < input_as_output.rank; ++d) {
-      input_as_output.dim(d).set_bounds(0, input_as_output.dim(d).max());
+      input_as_output.mutable_dim(d).set_bounds(0,
+                                                input_as_output.dim(d).max());
     }
     slinky::copy(input_as_output, output);
     return 0;

ynnpack/subgraph/reduce.cc

@@ -60,8 +60,8 @@ void prepare_reduction_output(slinky::buffer<void, YNN_MAX_TENSOR_RANK>& output,
   if (op.keep_dims) {
     for (int d = 0; d < input.rank; ++d) {
       if (op.k_dims[d]) {
-        output.dim(d) = input.dim(d);
-        output.dim(d).set_stride(0);
+        output.mutable_dim(d) = input.dim(d);
+        output.mutable_dim(d).set_stride(0);
       }
     }
   } else {

ynnpack/subgraph/runtime.cc

@@ -480,10 +480,10 @@ auto make_reshape_impl(ynn_runtime* runtime) {
         assert(i.data->rank == i.rank());
         for (size_t d = 0; d < i.rank(); ++d) {
           if (i.extents[d].defined()) {
-            i.data->dim(d).set_min_extent(
+            i.data->mutable_dim(d).set_min_extent(
                 0, evaluate(i.extents[d], ctx));
           } else {
-            i.data->dim(d).set_min_extent(0, 1);
+            i.data->mutable_dim(d).set_min_extent(0, 1);
           }
         }
         ynn::init_buffer_strides(*i.data);
@@ -553,7 +553,7 @@ ynn_runtime::ynn_runtime(ynn::ref_count<const ynn_subgraph> subgraph,
       for (size_t d = 0; d < value.extents.size(); ++d) {
         if (!value.extents[d].defined() ||
             slinky::is_constant(value.extents[d], 1)) {
-          value.data->dim(d) = slinky::dim::broadcast();
+          value.data->mutable_dim(d) = slinky::dim::broadcast();
         }
       }
 
@@ -632,7 +632,7 @@ ynn_status ynn_runtime::build() {
       for (size_t d = 0; d < value.extents.size(); ++d) {
         if (!value.extents[d].defined() ||
             slinky::is_constant(value.extents[d], 1)) {
-          value.data->dim(d) = slinky::dim::broadcast();
+          value.data->mutable_dim(d) = slinky::dim::broadcast();
         }
       }
     }

ynnpack/subgraph/static_transpose.cc

@@ -44,7 +44,7 @@ auto make_transpose_impl(int rank, size_t elem_count,
     int fuse_transpose[YNN_MAX_TENSOR_RANK];
     int fuse_batch[YNN_MAX_TENSOR_RANK];
     for (int d = 0; d < rank; ++d) {
-      sliced_input.dim(d) = input.dim(permutation[d]);
+      sliced_input.mutable_dim(d) = input.dim(permutation[d]);
       fuse_batch[d] = input_dim0 == -1 ? d : YNN_MAX_TENSOR_RANK;
       if (permutation[d] == 0) {
         input_dim0 = d;

ynnpack/subgraph/subgraph.cc

@@ -200,7 +200,7 @@ ynn_status ynn_value::set_external_shape(size_t rank, const size_t* dims) {
 
   for (size_t d = 0; d < rank; ++d) {
     if (!extents[d].defined() || slinky::is_constant(extents[d], 1)) {
-      data->dim(d) = slinky::dim::broadcast();
+      data->mutable_dim(d) = slinky::dim::broadcast();
     }
   }
 

ynnpack/subgraph/tensor.cc

@@ -45,7 +45,7 @@ void init_buffer_strides(slinky::raw_buffer& buffer) {
   slinky::index_t stride = buffer.elem_size;
   for (size_t i = 0; i < buffer.rank; ++i) {
     if (buffer.dim(i) != slinky::dim::broadcast()) {
-      buffer.dim(i).set_stride(stride);
+      buffer.mutable_dim(i).set_stride(stride);
       stride *= buffer.dim(i).extent();
     }
   }
@@ -60,8 +60,8 @@ void init_buffer(slinky::raw_buffer& buffer, size_t elem_size, size_t num_dims,
   buffer.base = const_cast<void*>(data);
   if (dims) {
     for (size_t i = 0; i < num_dims; ++i) {
-      buffer.dim(i).set_min_extent(0, dims[num_dims - i - 1]);
-      buffer.dim(i).set_fold_factor(slinky::dim::unfolded);
+      buffer.mutable_dim(i).set_min_extent(0, dims[num_dims - i - 1]);
+      buffer.mutable_dim(i).set_fold_factor(slinky::dim::unfolded);
     }
     init_buffer_strides(buffer);
   }
@@ -162,7 +162,7 @@ ynn_status ynn_define_tensor_value(ynn_subgraph_t subgraph, enum ynn_type type,
   for (size_t d = 0; d < value->extents.size(); ++d) {
     if (!value->extents[d].defined() ||
         slinky::is_constant(value->extents[d], 1)) {
-      value->data->dim(d) = slinky::dim::broadcast();
+      value->data->mutable_dim(d) = slinky::dim::broadcast();
     }
   }