xref: /aosp_15_r20/external/pytorch/tools/onnx/gen_diagnostics.sh (revision da0073e96a02ea20f0ac840b70461e3646d07c45)
1*da0073e9SAndroid Build Coastguard Worker#!/bin/bash
2*da0073e9SAndroid Build Coastguard Worker# Run this script inside its folder to generate PyTorch ONNX Export Diagnostic rules
3*da0073e9SAndroid Build Coastguard Worker# for C++, Python and documentations.
4*da0073e9SAndroid Build Coastguard Worker# The rules are defined in torch/onnx/_internal/diagnostics/rules.yaml.
5*da0073e9SAndroid Build Coastguard Worker
6*da0073e9SAndroid Build Coastguard Workerset -e -x
7*da0073e9SAndroid Build Coastguard WorkerROOT="${PWD}/../../"
8*da0073e9SAndroid Build Coastguard Workerpushd "$ROOT"
9*da0073e9SAndroid Build Coastguard Worker(
10*da0073e9SAndroid Build Coastguard Workerpython -m tools.onnx.gen_diagnostics \
11*da0073e9SAndroid Build Coastguard Worker    torch/onnx/_internal/diagnostics/rules.yaml \
12*da0073e9SAndroid Build Coastguard Worker    torch/onnx/_internal/diagnostics \
13*da0073e9SAndroid Build Coastguard Worker    torch/csrc/onnx/diagnostics/generated \
14*da0073e9SAndroid Build Coastguard Worker    torch/docs/source
15*da0073e9SAndroid Build Coastguard Worker)
16*da0073e9SAndroid Build Coastguard Workerpopd
17