Merge pull request #12 from zachgrayio/feature/fix-compiler-flag

be sure to include -O to compiler

Author: zachgrayio

Dockerfile

@@ -19,7 +19,7 @@ ENV CONFIG=debug
 # user can pass in LIVE=true
 ENV LIVE=false
 
-RUN swift build --configuration ${CONFIG}
+RUN swift build -Xswiftc -O --configuration ${CONFIG}
 
 RUN cp ./.build/${CONFIG}/libSTSLibrary.so /usr/lib/libSTSLibrary.so
 RUN cp ./.build/${CONFIG}/STSLibrary.swiftmodule /usr/lib/STSLibrary.swiftmodule

README.md

@@ -182,6 +182,12 @@ NOTE: if you don't include the `-b|--build` flag to `app run` then the previousl
 
 # Usage
 
+### Writing TensorFlow Code
+
+Something to keep in mind when writing your TensorFlow code, as to avoid issues with send/receive (from the [official FAQ](https://github.com/tensorflow/swift/blob/master/FAQ.md#why-do-i-get-error-internal-error-generating-tensorflow-graph-graphgen-cannot-lower-a-sendreceive-to-the-host-yet)):
+
+>We recommend separating functions that do tensor computation from host code in your programs. Those functions should be marked as `@inline(never)` (and have `public` access, to be safe). Within those functions, tensor computation should not be interrupted by host code. This ensures that the arguments and results of the extracted tensor program will be values on the host, as expected.
+
 ### SwiftPM Project Settings
 
 By default the following names are used:
@@ -247,7 +253,7 @@ If you wish to run `swift build` and `swift run` on your project outside of the
 
 ```bash
 export PATH=/Library/Developer/Toolchains/swift-latest/usr/bin:"${PATH}"
-swift run
+swift run -Xswiftc -O
 ```
 
 # License

entrypoint

@@ -4,7 +4,7 @@ CONFIG=$1
 LIVE=$2
 
 build() {
-    swift build --configuration ${CONFIG}
+    swift build -Xswiftc -O --configuration ${CONFIG}
 }
 
 install_application() {

sts

@@ -98,7 +98,7 @@ case ${STS_COMMAND} in
         if [ "$STS_LIVE_ENABLED" = true ] ; then
             echo -e "[STS][test][${STS_COMMAND_ARG}] ignoring unsupported flag: ${TXT_YELLOW}-l|--live${TXT_NC}"
         fi
-        RUN_COMMAND="docker run --rm -it ${STS_LIVE_FLAG} ${STS_VOLUME_FLAG} --entrypoint "/usr/bin/swift" ${STS_NAME} test"
+        RUN_COMMAND="docker run --rm -it ${STS_LIVE_FLAG} ${STS_VOLUME_FLAG} --entrypoint "/usr/bin/swift" ${STS_NAME} test -Xswiftc -O"
         ;;
         xcode)
         if [ "$STS_LIVE_ENABLED" = true ] ; then