Shushi Hong
b172d5ea32
This PR makes `arith::Analyzer` a first-class tvm-ffi object. The implementation splits the previous concrete `Analyzer` class into: - `AnalyzerObj`, the mutable object node that owns analyzer state, sub-analyzers, caches, and bindings - `Analyzer`, a reference-counted `ObjectRef` handle that can be passed across the tvm-ffi boundary This allows Python and C++ to share the same analyzer instance, so bindings, constraints, and cached facts can persist across FFI calls. Public APIs that accept an analyzer now use `const arith::Analyzer&`, while internal helper APIs that only borrow the object continue to use `AnalyzerObj*`. --------- Co-authored-by: Ubospica <ubospica@gmail.com>
41 lines
1.5 KiB
Python
41 lines
1.5 KiB
Python
# isort: skip_file
|
|
# Licensed to the Apache Software Foundation (ASF) under one
|
|
# or more contributor license agreements. See the NOTICE file
|
|
# distributed with this work for additional information
|
|
# regarding copyright ownership. The ASF licenses this file
|
|
# to you under the Apache License, Version 2.0 (the
|
|
# "License"); you may not use this file except in compliance
|
|
# with the License. You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing,
|
|
# software distributed under the License is distributed on an
|
|
# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
|
|
# KIND, either express or implied. See the License for the
|
|
# specific language governing permissions and limitations
|
|
# under the License.
|
|
"""Integer bound analysis, simplification and pattern detection."""
|
|
|
|
from .int_set import (
|
|
IntSet,
|
|
IntervalSet,
|
|
PresburgerSet,
|
|
estimate_region_lower_bound,
|
|
estimate_region_strict_bound,
|
|
estimate_region_upper_bound,
|
|
)
|
|
from .analyzer import ModularSet, ConstIntBound, Analyzer, ProofStrength, Extension, CompareResult
|
|
from .bound import deduce_bound
|
|
from .pattern import detect_linear_equation, detect_clip_bound
|
|
from .int_solver import solve_linear_equations, solve_linear_inequalities
|
|
from .iter_affine_map import IterMapExpr, IterMark, IterSplitExpr, IterSumExpr
|
|
from .iter_affine_map import (
|
|
detect_iter_map,
|
|
iter_map_simplify,
|
|
normalize_iter_map_to_expr,
|
|
normalize_to_iter_sum,
|
|
subspace_divide,
|
|
inverse_affine_iter_map,
|
|
)
|