Improve setup instructions (#24)

Summary: Pull Request resolved: #24

Reviewed By: kimishpatel, cccclai

Differential Revision: D47814545

Pulled By: msaroufim

fbshipit-source-id: 197ad900d2948f7caa5edaa2196654f030fb2282

Author: msaroufim

.gitignore

@@ -1,3 +1,6 @@
 buck-out/
 executorch.egg-info
 __pycache__/
+build/lib/
+exir/serialize/scalar_type.fbs
+exir/serialize/schema.fbs

docs/website/docs/tutorials/setting_up_executorch.md

@@ -28,9 +28,10 @@ git clone https://github.com/pytorch/executorch.git
 git clone [email protected]:pytorch/executorch.git
 
 # [Runtime requirement] Run the following to get all submodules, only need for runtime setup
+cd executorch
 git submodule update --init --recursive
 
-pip install executorch
+pip install .
 
 # cd into a directory that doesn't contain a `./executorch/exir` directory, since
 # otherwise python will try using it for `import executorch.exir...` instead of using the
@@ -53,7 +54,7 @@ Or via python interpreter:
 >>> from executorch.exir.tests.models import Mul
 >>> m = Mul()
 >>> print(exir.capture(m, m.get_random_inputs()).to_edge())
->>> exir.capture(m, m.get_random_inputs()).to_edge().to_executorch().buffer
+>>> open("add.ff", "wb").write(exir.capture(m, m.get_random_inputs()).to_edge().to_executorch().buffer)
 ```
 
 ## Runtime Setup

exir/capture/TARGETS

@@ -21,7 +21,6 @@ python_library(
         ":config",
         ":unlift",
         "//caffe2:torch",
-        "//caffe2/functorch:functorch_src",
         "//executorch/exir:error",
         "//executorch/exir:tracer",
         "//executorch/exir/program:lib",

exir/capture/_capture.py

@@ -17,13 +17,13 @@
     flatten_output,
     Value,
 )
-from functorch.experimental import functionalize
 from torch import _guards
 from torch._dispatch.python import enable_python_dispatcher
 from torch._dynamo.eval_frame import Constraint
 from torch._export import CallSpec, export, ExportGraphSignature
 from torch._export.passes import ReplaceViewOpsWithViewCopyOpsPass
 from torch._subclasses.fake_tensor import FakeTensor, FakeTensorMode
+from torch.func import functionalize
 from torch.fx._compatibility import compatibility
 from torch.fx.experimental.proxy_tensor import make_fx
 from torch.fx.experimental.symbolic_shapes import ShapeEnv

extension/pytree/__init__.py

@@ -1,6 +1,11 @@
 # flake8: noqa: F401
 
-import warnings
+import logging
+
+# Create a logger
+logger = logging.getLogger(__name__)
+logger.setLevel(logging.WARNING)
+
 
 try:
     """
@@ -22,7 +27,7 @@
         TreeSpec as TreeSpec,
     )
 except:
-    warnings.warn(
+    logger.info(
         "Unable to import executorch.extension.pytree, using native torch pytree instead."
     )
 

pyproject.toml

@@ -9,6 +9,9 @@ dependencies=[
   "ruamel.yaml",
   "pyyaml",
   "sympy",
+  "numpy",
+  "zstd",
+  "flatbuffers",
 ]
 
 [tool.setuptools.package-data]