diff --git a/devops/scripts/benchmarks/benches/base.py b/devops/scripts/benchmarks/benches/base.py index 830e299d87e0f..7ede51a66cd44 100644 --- a/devops/scripts/benchmarks/benches/base.py +++ b/devops/scripts/benchmarks/benches/base.py @@ -46,6 +46,7 @@ class TracingType(Enum): BenchmarkTag("inference", "Tests ML/AI inference performance"), BenchmarkTag("image", "Image processing benchmark"), BenchmarkTag("simulation", "Physics or scientific simulation benchmark"), + BenchmarkTag("pytorch", "Tests workloads close to Pytorch ones"), ] benchmark_tags_dict = {tag.name: tag for tag in benchmark_tags}