16 lines
485 B
Bash
16 lines
485 B
Bash
|
#!/bin/bash
|
||
|
|
ROOT=$(pwd)
|
||
|
|
FF_LOCATION="$ROOT/third_party/flatbuffers"
|
||
|
|
cd "$FF_LOCATION" || exit
|
||
|
|
mkdir build
|
||
|
|
cd build || exit
|
||
|
|
cmake ..
|
||
|
|
cmake --build . --target flatc
|
||
|
|
mkdir -p "$ROOT/build/torch/csrc/jit/serialization"
|
||
|
|
./flatc --cpp --gen-mutable --scoped-enums \
|
||
|
|
-o "$ROOT/torch/csrc/jit/serialization" \
|
||
|
|
-c "$ROOT/torch/csrc/jit/serialization/mobile_bytecode.fbs"
|
||
|
|
echo '// @generated' >> "$ROOT/torch/csrc/jit/serialization/mobile_bytecode_generated.h"
|
||
|
|
cd "$ROOT" || exit
|
||
|
|
exit
|