mirror of
https://github.com/codeflash-ai/codeflash-agent.git
synced 2026-05-04 18:25:19 +00:00
20f6c59f05
Remove .codeflash/ from ruff extend-exclude, add per-file ignores for .codeflash/, scripts/, evals/, and plugin/ (benchmark/script patterns like print, eval, magic values). Remove shebangs. Widen pre-commit hooks to check the full repo.
103 lines
2.6 KiB
Python
103 lines
2.6 KiB
Python
"""
|
|
Unified profiler for optimized image processing benchmark.
|
|
Profiles both CPU (cProfile) and memory (tracemalloc).
|
|
"""
|
|
|
|
import cProfile
|
|
import io
|
|
import os
|
|
import pstats
|
|
import sys
|
|
import tracemalloc
|
|
|
|
# Add odoo to path
|
|
sys.path.insert(0, "/Users/krrt7/Desktop/work/odoo_org/odoo")
|
|
venv_site_packages = (
|
|
"/Users/krrt7/Desktop/work/odoo_org/odoo/venv/lib/python3.14/site-packages"
|
|
)
|
|
if os.path.exists(venv_site_packages):
|
|
sys.path.insert(0, venv_site_packages)
|
|
|
|
# Import benchmark module
|
|
sys.path.insert(0, "/Users/krrt7/Desktop/work/odoo_org/odoo/.codeflash")
|
|
from benchmark_image_opt1 import (
|
|
benchmark_base64_operations,
|
|
benchmark_format_conversion,
|
|
benchmark_resize_pil,
|
|
benchmark_thumbnail_pil,
|
|
create_test_image,
|
|
)
|
|
|
|
|
|
def profile_image_operations():
|
|
"""Profile image operations with CPU and memory tracking."""
|
|
|
|
# Start memory profiling
|
|
tracemalloc.start()
|
|
|
|
# Start CPU profiling
|
|
profiler = cProfile.Profile()
|
|
profiler.enable()
|
|
|
|
# Run benchmark operations
|
|
print("Running profiled image operations (optimized)...")
|
|
|
|
# Create test images
|
|
small_img = create_test_image(512, 512)
|
|
medium_img = create_test_image(2048, 2048)
|
|
|
|
# Run operations multiple times for better profiling data
|
|
for _ in range(5):
|
|
benchmark_resize_pil(medium_img, (512, 512))
|
|
benchmark_thumbnail_pil(medium_img, (256, 256))
|
|
benchmark_format_conversion(small_img)
|
|
benchmark_base64_operations(medium_img)
|
|
|
|
# Stop CPU profiling
|
|
profiler.disable()
|
|
|
|
# Get memory snapshot
|
|
snapshot = tracemalloc.take_snapshot()
|
|
|
|
# Print CPU profile
|
|
print("\n" + "=" * 80)
|
|
print("CPU PROFILE (top 40 by cumulative time)")
|
|
print("=" * 80)
|
|
s = io.StringIO()
|
|
ps = pstats.Stats(profiler, stream=s)
|
|
ps.strip_dirs()
|
|
ps.sort_stats("cumulative")
|
|
ps.print_stats(40)
|
|
print(s.getvalue())
|
|
|
|
print("\n" + "=" * 80)
|
|
print("CPU PROFILE (top 40 by total time)")
|
|
print("=" * 80)
|
|
s = io.StringIO()
|
|
ps = pstats.Stats(profiler, stream=s)
|
|
ps.strip_dirs()
|
|
ps.sort_stats("tottime")
|
|
ps.print_stats(40)
|
|
print(s.getvalue())
|
|
|
|
# Save detailed profiles
|
|
profiler.dump_stats(
|
|
"/Users/krrt7/Desktop/work/odoo_org/odoo/.codeflash/cpu_profile_opt1.prof"
|
|
)
|
|
|
|
# Summary
|
|
current, peak = tracemalloc.get_traced_memory()
|
|
print("\n" + "=" * 80)
|
|
print("MEMORY SUMMARY")
|
|
print("=" * 80)
|
|
print(f"Current memory: {current / 1024 / 1024:.1f} MiB")
|
|
print(f"Peak memory: {peak / 1024 / 1024:.1f} MiB")
|
|
|
|
tracemalloc.stop()
|
|
|
|
return True
|
|
|
|
|
|
if __name__ == "__main__":
|
|
success = profile_image_operations()
|
|
sys.exit(0 if success else 1)
|