[ML] Harden pytorch_inference with TorchScript model graph validation#2936

Merged

edsavage merged 30 commits into

elastic:mainfrom

edsavage:feature/harden_pytorch_inference

Mar 12, 2026

Commits on Mar 1, 2026

[ML] Harden pytorch_inference by validating TorchScript model graphs
edsavage
committed
[ML] Move extract_model_ops to its own subdirectory with README and config
edsavage
committed
[ML] Add Elastic HuggingFace models to reference config and harden script
edsavage
committed

Commits on Mar 2, 2026

[ML] Remove models that fail tracing from reference config
edsavage
committed
[ML] Document rationale for dual forbidden/allowed operation lists
edsavage
committed
[ML] Pin extraction to PyTorch 2.7.1 matching libtorch build version
edsavage
committed
[ML] Log observed TorchScript ops at DEBUG level during validation
edsavage
committed
[ML] Inline TorchScript graph before validation and forbid prim::CallMethod
edsavage
committed
[ML] Enforce maximum graph node count to bound resource consumption
edsavage
committed
[ML] Add integration tests using real TorchScript modules
edsavage
committed
[ML] Fix clang-format in Main.cc node count check
edsavage
committed
[ML] Add allowlist validation script for pytorch_inference models
edsavage
committed
[ML] Add Elasticsearch ecosystem models to validation test suite
edsavage
committed
[ML] Validate allowlist against Elasticsearch integration test models
edsavage
committed
[ML] Add integration tests with malicious TorchScript model fixtures
edsavage
committed
[ML] Replace run_validation.sh with portable CMake script
edsavage
committed
[ML] Wire pytorch_inference validation into primary test targets
edsavage
committed
[ML] Fix clang-format in CModelGraphValidatorTest
edsavage
committed

Commits on Mar 9, 2026

Add CHANGELOG entry for TorchScript model graph validation
edsavage
committed

Commits on Mar 10, 2026

Commits on Mar 11, 2026