xref: /aosp_15_r20/external/pytorch/tools/gen_flatbuffers.sh (revision da0073e96a02ea20f0ac840b70461e3646d07c45)
1*da0073e9SAndroid Build Coastguard Worker#!/bin/bash
2*da0073e9SAndroid Build Coastguard WorkerROOT=$(pwd)
3*da0073e9SAndroid Build Coastguard WorkerFF_LOCATION="$ROOT/third_party/flatbuffers"
4*da0073e9SAndroid Build Coastguard Workercd "$FF_LOCATION" || exit
5*da0073e9SAndroid Build Coastguard Workermkdir build
6*da0073e9SAndroid Build Coastguard Workercd build || exit
7*da0073e9SAndroid Build Coastguard Workercmake ..
8*da0073e9SAndroid Build Coastguard Workercmake --build . --target flatc
9*da0073e9SAndroid Build Coastguard Workermkdir -p "$ROOT/build/torch/csrc/jit/serialization"
10*da0073e9SAndroid Build Coastguard Worker./flatc --cpp --gen-mutable --scoped-enums \
11*da0073e9SAndroid Build Coastguard Worker     -o "$ROOT/torch/csrc/jit/serialization" \
12*da0073e9SAndroid Build Coastguard Worker     -c "$ROOT/torch/csrc/jit/serialization/mobile_bytecode.fbs"
13*da0073e9SAndroid Build Coastguard Workerecho '// @generated' >> "$ROOT/torch/csrc/jit/serialization/mobile_bytecode_generated.h"
14*da0073e9SAndroid Build Coastguard Workercd "$ROOT" || exit
15*da0073e9SAndroid Build Coastguard Workerexit
16