mirror of https://github.com/llvm/torch-mlir
55 lines
1.6 KiB
C++
55 lines
1.6 KiB
C++
//===- Algorithm.h - Main algorithm ---------------------------------------===//
|
|
//
|
|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
// See https://llvm.org/LICENSE.txt for license information.
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
// Support types and utilities for the Cartesian Product Algorithm for
|
|
// Type Inference.
|
|
//
|
|
// See:
|
|
// http://citeseerx.ist.psu.edu/viewdoc/summary?doi=10.1.1.30.8177
|
|
// http://citeseerx.ist.psu.edu/viewdoc/summary?doi=10.1.1.129.2756
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#ifndef NPCOMP_TYPING_CPA_ALGORITHM_H
|
|
#define NPCOMP_TYPING_CPA_ALGORITHM_H
|
|
|
|
#include "npcomp/Typing/CPA/Support.h"
|
|
#include "llvm/ADT/DenseSet.h"
|
|
|
|
namespace mlir {
|
|
namespace NPCOMP {
|
|
namespace Typing {
|
|
namespace CPA {
|
|
|
|
/// Propagates constraints in an environment.
|
|
class PropagationWorklist {
|
|
public:
|
|
PropagationWorklist(Environment &env);
|
|
|
|
/// Propagates any current constraints that match the transitivity rule:
|
|
/// τv <: t, t <: τ (τv=ValueType, t=TypeVar, τ=TypeBase)
|
|
/// Expanding to:
|
|
/// τv <: τ
|
|
/// (τv=ValueType, t=TypeVar, τ=TypeBase)
|
|
void propagateTransitivity();
|
|
|
|
/// Commits the current round, returning true if any new constraints were
|
|
/// added.
|
|
bool commit();
|
|
|
|
private:
|
|
Environment &env;
|
|
llvm::DenseSet<Constraint *> currentConstraints;
|
|
int newConstraintCount = 0;
|
|
};
|
|
|
|
} // namespace CPA
|
|
} // namespace Typing
|
|
} // namespace NPCOMP
|
|
} // namespace mlir
|
|
|
|
#endif // NPCOMP_TYPING_CPA_ALGORITHM_H
|