ai-edge-torch-nightly 0.3.0.dev20240908__py3-none-any.whl → 0.3.0.dev20240910__py3-none-any.whl

ai_edge_torch/_convert/test/test_convert.py

@@ -25,6 +25,7 @@ from ai_edge_torch.testing import model_coverage
 import numpy as np
 import tensorflow as tf
 import torch
+from torch import nn
 import torchvision
 
 from absl.testing import absltest as googletest
@@ -51,7 +52,7 @@ class TestConvert(googletest.TestCase):
   def test_convert_add(self):
     """Tests conversion of a simple Add module."""
 
-    class Add(torch.nn.Module):
+    class Add(nn.Module):
 
       def forward(self, a, b):
         return a + b
@@ -70,7 +71,7 @@ class TestConvert(googletest.TestCase):
   def test_convert_dot_add(self):
     """Tests conversion of a matrix multiplication followed by an add."""
 
-    class DotAdd(torch.nn.Module):
+    class DotAdd(nn.Module):
 
       def forward(self, a, b, c):
         return a @ b + c
@@ -99,7 +100,7 @@ class TestConvert(googletest.TestCase):
   def test_signature_args_ordering(self):
     """Tests conversion of a model with more than 10 arguments."""
 
-    class AddChainWith11Args(torch.nn.Module):
+    class AddChainWith11Args(nn.Module):
       """A model with 11 arguments."""
 
       def forward(
@@ -152,7 +153,7 @@ class TestConvert(googletest.TestCase):
   def test_multi_output_model(self):
     """Tests conversion of a model that returns multiple outputs."""
 
-    class BasicAddModelWithMultipleOutputs(torch.nn.Module):
+    class BasicAddModelWithMultipleOutputs(nn.Module):
       """A model that returns multiple outputs."""
 
       def forward(self, arg0, arg1):
@@ -176,7 +177,7 @@ class TestConvert(googletest.TestCase):
   def test_12_outputs_model(self):
     """Tests conversion of a model that returns more than 10 outputs."""
 
-    class BasicAddModelWithMultipleOutputs(torch.nn.Module):
+    class BasicAddModelWithMultipleOutputs(nn.Module):
       """A model that returns multiple outputs."""
 
       def forward(self, arg0, arg1):
@@ -245,7 +246,7 @@ class TestConvert(googletest.TestCase):
   def test_convert_add_converter_flags(self):
     """Tests conversion of an add module setting a tflite converter flag."""
 
-    class Add(torch.nn.Module):
+    class Add(nn.Module):
 
       def forward(self, a, b):
         return a + b
@@ -267,6 +268,27 @@ class TestConvert(googletest.TestCase):
     )
     self.assertTrue(os.path.isdir(ir_dump_path))
 
+  def test_convert_conv_transpose_batch_norm(self):
+    """Tests conversion of a model with ConvTranspose2d and BatchNorm2d."""
+
+    channels = 2
+    size = 2
+    torch_model = nn.Sequential(
+        nn.ConvTranspose2d(
+            channels, channels, 1, stride=2, dilation=1, bias=False
+        ),
+        nn.BatchNorm2d(channels),
+    )
+
+    torch_model.eval()
+    sample_input = (torch.rand(1, channels, size, size),)
+    edge_model = ai_edge_torch.convert(torch_model, sample_input)
+
+    result = model_coverage.compare_tflite_torch(
+        edge_model, torch_model, sample_input
+    )
+    self.assertTrue(result)
+
   @googletest.skipIf(
       not config.Config.use_torch_xla,
       reason="Shape polymorphism is not yet support with odml_torch.",
@@ -274,7 +296,7 @@ class TestConvert(googletest.TestCase):
   def test_convert_model_with_dynamic_batch(self):
     """Test converting a simple model with dynamic batch size."""
 
-    class SampleModel(torch.nn.Module):
+    class SampleModel(nn.Module):
 
       def __init__(self):
         super().__init__()
@@ -304,7 +326,7 @@ class TestConvert(googletest.TestCase):
   def test_convert_model_with_kwargs(self):
     """Test converting a simple model with sample_kwargs."""
 
-    class SampleModel(torch.nn.Module):
+    class SampleModel(nn.Module):
 
       def forward(self, x, y):
         return x + y
@@ -323,7 +345,7 @@ class TestConvert(googletest.TestCase):
   def test_convert_model_with_args_kwargs(self):
     """Test converting a simple model with both sample_args and sample_kwargs."""
 
-    class SampleModel(torch.nn.Module):
+    class SampleModel(nn.Module):
 
       def forward(self, x, y):
         return x + y
@@ -343,7 +365,7 @@ class TestConvert(googletest.TestCase):
   def test_convert_model_with_args_nested_kwargs_1(self):
     """Test converting a simple model with both sample_args and nested sample_kwargs."""
 
-    class SampleModel(torch.nn.Module):
+    class SampleModel(nn.Module):
 
       def forward(self, x: torch.Tensor, y: torch.Tensor, z: TestContainer1):
         return x + y + z.data_1 + z.data_2[0] + z.data_2[1]
@@ -370,7 +392,7 @@ class TestConvert(googletest.TestCase):
   def test_convert_model_with_args_nested_kwargs_2(self):
     """Test converting a simple model with both sample_args and nested sample_kwargs."""
 
-    class SampleModel(torch.nn.Module):
+    class SampleModel(nn.Module):
 
       def forward(self, x, y, z):
         return x + y + z.data_1 + z.data_2[0][0] + z.data_2[1]
@@ -397,7 +419,7 @@ class TestConvert(googletest.TestCase):
   def test_convert_model_with_args_nested_kwargs_3(self):
     """Test converting a simple model with both sample_args and nested sample_kwargs."""
 
-    class SampleModel(torch.nn.Module):
+    class SampleModel(nn.Module):
 
       def forward(self, x, y, z):
         return x + y + z.data_1 + z.data_2[0]["foo"] + z.data_2[1]
@@ -424,7 +446,7 @@ class TestConvert(googletest.TestCase):
   def test_convert_model_non_flat_output_dict(self):
     """Test converting a model with non-flat output structure."""
 
-    class SampleModel(torch.nn.Module):
+    class SampleModel(nn.Module):
 
       def forward(self, x, y, z):
         return {"x": x, "y": TestContainer1(data_1=y, data_2=[y, z])}

ai_edge_torch/odml_torch/lowerings/_convolution.py

@@ -12,22 +12,171 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 # ==============================================================================
-"""Provides lowering for coreaten to mlir stablehlo op: Convolution"""
+"""Provides lowering for coreaten to stablehlo for Convolution."""
 
 import math
 from typing import Optional
 
+from ai_edge_torch.odml_torch.lowerings import registry
 from jax._src.lib.mlir import ir
 from jax._src.lib.mlir.dialects import hlo as stablehlo
 import torch
 
-from .registry import lower
+
+def make_padding(padding):
+  """Change the padding from pytorch to stablehlo style.
+
+  Stablehlo allows start and end padding for each dimension while aten only
+  allows symmetric padding and so only has one number per dimension.
+
+  Args:
+    padding: The padding of the convolution
+
+  Returns:
+    The padding in stablehlo style
+  """
+  return tuple((p, p) for p in padding)
+
+
+def create_conv_dimension_numbers(lhs, transposed: bool = False):
+  """Create the dimension numbers for the convolution.
+
+  Args:
+    lhs: The input tensor
+    transposed: Whether the convolution is transposed
+
+  Returns:
+    The dimension numbers for the convolution
+  """
+  num_spatial_dims = len(lhs.type.shape) - 2
+  spatial_dimensions = []
+  for i in range(0, num_spatial_dims):
+    spatial_dimensions.append(i + 2)
+
+  # Regular kernels are OIHW
+  # TransposedConv kernels are IOHW
+  dimension_numbers = stablehlo.ConvDimensionNumbers.get(
+      input_batch_dimension=0,
+      input_feature_dimension=1,
+      input_spatial_dimensions=spatial_dimensions,
+      kernel_input_feature_dimension=0 if transposed else 1,
+      kernel_output_feature_dimension=1 if transposed else 0,
+      kernel_spatial_dimensions=spatial_dimensions,
+      output_batch_dimension=0,
+      output_feature_dimension=1,
+      output_spatial_dimensions=spatial_dimensions,
+  )
+  return dimension_numbers
+
+
+def infer_output_shape(
+    lhs,
+    rhs,
+    stride,
+    dilation,
+    padding,
+    transposed: bool = False,
+    output_padding: list[int] = 0,
+):
+  """Infer the output shape of the convolution.
+
+  Args:
+    lhs: The input tensor
+    rhs: The kernel tensor
+    stride: The stride of the convolution (dilation of input in transposed conv)
+    dilation: The kernel dilation of the convolution
+    padding: The padding of the convolution
+    transposed: Whether the convolution is transposed
+    output_padding: The output padding of the convolution
+
+  Returns:
+    The output shape of the convolution
+  """
+  lhs_type: ir.RankedTensorType = lhs.type
+  lhs_shape: list[int] = lhs_type.shape
+  rhs_shape: list[int] = rhs.type.shape
+
+  # Input layout is: (N)CHW and Kernel layout is: (O)IHW for regular conv
+  # Input layout is: (N)CHW and Kernel layout is: I(O)HW for transposed conv
+  output_shape = (
+      [lhs_shape[0], rhs_shape[1]]
+      if transposed
+      else [lhs_shape[0], rhs_shape[0]]
+  )
+  num_spatial_dims = len(lhs.type.shape) - 2
+
+  # looping over the spatial dims (skipping the first 2 dims which are
+  # batch and features)
+  for spatial_dim in range(0, num_spatial_dims):
+    dim = spatial_dim + 2
+    dim_size = lhs_shape[dim]
+    kernel_dim_size = rhs_shape[dim]
+
+    if transposed:
+      output_dim_size = (
+          (dim_size - 1) * stride[spatial_dim]
+          - 2 * padding[spatial_dim]
+          + dilation[spatial_dim] * (kernel_dim_size - 1)
+          + output_padding[spatial_dim]
+          + 1
+      )
+    else:
+      output_dim_size = math.floor(
+          (
+              (
+                  dim_size
+                  + 2 * padding[spatial_dim]
+                  - dilation[spatial_dim] * (kernel_dim_size - 1)
+                  - 1
+              )
+              / stride[spatial_dim]
+          )
+          + 1
+      )
+
+    output_shape.append(output_dim_size)
+
+  return output_shape
+
+
+def build_transpose_conv(
+    lctx,
+    output_type: ir.RankedTensorType,
+    lhs: ir.Value,
+    rhs: ir.Value,
+    stride: list[int],
+    padding: list[int],
+    dilation: list[int],
+    output_padding: list[int],
+    groups: int,
+):
+  lhs_type: ir.RankedTensorType = lhs.type
+  num_spatial_dims = len(lhs_type.shape) - 2
+  rhs = stablehlo.reverse(rhs, list(range(2, 2 + num_spatial_dims)))
+
+  kernel_size = rhs.type.shape
+  # We need to additional padding on the input to get the right output size.
+  adjusted_padding = [
+      dilation[dim] * (kernel_size[dim + 2] - 1) - padding[dim]
+      for dim in range(num_spatial_dims)
+  ]
+  return stablehlo.convolution(
+      result=output_type,
+      lhs=lhs,
+      rhs=rhs,
+      dimension_numbers=create_conv_dimension_numbers(lhs, True),
+      feature_group_count=groups,
+      batch_group_count=1,
+      padding=make_padding(adjusted_padding),
+      lhs_dilation=stride,
+      rhs_dilation=dilation,
+  )
 
 
 # convolution(Tensor input, Tensor weight, Tensor? bias, SymInt[] stride,
 #   SymInt[] padding, SymInt[] dilation, bool transposed,
 #   SymInt[] output_padding, SymInt groups) -> Tensor
-# @lower(torch.ops.aten.convolution)
+@registry.lower(torch.ops.aten.convolution)
 def _aten_convolution(
     lctx,
     lhs: ir.Value,
@@ -40,80 +189,53 @@ def _aten_convolution(
     output_padding: list[int],
     groups: int,
 ):
-  if transposed:
-    raise NotImplementedError("Transposed convolution is not implemented.")
 
-  if bias is not None:
-    raise NotImplementedError("Bias on convolution is not implemented.")
-
-  # Stablehlo allows start and end padding for each dimension while aten only
-  # allows symmetric padding and so only has one number per dimension.
-  def make_padding(padding):
-    return tuple((p, p) for p in padding)
-
-  def create_conv_dimension_numbers():
-    num_spatial_dims = len(lhs.type.shape) - 2
-    spatial_dimensions = []
-    for i in range(0, num_spatial_dims):
-      spatial_dimensions.append(i + 2)
-
-    dimension_numbers = stablehlo.ConvDimensionNumbers.get(
-        input_batch_dimension=0,
-        input_feature_dimension=1,
-        input_spatial_dimensions=spatial_dimensions,
-        kernel_input_feature_dimension=1,
-        kernel_output_feature_dimension=0,
-        kernel_spatial_dimensions=spatial_dimensions,
-        output_batch_dimension=0,
-        output_feature_dimension=1,
-        output_spatial_dimensions=spatial_dimensions,
+  # TODO(b/365559296) Add support for output_padding
+  if any(output_padding):
+    raise NotImplementedError(
+        "Output padding on convolution is not implemented."
     )
-    return dimension_numbers
-
-  def infer_output_shape():
-    lhs_type: ir.RankedTensorType = lhs.type
-    lhs_shape: list[int] = lhs_type.shape
-    rhs_shape: list[int] = rhs.type.shape
-
-    # Input layout is: (N)CHW and Kernel layout is: (O)IHW
-    output_shape = [lhs_shape[0], rhs_shape[0]]
-    num_spatial_dims = len(lhs.type.shape) - 2
-
-    # looping over the spatial dims (skipping the first 2 dims which are
-    # batch and features)
-    for spatial_dim in range(0, num_spatial_dims):
-      dim_size = lhs_shape[spatial_dim + 2]
-      kernel_dim_size = rhs_shape[spatial_dim + 2]
-
-      # for example, a dilation of 2 increases the dimension size by 2
-      dim_size *= dilation[spatial_dim]
-
-      # padding added to both sides
-      dim_size += 2 * padding[spatial_dim]
-
-      output_dim_size = math.ceil(
-          (dim_size - kernel_dim_size + 1) / stride[spatial_dim]
-      )
-
-      output_shape.append(output_dim_size)
-
-    return output_shape
 
   lhs_type: ir.RankedTensorType = lhs.type
-
-  op = stablehlo.ConvolutionOp(
-      result=ir.RankedTensorType.get(
-          infer_output_shape(), lhs_type.element_type
-      ),
-      lhs=lhs,
-      rhs=rhs,
-      dimension_numbers=create_conv_dimension_numbers(),
-      feature_group_count=groups,
-      batch_group_count=1,
-      window_strides=stride,
-      padding=make_padding(padding),
-      lhs_dilation=(1,) * len(stride),
-      rhs_dilation=dilation,
+  output_shape = infer_output_shape(
+      lhs, rhs, stride, dilation, padding, transposed, output_padding
+  )
+  output_type = ir.RankedTensorType.get(
+      output_shape,
+      lhs_type.element_type,
   )
 
-  return op.result
+  if transposed:
+    res = build_transpose_conv(
+        lctx,
+        output_type,
+        lhs,
+        rhs,
+        stride,
+        padding,
+        dilation,
+        output_padding,
+        groups,
+    )
+  else:
+    res = stablehlo.convolution(
+        result=output_type,
+        lhs=lhs,
+        rhs=rhs,
+        dimension_numbers=create_conv_dimension_numbers(lhs),
+        feature_group_count=groups,
+        batch_group_count=1,
+        window_strides=stride,
+        padding=make_padding(padding),
+        rhs_dilation=dilation,
+    )
+
+  if bias is not None:
+    # broadcast [C] to [NCHW]
+    broadcasted_bias = stablehlo.broadcast_in_dim(output_type, bias, [1])
+    res = stablehlo.add(
+        lhs=res,
+        rhs=broadcasted_bias,
+    )
+
+  return res

ai_edge_torch/odml_torch/lowerings/_jax_lowerings.py

@@ -105,7 +105,6 @@ lower_by_torch_xla2(torch.ops.aten.clamp.default)
 lower_by_torch_xla2(torch.ops.aten.clone)
 lower_by_torch_xla2(torch.ops.aten.clone.default)
 lower_by_torch_xla2(torch.ops.aten.constant_pad_nd)
-lower_by_torch_xla2(torch.ops.aten.convolution)
 lower_by_torch_xla2(torch.ops.aten.cos)
 lower_by_torch_xla2(torch.ops.aten.cosh)
 lower_by_torch_xla2(torch.ops.aten.cumsum)

ai_edge_torch/version.py

@@ -13,4 +13,4 @@
 # limitations under the License.
 # ==============================================================================
 
-__version__ = "0.3.0.dev20240908"
+__version__ = "0.3.0.dev20240910"

{ai_edge_torch_nightly-0.3.0.dev20240908.dist-info → ai_edge_torch_nightly-0.3.0.dev20240910.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ai-edge-torch-nightly
-Version: 0.3.0.dev20240908
+Version: 0.3.0.dev20240910
 Summary: Supporting PyTorch models with the Google AI Edge TFLite runtime.
 Home-page: https://github.com/google-ai-edge/ai-edge-torch
 Keywords: On-Device ML,AI,Google,TFLite,PyTorch,LLMs,GenAI

{ai_edge_torch_nightly-0.3.0.dev20240908.dist-info → ai_edge_torch_nightly-0.3.0.dev20240910.dist-info}/RECORD

@@ -2,7 +2,7 @@ ai_edge_torch/__init__.py,sha256=48qP37uHT90YPs4eIUQxCiWVwqGEX3idCUs6mQKvX1U,116
 ai_edge_torch/config.py,sha256=PCd9PVrbUNeVIUDFUCnW4goDWU4bjouK28yMYU6VOi0,877
 ai_edge_torch/conftest.py,sha256=r0GTrhMRhlmOGrrkvumHN8hkmyug6WvF60vWq8wRIBI,758
 ai_edge_torch/model.py,sha256=NYV6Mkaje_ditIEI_s_7nLP_-8i4kbGM8nRzieVkbUI,5397
-ai_edge_torch/version.py,sha256=gyyuDD-i83EGkYIzPuqIjPUIC8huFW09RDInbOSOx1c,706
+ai_edge_torch/version.py,sha256=e4sh_RFYgNHGoVuOeICnFZtLu1MQCNv7qpq94nKFarU,706
 ai_edge_torch/_convert/__init__.py,sha256=hHLluseD2R0Hh4W6XZRIXY_dRQeYudjsrKGf6LZz65g,671
 ai_edge_torch/_convert/conversion.py,sha256=kcv_QgNgeyDmrqwdzHicGNP68w6zF7GJg7YkMEIXp4Q,3759
 ai_edge_torch/_convert/conversion_utils.py,sha256=Sr8qXVcTwc-ZnZmK7yxVrIOOp1S_vNrwzC0zUvLTI2o,2160
@@ -26,7 +26,7 @@ ai_edge_torch/_convert/fx_passes/optimize_layout_transposes_pass/layout_partitio
 ai_edge_torch/_convert/fx_passes/optimize_layout_transposes_pass/layout_partitioners/greedy.py,sha256=L_x8BrF7UDah-SYl-pG11I6CIckdU9kBTUHcmwW4cts,2420
 ai_edge_torch/_convert/fx_passes/optimize_layout_transposes_pass/layout_partitioners/min_cut.py,sha256=mzfL9cf0qBnpmxM_OlMQFvQsEZV2B_Mia9yEJV4J7rI,7135
 ai_edge_torch/_convert/test/__init__.py,sha256=hHLluseD2R0Hh4W6XZRIXY_dRQeYudjsrKGf6LZz65g,671
-ai_edge_torch/_convert/test/test_convert.py,sha256=pUYSXuqFg8CAeJ8JkoYf7S0RDLRPVuZUwVOd0xObM6w,14411
+ai_edge_torch/_convert/test/test_convert.py,sha256=FSufFZEeTLBpUnzE1Iy-LvNN0mhDynWMNg7Mei8RpLQ,14973
 ai_edge_torch/_convert/test/test_convert_composites.py,sha256=BCIODgxMI_3MxMLfNWYMGjcz-al-J3z5eDHCiZJXNwY,7992
 ai_edge_torch/_convert/test/test_convert_multisig.py,sha256=6_C2R9--KyNR7_oezZIAfyTSR97tOeEWy4XGcbSxBDE,5778
 ai_edge_torch/_convert/test/test_to_channel_last_io.py,sha256=1o-gUiwzIuO67FNAJ8DeyKv8fVUeZVNNNwofNVDjYeU,3024
@@ -148,8 +148,8 @@ ai_edge_torch/odml_torch/jax_bridge/utils.py,sha256=T8isGc896VrHZ6c_L5pYmLpolQ7i
 ai_edge_torch/odml_torch/lowerings/__init__.py,sha256=GqYk6oBJw7KWeG4_6gxSu_OvYhjJcC2FpGzWPPEdH6w,933
 ai_edge_torch/odml_torch/lowerings/_basic.py,sha256=wV8AUK8dvjLUy3qjqw_IxpiYVDWUMPNZRfi3XYE_hDs,6972
 ai_edge_torch/odml_torch/lowerings/_batch_norm.py,sha256=PaLI0BB6pdBW1VyfW8VTOT_Be-ZcqYdNOsyfzKfq8Cg,2064
-ai_edge_torch/odml_torch/lowerings/_convolution.py,sha256=B6BILeu-UlwGB1O6g7111X1TaIFznsfxXrB72ygBsBA,3885
-ai_edge_torch/odml_torch/lowerings/_jax_lowerings.py,sha256=I0Y4IK7Zap8m6xfxMw7DfQ9Mg4htKOoypdHVAMHqx9c,10669
+ai_edge_torch/odml_torch/lowerings/_convolution.py,sha256=v1VdKmL8YLJv3PR9VgyNghO83A25PpTzY2ZUAJqlq3Q,6847
+ai_edge_torch/odml_torch/lowerings/_jax_lowerings.py,sha256=s-cT_tIQHu7w5hXl8MCixRxLlHplpXW-UCzHT9TY--o,10621
 ai_edge_torch/odml_torch/lowerings/context.py,sha256=jslcCv7r_HtImSRTxJwHAUV_QCu9Jub51lovmoBkmFA,1295
 ai_edge_torch/odml_torch/lowerings/registry.py,sha256=ES3x_RJ22T5rlmMrlomex2DdcZbhlyVJ7_HS3rjz3Uk,2851
 ai_edge_torch/odml_torch/lowerings/utils.py,sha256=NczqpsSd3Fn7yVcPC3qllemiZxxDAZgcW1T5l8-W9fE,5593
@@ -161,8 +161,8 @@ ai_edge_torch/quantize/quant_config.py,sha256=U0KisSW-uZkoMJcy-ZP9W57p3tsa594fr9
 ai_edge_torch/testing/__init__.py,sha256=hHLluseD2R0Hh4W6XZRIXY_dRQeYudjsrKGf6LZz65g,671
 ai_edge_torch/testing/model_coverage/__init__.py,sha256=5P8J6Zk5YYtDvTBucFvB9NGSRI7Gw_24WnrbhXgycEE,765
 ai_edge_torch/testing/model_coverage/model_coverage.py,sha256=UPB448aMDUyC0HNYVqio2rcJPnDN0tBQMP08J6vPYew,4718
-ai_edge_torch_nightly-0.3.0.dev20240908.dist-info/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
-ai_edge_torch_nightly-0.3.0.dev20240908.dist-info/METADATA,sha256=kDesFhNNQZ4QCG572z3cDo3eyiSnnFTUMwzKOmdsrGo,1859
-ai_edge_torch_nightly-0.3.0.dev20240908.dist-info/WHEEL,sha256=eOLhNAGa2EW3wWl_TU484h7q1UNgy0JXjjoqKoxAAQc,92
-ai_edge_torch_nightly-0.3.0.dev20240908.dist-info/top_level.txt,sha256=5KXRaF2hwkApYxf7Y8y_tVb9aulGTlbOoNdbx1aKRkE,14
-ai_edge_torch_nightly-0.3.0.dev20240908.dist-info/RECORD,,
+ai_edge_torch_nightly-0.3.0.dev20240910.dist-info/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
+ai_edge_torch_nightly-0.3.0.dev20240910.dist-info/METADATA,sha256=WFNExTO6eF-tAWPmDdQDlr9dvplcoNB0uPdVxSNXYHk,1859
+ai_edge_torch_nightly-0.3.0.dev20240910.dist-info/WHEEL,sha256=eOLhNAGa2EW3wWl_TU484h7q1UNgy0JXjjoqKoxAAQc,92
+ai_edge_torch_nightly-0.3.0.dev20240910.dist-info/top_level.txt,sha256=5KXRaF2hwkApYxf7Y8y_tVb9aulGTlbOoNdbx1aKRkE,14
+ai_edge_torch_nightly-0.3.0.dev20240910.dist-info/RECORD,,