Skip to content

Commit 58e8190

Browse files
authored
Update README.md
1 parent d2028bb commit 58e8190

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

README.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,7 @@
11
# pytorch2c
22

3+
**NOTE: this project is currently under being reworked; instead of graph traversal, it will be based on the new tracing functionality being implemented in PyTorch as of 0.2.0+ **
4+
35
A Python module for compiling (static) [PyTorch](http://pytorch.org) graphs to C (relying on TH and THNN).
46

57
PyTorch2c inspects the computation graph and emits C code that performs the same computation. As long as a network is static (i.e. the graph doesn't change dynamically) it should produce a C source file that links to TH and THNN and can be compiled stand-alone. Interestingly, compiled graphs can be tested automatically by comparing what PyTorch produces to what the compiled code produces, given the same input.
@@ -10,8 +12,6 @@ Caveats:
1012
* I'm developing on macOS and Python 3.5 at the moment
1113
* PyTorch2c currently supports PyTorch version 0.1.10
1214

13-
**NOTE: autograd is currently being refactored in PyTorch: this project is of limited utility until the autograd branch in PyTorch is merged**
14-
1515
## TODO
1616

1717
* [x] Solve storage serialization issues

0 commit comments

Comments
 (0)