Extend static analysis for arithmetic operators

[rostan-t]

Category:

New feature (non-breaking change which adds functionality)

Description:

This extends dynamic mode static analysis to support arithmetic operators.

PR #6475 promotes bools, ints, and floats to constants in arithmetic ops. With this PR, we can limit it to only those that are provably constant.

To prevent a regression on Python 3.10 when a call is on the same line as an arithmetic operation, we also add disambiguation on the bytecode since exact span checking is absent.

Additional information:

Affected modules and functionalities:

Dynamic mode.

Key points relevant for the review:

Commits a677329 and 0374268 belong to #6475.

Tests:

• Existing tests apply: test_arithm_ops.py
• New tests added
  • Python tests
  • GTests
  • Benchmark
  • Other
• N/A

Checklist

Documentation

• Existing documentation applies
• Documentation updated
  • Docstring
  • Doxygen
  • RST
  • Jupyter
  • Other
• N/A

DALI team only

Requirements

• Implements new requirements
• Affects existing requirements
• N/A

REQ IDs: N/A

JIRA TASK: N/A

[rostan-t]

!build

[greptile-apps]

Confidence Score: 5/5

The PR appears safe to merge; the only outstanding concern from the previous review was fixed by strict operand-classification iteration.

Summary

• Resolves the previous operand-truncation concern by enforcing equal-length iteration with zip(..., strict=True).
• Adds arithmetic-dunder transparency for Tensor and Batch.
• Generalizes call-site source analysis to calls, binary operations, and augmented assignments, including Python 3.10 bytecode disambiguation.
• Passes classified integer and real constants through the arithmetic operator schema.

Diagram

%%{init: {'theme': 'neutral'}}%%
flowchart LR
    A[Python arithmetic expression] --> B[Transparent arithmetic dunder]
    B --> C[Resolve source site]
    C --> D[Classify operands]
    D -->|Provably constant| E[Integer or real expression constant]
    D -->|Dynamic| F[Tensor or Batch operator input]
    E --> G[Arithmetic expression operator]
    F --> G

Loading

[dali-automaton]

CI MESSAGE: [66962570]: BUILD STARTED

[rostan-t]

!build

[dali-automaton]

CI MESSAGE: [66975775]: BUILD STARTED

[dali-automaton]

CI MESSAGE: [66975775]: BUILD PASSED

[dali-automaton]

CI MESSAGE: [66962570]: BUILD PASSED