Commit 09fa96c
tracer improvements (#970)
* Consolidate FunctionRanker: merge rank/rerank/filter methods into single rank_functions
* calculate in own file time
remove unittests remnants
* implement suggestions
* cleanup code
* let's make it clear it's an sqlite3 db
* forgot this one
* cleanup
* tessl add
* improve filtering
* cleanup
* Optimize FunctionRanker.get_function_stats_summary (#971)
The optimization replaces an O(N) linear search through all functions with an O(1) hash table lookup followed by iteration over only matching function names.
**Key Changes:**
- Added `_function_stats_by_name` index in `__init__` that maps function names to lists of (key, stats) tuples
- Modified `get_function_stats_summary` to first lookup candidates by function name, then iterate only over those candidates
**Why This is Faster:**
The original code iterates through ALL function stats (22,603 iterations in the profiler results) for every lookup. The optimized version uses a hash table to instantly find only the functions with matching names, then iterates through just those candidates (typically 1-2 functions).
**Performance Impact:**
- **Small datasets**: 15-30% speedup as shown in basic test cases
- **Large datasets**: Dramatic improvement - the `test_large_scale_performance` case with 900 functions shows **3085% speedup** (66.7μs → 2.09μs)
- **Overall benchmark**: 2061% speedup demonstrates the optimization scales excellently with dataset size
**When This Optimization Shines:**
- Large codebases with many profiled functions (where the linear search becomes expensive)
- Repeated function lookups (if this method is called frequently)
- Cases with many unique function names but few duplicates per name
The optimization maintains identical behavior while transforming the algorithm from O(N) per lookup to O(average functions per name) per lookup, which is typically O(1) in practice.
Co-authored-by: codeflash-ai[bot] <148906541+codeflash-ai[bot]@users.noreply.github.com>
* Revert "let's make it clear it's an sqlite3 db"
This reverts commit 713f135.
* cleanup trace file
* cleanup
* addressable time
* Optimize TestResults.add
The optimization applies **local variable caching** to eliminate repeated attribute lookups on `self.test_result_idx` and `self.test_results`.
**Key Changes:**
- Added `test_result_idx = self.test_result_idx` and `test_results = self.test_results` to cache references locally
- Used these local variables instead of accessing `self.*` attributes multiple times
**Why This Works:**
In Python, attribute access (e.g., `self.test_result_idx`) involves dictionary lookups in the object's `__dict__`, which is slower than accessing local variables. By caching these references, we eliminate redundant attribute resolution overhead on each access.
**Performance Impact:**
The line profiler shows the optimization reduces total execution time from 12.771ms to 19.482ms in the profiler run, but the actual runtime improved from 2.13ms to 1.89ms (12% speedup). The test results consistently show 10-20% improvements across various scenarios, particularly benefiting:
- Large-scale operations (500+ items): 14-16% faster
- Multiple unique additions: 15-20% faster
- Mixed workloads with duplicates: 7-15% faster
**Real-World Benefits:**
This optimization is especially valuable for high-frequency test result collection scenarios where the `add` method is called repeatedly in tight loops, as the cumulative effect of eliminating attribute lookups becomes significant at scale.
* bugfix
* cleanup
* type checks
* pre-commit
* ⚡️ Speed up function `get_cached_gh_event_data` by 13% (#975)
* Optimize get_cached_gh_event_data
The optimization replaces `Path(event_path).open(encoding="utf-8")` with the built-in `open(event_path, encoding="utf-8")`, achieving a **12% speedup** by eliminating unnecessary object allocation overhead.
**Key optimization:**
- **Removed Path object creation**: The original code creates a `pathlib.Path` object just to call `.open()` on it, when the built-in `open()` function can directly accept the string path from `event_path`.
- **Reduced memory allocation**: Avoiding the intermediate `Path` object saves both allocation time and memory overhead.
**Why this works:**
In Python, `pathlib.Path().open()` internally calls the same file opening mechanism as the built-in `open()`, but with additional overhead from object instantiation and method dispatch. Since `event_path` is already a string from `os.getenv()`, passing it directly to `open()` is more efficient.
**Performance impact:**
The test results show consistent improvements across all file-reading scenarios:
- Simple JSON files: 12-20% faster
- Large files (1000+ elements): 3-27% faster
- Error cases (missing files): Up to 71% faster
- The cached calls remain unaffected (0% change as expected)
**Workload benefits:**
Based on the function references, `get_cached_gh_event_data()` is called by multiple GitHub-related utility functions (`get_pr_number()`, `is_repo_a_fork()`, `is_pr_draft()`). While the `@lru_cache(maxsize=1)` means the file is only read once per program execution, this optimization reduces the initial cold-start latency for GitHub Actions workflows or CI/CD pipelines where these functions are commonly used.
The optimization is particularly effective for larger JSON files and error handling scenarios, making it valuable for robust CI/CD environments that may encounter various file conditions.
* ignore
---------
Co-authored-by: codeflash-ai[bot] <148906541+codeflash-ai[bot]@users.noreply.github.com>
Co-authored-by: Kevin Turcios <[email protected]>
* ⚡️ Speed up function `function_is_a_property` by 60% (#974)
* Optimize function_is_a_property
The optimized version achieves a **60% speedup** by replacing Python's `any()` generator expression with a manual loop and making three key micro-optimizations:
**What was optimized:**
1. **Replaced `isinstance()` with `type() is`**: Direct type comparison (`type(node) is ast_Name`) is faster than `isinstance(node, ast.Name)` for AST nodes where subclassing is rare
2. **Eliminated repeated lookups**: Cached `"property"` as `property_id` and `ast.Name` as `ast_Name` in local variables to avoid global/attribute lookups in the loop
3. **Manual loop with early return**: Replaced `any()` generator with explicit `for` loop that returns `True` immediately upon finding a match, avoiding generator overhead
**Why it's faster:**
- The `any()` function creates generator machinery that adds overhead, especially for small decorator lists
- `isinstance()` performs multiple checks while `type() is` does a single identity comparison
- Local variable access is significantly faster than repeated global/attribute lookups in tight loops
**Performance characteristics from tests:**
- **Small decorator lists** (1-3 decorators): 50-80% faster due to reduced per-iteration overhead
- **Large decorator lists** (1000+ decorators): 55-60% consistent speedup, with early termination providing additional benefits when `@property` appears early
- **Empty decorator lists**: 77% faster due to avoiding `any()` generator setup entirely
**Impact on workloads:**
Based on the function references, this function is called during AST traversal in `visit_FunctionDef` and `visit_AsyncFunctionDef` methods - likely part of a code analysis pipeline that processes many functions. The 60% speedup will be particularly beneficial when analyzing codebases with many decorated functions, as this optimization reduces overhead in a hot path that's called once per function definition.
* format
---------
Co-authored-by: codeflash-ai[bot] <148906541+codeflash-ai[bot]@users.noreply.github.com>
Co-authored-by: Kevin Turcios <[email protected]>
* Optimize function_is_a_property (#976)
The optimization achieves an **11% speedup** through two key changes:
**1. Constant Hoisting:** The original code repeatedly assigns `property_id = "property"` and `ast_name = ast.Name` on every function call. The optimized version moves these to module-level constants `_property_id` and `_ast_name`, eliminating 4,130 redundant assignments per profiling run (saving ~2.12ms total time).
**2. isinstance() vs type() comparison:** Replaced `type(node) is ast_name` with `isinstance(node, _ast_name)`. While both are correct for AST nodes (which use single inheritance), `isinstance()` is slightly more efficient for type checking in Python's implementation.
**Performance Impact:** The function is called in AST traversal loops when discovering functions to optimize (`visit_FunctionDef` and `visit_AsyncFunctionDef`). Since these visitors process entire codebases, the 11% per-call improvement compounds significantly across large projects.
**Test Case Performance:** The optimization shows consistent gains across all test scenarios:
- **Simple cases** (no decorators): 29-42% faster due to eliminated constant assignments
- **Property detection cases**: 11-26% faster from combined optimizations
- **Large-scale tests** (500-1000 functions): 18.5% faster, demonstrating the cumulative benefit when processing many functions
The optimizations are particularly effective for codebases with many function definitions, where this function gets called repeatedly during AST analysis.
Co-authored-by: codeflash-ai[bot] <148906541+codeflash-ai[bot]@users.noreply.github.com>
* Address PR review comments
- Add mkdir for test file directory to prevent FileNotFoundError
- Use addressable_time_ns for importance filtering instead of own_time_ns
- Remove unnecessary list() wrappers in make_pstats_compatible
- Remove old .sqlite3 file with wrong extension
Co-Authored-By: Warp <[email protected]>
* Check addressable_time_ns instead of own_time_ns for filtering
This ensures we consider functions that may have low own_time but high
time in first-order dependent functions (callees).
Co-Authored-By: Warp <[email protected]>
---------
Co-authored-by: codeflash-ai[bot] <148906541+codeflash-ai[bot]@users.noreply.github.com>
Co-authored-by: Saurabh Misra <[email protected]>
Co-authored-by: Warp <[email protected]>1 parent 872ec28 commit 09fa96c
File tree
14 files changed
+474
-266
lines changed- codeflash
- benchmarking
- code_utils
- discovery
- models
- optimization
- tracing
- tests
14 files changed
+474
-266
lines changed| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
254 | 254 | | |
255 | 255 | | |
256 | 256 | | |
257 | | - | |
| 257 | + | |
| 258 | + | |
| 259 | + | |
| 260 | + | |
| 261 | + | |
| 262 | + | |
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
315 | 315 | | |
316 | 316 | | |
317 | 317 | | |
318 | | - | |
| 318 | + | |
| 319 | + | |
| 320 | + | |
| 321 | + | |
| 322 | + | |
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
2 | 2 | | |
3 | 3 | | |
4 | 4 | | |
5 | | - | |
| 5 | + | |
6 | 6 | | |
7 | 7 | | |
8 | 8 | | |
| |||
12 | 12 | | |
13 | 13 | | |
14 | 14 | | |
| 15 | + | |
| 16 | + | |
| 17 | + | |
| 18 | + | |
| 19 | + | |
| 20 | + | |
| 21 | + | |
| 22 | + | |
| 23 | + | |
| 24 | + | |
| 25 | + | |
| 26 | + | |
| 27 | + | |
| 28 | + | |
| 29 | + | |
| 30 | + | |
| 31 | + | |
| 32 | + | |
| 33 | + | |
| 34 | + | |
| 35 | + | |
| 36 | + | |
| 37 | + | |
| 38 | + | |
| 39 | + | |
15 | 40 | | |
16 | 41 | | |
17 | | - | |
| 42 | + | |
18 | 43 | | |
19 | | - | |
20 | | - | |
| 44 | + | |
| 45 | + | |
21 | 46 | | |
22 | | - | |
23 | | - | |
| 47 | + | |
| 48 | + | |
| 49 | + | |
24 | 50 | | |
25 | 51 | | |
26 | | - | |
| 52 | + | |
27 | 53 | | |
28 | 54 | | |
29 | 55 | | |
30 | 56 | | |
31 | 57 | | |
32 | 58 | | |
33 | 59 | | |
| 60 | + | |
34 | 61 | | |
35 | 62 | | |
| 63 | + | |
| 64 | + | |
| 65 | + | |
| 66 | + | |
| 67 | + | |
| 68 | + | |
36 | 69 | | |
37 | 70 | | |
| 71 | + | |
38 | 72 | | |
39 | 73 | | |
40 | 74 | | |
| |||
45 | 79 | | |
46 | 80 | | |
47 | 81 | | |
| 82 | + | |
| 83 | + | |
| 84 | + | |
| 85 | + | |
48 | 86 | | |
49 | 87 | | |
50 | 88 | | |
| |||
56 | 94 | | |
57 | 95 | | |
58 | 96 | | |
59 | | - | |
60 | | - | |
| 97 | + | |
| 98 | + | |
61 | 99 | | |
62 | 100 | | |
63 | 101 | | |
| |||
70 | 108 | | |
71 | 109 | | |
72 | 110 | | |
73 | | - | |
| 111 | + | |
74 | 112 | | |
75 | 113 | | |
76 | | - | |
| 114 | + | |
| 115 | + | |
| 116 | + | |
| 117 | + | |
77 | 118 | | |
78 | 119 | | |
79 | 120 | | |
80 | 121 | | |
81 | 122 | | |
82 | | - | |
| 123 | + | |
83 | 124 | | |
84 | | - | |
85 | | - | |
86 | | - | |
87 | | - | |
| 125 | + | |
| 126 | + | |
| 127 | + | |
| 128 | + | |
| 129 | + | |
| 130 | + | |
| 131 | + | |
| 132 | + | |
| 133 | + | |
| 134 | + | |
88 | 135 | | |
89 | 136 | | |
90 | 137 | | |
91 | 138 | | |
92 | 139 | | |
93 | 140 | | |
94 | 141 | | |
95 | | - | |
96 | | - | |
97 | | - | |
| 142 | + | |
| 143 | + | |
98 | 144 | | |
99 | | - | |
100 | | - | |
101 | | - | |
102 | | - | |
103 | | - | |
104 | | - | |
| 145 | + | |
| 146 | + | |
| 147 | + | |
| 148 | + | |
| 149 | + | |
105 | 150 | | |
106 | | - | |
107 | | - | |
| 151 | + | |
| 152 | + | |
108 | 153 | | |
109 | | - | |
110 | | - | |
| 154 | + | |
| 155 | + | |
111 | 156 | | |
112 | | - | |
113 | | - | |
114 | | - | |
115 | | - | |
116 | | - | |
117 | | - | |
| 157 | + | |
| 158 | + | |
| 159 | + | |
118 | 160 | | |
119 | | - | |
| 161 | + | |
| 162 | + | |
| 163 | + | |
120 | 164 | | |
121 | | - | |
122 | | - | |
| 165 | + | |
| 166 | + | |
123 | 167 | | |
124 | | - | |
125 | | - | |
126 | | - | |
| 168 | + | |
| 169 | + | |
127 | 170 | | |
128 | | - | |
129 | 171 | | |
130 | | - | |
131 | | - | |
| 172 | + | |
| 173 | + | |
132 | 174 | | |
133 | 175 | | |
134 | | - | |
| 176 | + | |
| 177 | + | |
| 178 | + | |
| 179 | + | |
| 180 | + | |
| 181 | + | |
| 182 | + | |
| 183 | + | |
| 184 | + | |
| 185 | + | |
| 186 | + | |
| 187 | + | |
| 188 | + | |
| 189 | + | |
| 190 | + | |
| 191 | + | |
| 192 | + | |
| 193 | + | |
| 194 | + | |
| 195 | + | |
| 196 | + | |
| 197 | + | |
135 | 198 | | |
136 | 199 | | |
137 | 200 | | |
138 | | - | |
139 | | - | |
140 | | - | |
141 | | - | |
142 | | - | |
143 | | - | |
144 | | - | |
145 | | - | |
146 | | - | |
147 | | - | |
148 | | - | |
149 | | - | |
150 | | - | |
151 | | - | |
152 | | - | |
153 | | - | |
154 | | - | |
| 201 | + | |
| 202 | + | |
| 203 | + | |
| 204 | + | |
| 205 | + | |
| 206 | + | |
| 207 | + | |
| 208 | + | |
| 209 | + | |
| 210 | + | |
| 211 | + | |
| 212 | + | |
| 213 | + | |
| 214 | + | |
| 215 | + | |
| 216 | + | |
| 217 | + | |
| 218 | + | |
| 219 | + | |
| 220 | + | |
| 221 | + | |
| 222 | + | |
| 223 | + | |
155 | 224 | | |
156 | | - | |
157 | | - | |
158 | | - | |
| 225 | + | |
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
64 | 64 | | |
65 | 65 | | |
66 | 66 | | |
67 | | - | |
68 | | - | |
69 | | - | |
70 | | - | |
| 67 | + | |
71 | 68 | | |
72 | 69 | | |
73 | 70 | | |
74 | 71 | | |
75 | 72 | | |
76 | 73 | | |
77 | 74 | | |
78 | | - | |
79 | 75 | | |
80 | 76 | | |
81 | 77 | | |
82 | 78 | | |
83 | 79 | | |
84 | 80 | | |
85 | 81 | | |
86 | | - | |
87 | | - | |
88 | 82 | | |
89 | | - | |
90 | | - | |
| 83 | + | |
91 | 84 | | |
92 | 85 | | |
93 | 86 | | |
| |||
158 | 151 | | |
159 | 152 | | |
160 | 153 | | |
161 | | - | |
162 | | - | |
163 | | - | |
164 | | - | |
165 | | - | |
166 | | - | |
167 | | - | |
| 154 | + | |
168 | 155 | | |
169 | 156 | | |
170 | 157 | | |
| |||
223 | 210 | | |
224 | 211 | | |
225 | 212 | | |
226 | | - | |
| 213 | + | |
227 | 214 | | |
228 | | - | |
229 | 215 | | |
230 | | - | |
| 216 | + | |
231 | 217 | | |
232 | 218 | | |
233 | 219 | | |
234 | 220 | | |
235 | | - | |
236 | | - | |
237 | | - | |
| 221 | + | |
238 | 222 | | |
239 | 223 | | |
240 | 224 | | |
241 | 225 | | |
242 | 226 | | |
243 | 227 | | |
244 | | - | |
245 | 228 | | |
246 | 229 | | |
247 | 230 | | |
248 | 231 | | |
249 | | - | |
| 232 | + | |
250 | 233 | | |
251 | 234 | | |
252 | 235 | | |
| |||
293 | 276 | | |
294 | 277 | | |
295 | 278 | | |
296 | | - | |
297 | | - | |
298 | | - | |
299 | | - | |
| 279 | + | |
300 | 280 | | |
301 | 281 | | |
302 | 282 | | |
| |||
0 commit comments