[python_pytorch] Initial commit

Author: KowalskiThomas

scenarios/python_pytorch_3.11/Dockerfile

@@ -0,0 +1,17 @@
+FROM python:3.11
+
+WORKDIR /usr/src/app
+
+COPY ./scenarios/python_pytorch_3.11/ /usr/src/app
+
+RUN pip install --no-cache-dir torch --index-url https://download.pytorch.org/whl/cpu
+RUN pip install --no-cache-dir -r requirements.txt
+
+ENV EXECUTION_TIME_SEC 30
+ENV DD_PROFILING_ENABLED true
+ENV DD_TRACE_ENABLED false
+ENV DD_TRACE_DEBUG false
+ENV DD_PROFILING_MEMORY_ENABLED=false
+ENV DD_PROFILING_OUTPUT_PPROF="/app/data/profiles"
+
+CMD ddtrace-run python main.py

scenarios/python_pytorch_3.11/expected_profile.json

@@ -0,0 +1,37 @@
+{
+  "test_name": "python_pytorch",
+  "stacks": [
+    {
+      "profile-type": "cpu-time",
+      "stack-content": [
+        {
+          "regular_expression": "<module>;.*main;.*heavy_compute",
+          "percent": 66,
+          "error_margin": 5,
+          "labels": [
+            {
+              "key": "thread name",
+              "values": [
+                "MainThread"
+              ]
+            }
+          ]
+        },
+        {
+          "regular_expression": "<module>;.*main;.*light_compute",
+          "percent": 33,
+          "error_margin": 6,
+          "labels": [
+            {
+              "key": "thread name",
+              "values": [
+                "MainThread"
+              ]
+            }
+          ]
+        }
+      ]
+    }
+  ],
+  "scale_by_duration": true
+}

scenarios/python_pytorch_3.11/main.py

@@ -0,0 +1,28 @@
+import os
+from time import time
+
+import torch
+
+
+def light_compute(a: torch.Tensor, b: torch.Tensor) -> None:
+    for _ in range(20):
+        torch.mm(a, b)
+
+
+def heavy_compute(a: torch.Tensor, b: torch.Tensor) -> None:
+    for _ in range(40):
+        torch.mm(a, b)
+
+
+def main() -> None:
+    torch.set_num_threads(1)
+    a = torch.randn(300, 300)
+    b = torch.randn(300, 300)
+    execution_time_sec = float(os.getenv("EXECUTION_TIME_SEC", "10"))
+    end = time() + execution_time_sec
+    while time() < end:
+        light_compute(a, b)
+        heavy_compute(a, b)
+
+
+main()

scenarios/python_pytorch_3.11/requirements.txt

@@ -0,0 +1 @@
+ddtrace