mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
8f4edf1e1d
This PR introduces a general Python diagnostics infrastructure powered by SARIF, and the exporter diagnostics module that builds on top of it. Pull Request resolved: https://github.com/pytorch/pytorch/pull/85107 Approved by: https://github.com/abock, https://github.com/justinchuby
17 lines
456 B
Bash
Executable File
17 lines
456 B
Bash
Executable File
#!/bin/bash
|
|
# Run this script inside its folder to generate PyTorch ONNX Export Diagnostic rules
|
|
# for C++, Python and documentations.
|
|
# The rules are defined in torch/onnx/_internal/diagnostics/rules.yaml.
|
|
|
|
set -e -x
|
|
ROOT="${PWD}/../../"
|
|
pushd "$ROOT"
|
|
(
|
|
python -m tools.onnx.gen_diagnostics \
|
|
torch/onnx/_internal/diagnostics/rules.yaml \
|
|
torch/onnx/_internal/diagnostics \
|
|
torch/csrc/onnx/diagnostics/generated \
|
|
torch/docs/source
|
|
)
|
|
popd
|