Update on "[xnnpack][lite-int] Handle Constant Data"

mcr229 · mcr229 · commit 60f0edfd2bb3 · 2022-11-21T10:58:05.000-08:00
Handling constant data for xnnpack delegation. This allows us to handle new modules like such: ``` class Module(torch.nn.Module): def __init__(self): super().__init__() self._constant = torch.ones(4, 4, 4) def forward(self, x): return x + self._constant ``` this is the precursor work to handling convolution, as we need to serialize constant data(weights) Differential Revision: [D41050349](https://our.internmc.facebook.com/intern/diff/D41050349/) [ghstack-poisoned]
diff --git a/torch/csrc/jit/backends/xnnpack/compiler/xnn_compiler.cpp b/torch/csrc/jit/backends/xnnpack/compiler/xnn_compiler.cpp
@@ -51,8 +51,8 @@ void XNNCompiler::compileModel(
         const auto& constant_buffer = *flatbuffer_graph->constant_buffer();
         auto buffer_idx = tensor_value->constant_buffer_idx();
         const auto buffer_ptr = buffer_idx == 0
-                ? nullptr
-                : constant_buffer[buffer_idx]->storage()->data();
+            ? nullptr
+            : constant_buffer[buffer_idx]->storage()->data();
         status = xnn_define_tensor_value(
             /*subgraph=*/subgraph_ptr,
             /*datatype=*/xnn_datatype_fp32,
diff --git a/torch/csrc/jit/backends/xnnpack/serialization/serializer.h b/torch/csrc/jit/backends/xnnpack/serialization/serializer.h
@@ -13,7 +13,6 @@ namespace delegate {
 
 using namespace fb_xnnpack; // Specified in the schema
 
-
 class XNNSerializer {
  public:
   // Constructors
@@ -26,8 +25,9 @@ class XNNSerializer {
       : _builder(bufferSize),
         _nodes(),
         _values(),
-        _constantBuffer(
-            {CreateBuffer(_builder, {})}), // index 0 is reserved for non-const data
+        _constantBuffer({CreateBuffer(
+            _builder,
+            {})}), // index 0 is reserved for non-const data
         _bufferSizes({0}) {}
 
   // Serializing Nodes
