ci(profiling): debug pprof output.

peterg17 · peterg17 · commit d5f94d60bf03 · 2024-12-13T03:53:24.000Z
diff --git a/tests/profiling/test_main.py b/tests/profiling/test_main.py
@@ -43,6 +43,8 @@ def test_call_script_gevent(monkeypatch):
 def test_call_script_pytorch_gpu(tmp_path, monkeypatch):
     filename = str(tmp_path / "pprof")
     monkeypatch.setenv("DD_PROFILING_OUTPUT_PPROF", filename)
+    monkeypatch.setenv("DD_PROFILING_ENABLED", "1")
+    monkeypatch.setenv("DD_PROFILING_PYTORCH_ENABLED", "1")
     stdout, stderr, exitcode, pid = call_program(
         "ddtrace-run", sys.executable, os.path.join(os.path.dirname(__file__), "simple_program_pytorch_gpu.py")
     )

Original file line number	Diff line number	Diff line change
`@@ -43,6 +43,8 @@ def test_call_script_gevent(monkeypatch):`
`43`	`43`	`def test_call_script_pytorch_gpu(tmp_path, monkeypatch):`
`44`	`44`	`filename = str(tmp_path / "pprof")`
`45`	`45`	`monkeypatch.setenv("DD_PROFILING_OUTPUT_PPROF", filename)`
	`46`	`+ monkeypatch.setenv("DD_PROFILING_ENABLED", "1")`
	`47`	`+ monkeypatch.setenv("DD_PROFILING_PYTORCH_ENABLED", "1")`
`46`	`48`	`stdout, stderr, exitcode, pid = call_program(`
`47`	`49`	`"ddtrace-run", sys.executable, os.path.join(os.path.dirname(__file__), "simple_program_pytorch_gpu.py")`
`48`	`50`	`)`