This is an archive of the discontinued LLVM Phabricator instance.

Differential D76318

NFC - Rename QuantOps to Quant
ClosedPublic

Authored by rsuderman on Mar 17 2020, 1:27 PM.

Details

Reviewers
rriddle
Commits
rG363dd3f394b5: [mlir][NFC] Rename QuantOps to Quant
Summary

Renamed QuantOps to Quant to avoid the Ops suffix. All dialects will contain
ops, so the Ops suffix is redundant.

(github)

Diff Detail

Event Timeline

rsuderman created this revision.Mar 17 2020, 1:27 PM
Herald added a project: Restricted Project. · View Herald TranscriptMar 17 2020, 1:27 PM
Herald added subscribers: llvm-commits, Joonsoo, liufengdb and 11 others. · View Herald Transcript
Harbormaster completed remote builds in B49487: Diff 250887.Mar 17 2020, 2:02 PM
rriddle accepted this revision.Mar 17 2020, 3:00 PM
This revision is now accepted and ready to land.Mar 17 2020, 3:00 PM
Closed by commit rG363dd3f394b5: [mlir][NFC] Rename QuantOps to Quant (authored by rsuderman, committed by rriddle). · Explain WhyMar 17 2020, 3:41 PM
This revision was automatically updated to reflect the committed changes.

Revision Contents

PathSize
mlir/
include/
mlir/
Dialect/
2 lines
FxpMathOps/
2 lines
Quant/
QuantOps/
8 lines
6 lines
10 lines
8 lines
6 lines
6 lines
6 lines
8 lines
QuantOps/
2 lines
2 lines
Quantizer/
Support/
2 lines
2 lines
lib/
Dialect/
2 lines
FxpMathOps/
IR/
2 lines
Transforms/
6 lines
Quant/
QuantOps/
IR/
8 lines
2 lines
4 lines
Transforms/
8 lines
8 lines
Utils/
4 lines
4 lines
2 lines
QuantOps/
IR/
Transforms/
Utils/
Quantizer/
Configurations/
4 lines
Support/
2 lines
Transforms/
4 lines
4 lines
2 lines
test/
Dialect/
Quant/
QuantOps/
QuantOps/
unittests/
Dialect/
Quant/
QuantOps/
6 lines
QuantOps/

Diff 250931

mlir/include/mlir/Dialect/CMakeLists.txt

add_subdirectory(AffineOps) add_subdirectory(AffineOps)
add_subdirectory(FxpMathOps) add_subdirectory(FxpMathOps)
add_subdirectory(GPU) add_subdirectory(GPU)
add_subdirectory(Linalg) add_subdirectory(Linalg)
add_subdirectory(LLVMIR) add_subdirectory(LLVMIR)
add_subdirectory(LoopOps) add_subdirectory(LoopOps)
add_subdirectory(OpenMP) add_subdirectory(OpenMP)
add_subdirectory(QuantOps) add_subdirectory(Quant)
add_subdirectory(Shape) add_subdirectory(Shape)
add_subdirectory(SPIRV) add_subdirectory(SPIRV)
add_subdirectory(StandardOps) add_subdirectory(StandardOps)
add_subdirectory(VectorOps) add_subdirectory(VectorOps)

mlir/include/mlir/Dialect/FxpMathOps/FxpMathOps.td

Show All 9 Lines
// equivalents). // equivalents).
// //
//===----------------------------------------------------------------------===// //===----------------------------------------------------------------------===//
#ifndef DIALECT_FXPMATHOPS_FXPMATH_OPS_ #ifndef DIALECT_FXPMATHOPS_FXPMATH_OPS_
#define DIALECT_FXPMATHOPS_FXPMATH_OPS_ #define DIALECT_FXPMATHOPS_FXPMATH_OPS_
include "mlir/IR/OpBase.td" include "mlir/IR/OpBase.td"
include "mlir/Dialect/QuantOps/QuantOpsBase.td" include "mlir/Dialect/Quant/QuantOpsBase.td"
include "mlir/Interfaces/SideEffects.td" include "mlir/Interfaces/SideEffects.td"
def FxpMathOps_Dialect : Dialect { def FxpMathOps_Dialect : Dialect {
let name = "fxpmath"; let name = "fxpmath";
} }
//===----------------------------------------------------------------------===// //===----------------------------------------------------------------------===//
// Attributes // Attributes
▲ Show 20 Lines • Show All 252 Lines • Show Last 20 Lines

mlir/include/mlir/Dialect/Quant/CMakeLists.txt

  • This file was moved from mlir/include/mlir/Dialect/QuantOps/CMakeLists.txt.
The contents of this file were not changed.

mlir/include/mlir/Dialect/Quant/FakeQuantSupport.h

  • This file was moved from mlir/include/mlir/Dialect/QuantOps/FakeQuantSupport.h.
Show All 28 Lines
// //
// As canonically used in TensorFlow graphs, the presence of a FakeQuant node // As canonically used in TensorFlow graphs, the presence of a FakeQuant node
// is a hint that the specific math represented here has been simulated at // is a hint that the specific math represented here has been simulated at
// training time. As such, it is usually not advised to arbitrarily change // training time. As such, it is usually not advised to arbitrarily change
// quantization parameters derived from FakeQuant. // quantization parameters derived from FakeQuant.
// //
//===----------------------------------------------------------------------===// //===----------------------------------------------------------------------===//
#ifndef MLIR_DIALECT_QUANTOPS_FAKEQUANTSUPPORT_H_ #ifndef MLIR_DIALECT_QUANT_FAKEQUANTSUPPORT_H_
#define MLIR_DIALECT_QUANTOPS_FAKEQUANTSUPPORT_H_ #define MLIR_DIALECT_QUANT_FAKEQUANTSUPPORT_H_
#include "mlir/Dialect/QuantOps/QuantTypes.h" #include "mlir/Dialect/Quant/QuantTypes.h"
namespace mlir { namespace mlir {
namespace quant { namespace quant {
/// Converts per-layer FakeQuant attributes to the corresponding type. /// Converts per-layer FakeQuant attributes to the corresponding type.
/// In the event that the parameters cannot be converted, returns a nullptr /// In the event that the parameters cannot be converted, returns a nullptr
/// convertible Type and issues an appropriate error. /// convertible Type and issues an appropriate error.
/// Note that there are multiple variants of a per-layer FakeQuant op, so /// Note that there are multiple variants of a per-layer FakeQuant op, so
Show All 10 Lines
UniformQuantizedPerAxisType UniformQuantizedPerAxisType
fakeQuantAttrsToType(Location loc, unsigned numBits, int32_t quantizedDimension, fakeQuantAttrsToType(Location loc, unsigned numBits, int32_t quantizedDimension,
ArrayRef<double> rmins, ArrayRef<double> rmax, ArrayRef<double> rmins, ArrayRef<double> rmax,
bool narrowRange, Type expressedType, bool narrowRange, Type expressedType,
bool isSigned = false); bool isSigned = false);
} // namespace quant } // namespace quant
} // namespace mlir } // namespace mlir
#endif // MLIR_DIALECT_QUANTOPS_FAKEQUANTSUPPORT_H_ #endif // MLIR_DIALECT_QUANT_FAKEQUANTSUPPORT_H_

mlir/include/mlir/Dialect/Quant/Passes.h

  • This file was moved from mlir/include/mlir/Dialect/QuantOps/Passes.h.
//===- Passes.h - Quantization Passes ------ --------------------*- C++ -*-===// //===- Passes.h - Quantization Passes ------ --------------------*- C++ -*-===//
// //
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information. // See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
// //
//===----------------------------------------------------------------------===// //===----------------------------------------------------------------------===//
// //
// This file defines all of the passes owned by the quantization dialect. As // This file defines all of the passes owned by the quantization dialect. As
// things mature, it is expected that passes specific to certain frontend or // things mature, it is expected that passes specific to certain frontend or
// backend dialects will move to those dialects directly. For now, they are // backend dialects will move to those dialects directly. For now, they are
// incubated here. // incubated here.
// //
//===----------------------------------------------------------------------===// //===----------------------------------------------------------------------===//
#ifndef MLIR_DIALECT_QUANTOPS_PASSES_H #ifndef MLIR_DIALECT_QUANT_PASSES_H
#define MLIR_DIALECT_QUANTOPS_PASSES_H #define MLIR_DIALECT_QUANT_PASSES_H
#include <memory> #include <memory>
namespace mlir { namespace mlir {
class FuncOp; class FuncOp;
template <typename T> class OpPassBase; template <typename T> class OpPassBase;
namespace quant { namespace quant {
/// Creates a pass that converts quantization simulation operations (i.e. /// Creates a pass that converts quantization simulation operations (i.e.
/// FakeQuant and those like it) to casts into/out of supported QuantizedTypes. /// FakeQuant and those like it) to casts into/out of supported QuantizedTypes.
std::unique_ptr<OpPassBase<FuncOp>> createConvertSimulatedQuantPass(); std::unique_ptr<OpPassBase<FuncOp>> createConvertSimulatedQuantPass();
/// Creates a pass that converts constants followed by a qbarrier to a /// Creates a pass that converts constants followed by a qbarrier to a
/// constant whose value is quantized. This is typically one of the last /// constant whose value is quantized. This is typically one of the last
/// passes done when lowering to express actual quantized arithmetic in a /// passes done when lowering to express actual quantized arithmetic in a
/// low level representation. Because it modifies the constant, it is /// low level representation. Because it modifies the constant, it is
/// destructive and cannot be undone. /// destructive and cannot be undone.
std::unique_ptr<OpPassBase<FuncOp>> createConvertConstPass(); std::unique_ptr<OpPassBase<FuncOp>> createConvertConstPass();
} // namespace quant } // namespace quant
} // namespace mlir } // namespace mlir
#endif // MLIR_DIALECT_QUANTOPS_PASSES_H #endif // MLIR_DIALECT_QUANT_PASSES_H

mlir/include/mlir/Dialect/Quant/QuantOps.h

  • This file was moved from mlir/include/mlir/Dialect/QuantOps/QuantOps.h.
//===- QuantOps.h - Quantization Ops and Types ------------------*- C++ -*-===// //===- QuantOps.h - Quantization Ops and Types ------------------*- C++ -*-===//
// //
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information. // See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
// //
//===----------------------------------------------------------------------===// //===----------------------------------------------------------------------===//
#ifndef MLIR_DIALECT_QUANTOPS_QUANTOPS_H_ #ifndef MLIR_DIALECT_QUANT_QUANTOPS_H_
#define MLIR_DIALECT_QUANTOPS_QUANTOPS_H_ #define MLIR_DIALECT_QUANT_QUANTOPS_H_
#include "mlir/IR/Attributes.h" #include "mlir/IR/Attributes.h"
#include "mlir/IR/Builders.h" #include "mlir/IR/Builders.h"
#include "mlir/IR/Dialect.h" #include "mlir/IR/Dialect.h"
#include "mlir/IR/OpDefinition.h" #include "mlir/IR/OpDefinition.h"
#include "mlir/IR/StandardTypes.h" #include "mlir/IR/StandardTypes.h"
#include "mlir/IR/Types.h" #include "mlir/IR/Types.h"
#include "mlir/Interfaces/SideEffects.h" #include "mlir/Interfaces/SideEffects.h"
#include "llvm/Support/MathExtras.h" #include "llvm/Support/MathExtras.h"
namespace mlir { namespace mlir {
namespace quant { namespace quant {
#include "mlir/Dialect/QuantOps/QuantOpsDialect.h.inc" #include "mlir/Dialect/Quant/QuantOpsDialect.h.inc"
#define GET_OP_CLASSES #define GET_OP_CLASSES
#include "mlir/Dialect/QuantOps/QuantOps.h.inc" #include "mlir/Dialect/Quant/QuantOps.h.inc"
} // namespace quant } // namespace quant
} // namespace mlir } // namespace mlir
#endif // MLIR_DIALECT_QUANTOPS_QUANTOPS_H_ #endif // MLIR_DIALECT_QUANT_QUANTOPS_H_

mlir/include/mlir/Dialect/Quant/QuantOps.td

  • This file was moved from mlir/include/mlir/Dialect/QuantOps/QuantOps.td.
//===- QuantOps.td - Quantization operation definition -----*- tablegen -*-===// //===- QuantOps.td - Quantization operation definition -----*- tablegen -*-===//
// //
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information. // See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
// //
//===----------------------------------------------------------------------===// //===----------------------------------------------------------------------===//
// //
// This is the operation definition file for Quantization. // This is the operation definition file for Quantization.
// //
//===----------------------------------------------------------------------===// //===----------------------------------------------------------------------===//
#ifndef DIALECT_QUANTOPS_QUANT_OPS_ #ifndef DIALECT_QUANT_QUANT_OPS_
#define DIALECT_QUANTOPS_QUANT_OPS_ #define DIALECT_QUANT_QUANT_OPS_
include "mlir/Dialect/QuantOps/QuantOpsBase.td" include "mlir/Dialect/Quant/QuantOpsBase.td"
include "mlir/Interfaces/SideEffects.td" include "mlir/Interfaces/SideEffects.td"
//===----------------------------------------------------------------------===// //===----------------------------------------------------------------------===//
// Base classes // Base classes
//===----------------------------------------------------------------------===// //===----------------------------------------------------------------------===//
class quant_Op<string mnemonic, list<OpTrait> traits> : class quant_Op<string mnemonic, list<OpTrait> traits> :
Op<Quantization_Dialect, mnemonic, traits>; Op<Quantization_Dialect, mnemonic, traits>;
▲ Show 20 Lines • Show All 251 Lines • ▼ Show 20 Linesdef quant_CoupledRefOp : quant_Op<"coupled_ref", [SameOperandsAndResultType]> {
}]; }];
let arguments = (ins let arguments = (ins
quant_RealValueType:$arg, quant_RealValueType:$arg,
StrAttr:$coupledKey); StrAttr:$coupledKey);
let results = (outs quant_RealValueType); let results = (outs quant_RealValueType);
} }
#endif // DIALECT_QUANTOPS_QUANT_OPS_ #endif // DIALECT_QUANT_QUANT_OPS_

mlir/include/mlir/Dialect/Quant/QuantOpsBase.td

  • This file was moved from mlir/include/mlir/Dialect/QuantOps/QuantOpsBase.td.
//===- QuantOpsBase.td - Quantization dialect base ---------*- tablegen -*-===// //===- QuantOpsBase.td - Quantization dialect base ---------*- tablegen -*-===//
// //
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information. // See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
// //
//===----------------------------------------------------------------------===// //===----------------------------------------------------------------------===//
// //
// Predicates for types in the Quantization dialect. // Predicates for types in the Quantization dialect.
// //
//===----------------------------------------------------------------------===// //===----------------------------------------------------------------------===//
#ifndef DIALECT_QUANTOPS_QUANT_OPS_BASE_ #ifndef DIALECT_QUANT_QUANT_OPS_BASE_
#define DIALECT_QUANTOPS_QUANT_OPS_BASE_ #define DIALECT_QUANT_QUANT_OPS_BASE_
include "mlir/IR/OpBase.td" include "mlir/IR/OpBase.td"
def Quantization_Dialect : Dialect { def Quantization_Dialect : Dialect {
let name = "quant"; let name = "quant";
} }
//===----------------------------------------------------------------------===// //===----------------------------------------------------------------------===//
Show All 40 Linesdef quant_UniformQuantizedType :
DialectType<Quantization_Dialect, DialectType<Quantization_Dialect,
CPred<"$_self.isa<UniformQuantizedType>()">, CPred<"$_self.isa<UniformQuantizedType>()">,
"UniformQuantizedType">; "UniformQuantizedType">;
// Predicate for detecting a container or primitive of UniformQuantizedType. // Predicate for detecting a container or primitive of UniformQuantizedType.
def quant_UniformQuantizedValueType : def quant_UniformQuantizedValueType :
quant_TypedPrimitiveOrContainer<quant_UniformQuantizedType>; quant_TypedPrimitiveOrContainer<quant_UniformQuantizedType>;
#endif // DIALECT_QUANTOPS_QUANT_OPS_BASE_ #endif // DIALECT_QUANT_QUANT_OPS_BASE_

mlir/include/mlir/Dialect/Quant/QuantTypes.h

  • This file was moved from mlir/include/mlir/Dialect/QuantOps/QuantTypes.h.
//===- QuantTypes.h - Quantization Ops and Types ----------------*- C++ -*-===// //===- QuantTypes.h - Quantization Ops and Types ----------------*- C++ -*-===//
// //
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information. // See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
// //
//===----------------------------------------------------------------------===// //===----------------------------------------------------------------------===//
#ifndef MLIR_DIALECT_QUANTOPS_QUANT_TYPES_H_ #ifndef MLIR_DIALECT_QUANT_QUANT_TYPES_H_
#define MLIR_DIALECT_QUANTOPS_QUANT_TYPES_H_ #define MLIR_DIALECT_QUANT_QUANT_TYPES_H_
#include "mlir/IR/Attributes.h" #include "mlir/IR/Attributes.h"
#include "mlir/IR/Builders.h" #include "mlir/IR/Builders.h"
#include "mlir/IR/Dialect.h" #include "mlir/IR/Dialect.h"
#include "mlir/IR/OpDefinition.h" #include "mlir/IR/OpDefinition.h"
#include "mlir/IR/StandardTypes.h" #include "mlir/IR/StandardTypes.h"
#include "mlir/IR/Types.h" #include "mlir/IR/Types.h"
#include "llvm/Support/MathExtras.h" #include "llvm/Support/MathExtras.h"
▲ Show 20 Lines • Show All 375 Lines • ▼ Show 20 Lines bool isFixedPoint() const {
return llvm::all_of(getZeroPoints(), return llvm::all_of(getZeroPoints(),
[](int64_t zeroPoint) { return zeroPoint != 0; }); [](int64_t zeroPoint) { return zeroPoint != 0; });
} }
}; };
} // namespace quant } // namespace quant
} // namespace mlir } // namespace mlir
#endif // MLIR_DIALECT_QUANTOPS_QUANT_TYPES_H_ #endif // MLIR_DIALECT_QUANT_QUANT_TYPES_H_

mlir/include/mlir/Dialect/Quant/QuantizeUtils.h

  • This file was moved from mlir/include/mlir/Dialect/QuantOps/QuantizeUtils.h.
//===- QuantizeUtils.h - Support utilities for quantization -----*- C++ -*-===// //===- QuantizeUtils.h - Support utilities for quantization -----*- C++ -*-===//
// //
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information. // See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
// //
//===----------------------------------------------------------------------===// //===----------------------------------------------------------------------===//
#ifndef MLIR_DIALECT_QUANTOPS_QUANTIZEUTILS_H_ #ifndef MLIR_DIALECT_QUANT_QUANTIZEUTILS_H_
#define MLIR_DIALECT_QUANTOPS_QUANTIZEUTILS_H_ #define MLIR_DIALECT_QUANT_QUANTIZEUTILS_H_
namespace mlir { namespace mlir {
class Attribute; class Attribute;
class Type; class Type;
namespace quant { namespace quant {
class QuantizedType; class QuantizedType;
class UniformQuantizedType; class UniformQuantizedType;
Show All 34 Lines
/// -> (DenseElementsAttr[tensor<2x2xi8>], outConvertedType: tensor<2x2xi8>) /// -> (DenseElementsAttr[tensor<2x2xi8>], outConvertedType: tensor<2x2xi8>)
Attribute quantizeAttrUniform(Attribute realValue, Attribute quantizeAttrUniform(Attribute realValue,
UniformQuantizedType quantizedElementType, UniformQuantizedType quantizedElementType,
const UniformQuantizedValueConverter &converter, const UniformQuantizedValueConverter &converter,
Type &outConvertedType); Type &outConvertedType);
} // namespace quant } // namespace quant
} // namespace mlir } // namespace mlir
#endif // MLIR_DIALECT_QUANTOPS_QUANTIZEUTILS_H_ #endif // MLIR_DIALECT_QUANT_QUANTIZEUTILS_H_

mlir/include/mlir/Dialect/Quant/UniformSupport.h

  • This file was moved from mlir/include/mlir/Dialect/QuantOps/UniformSupport.h.
//===- UniformSupport.h - Support utilities for uniform quant ---*- C++ -*-===// //===- UniformSupport.h - Support utilities for uniform quant ---*- C++ -*-===//
// //
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information. // See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
// //
//===----------------------------------------------------------------------===// //===----------------------------------------------------------------------===//
#ifndef MLIR_DIALECT_QUANTOPS_UNIFORMSUPPORT_H_ #ifndef MLIR_DIALECT_QUANT_UNIFORMSUPPORT_H_
#define MLIR_DIALECT_QUANTOPS_UNIFORMSUPPORT_H_ #define MLIR_DIALECT_QUANT_UNIFORMSUPPORT_H_
#include "mlir/Dialect/QuantOps/QuantTypes.h" #include "mlir/Dialect/Quant/QuantTypes.h"
#include "mlir/IR/StandardTypes.h" #include "mlir/IR/StandardTypes.h"
#include "mlir/IR/Types.h" #include "mlir/IR/Types.h"
#include "llvm/ADT/APFloat.h" #include "llvm/ADT/APFloat.h"
#include "llvm/ADT/APInt.h" #include "llvm/ADT/APInt.h"
#include "llvm/ADT/APSInt.h" #include "llvm/ADT/APSInt.h"
namespace mlir { namespace mlir {
namespace quant { namespace quant {
▲ Show 20 Lines • Show All 189 Lines • ▼ Show 20 Linesprivate:
const uint32_t storageBitWidth; const uint32_t storageBitWidth;
const bool isSigned; const bool isSigned;
int32_t quantizationDim; int32_t quantizationDim;
}; };
} // namespace quant } // namespace quant
} // namespace mlir } // namespace mlir
#endif // MLIR_DIALECT_QUANTOPS_UNIFORMSUPPORT_H_ #endif // MLIR_DIALECT_QUANT_UNIFORMSUPPORT_H_

mlir/include/mlir/Dialect/QuantOps/CMakeLists.txt

  • This file was moved to mlir/include/mlir/Dialect/Quant/CMakeLists.txt.

mlir/include/mlir/Dialect/QuantOps/FakeQuantSupport.h

  • This file was moved to mlir/include/mlir/Dialect/Quant/FakeQuantSupport.h.

mlir/include/mlir/Dialect/QuantOps/Passes.h

  • This file was moved to mlir/include/mlir/Dialect/Quant/Passes.h.

mlir/include/mlir/Dialect/QuantOps/QuantOps.h

  • This file was moved to mlir/include/mlir/Dialect/Quant/QuantOps.h.

mlir/include/mlir/Dialect/QuantOps/QuantOps.td

  • This file was moved to mlir/include/mlir/Dialect/Quant/QuantOps.td.

mlir/include/mlir/Dialect/QuantOps/QuantOpsBase.td

  • This file was moved to mlir/include/mlir/Dialect/Quant/QuantOpsBase.td.

mlir/include/mlir/Dialect/QuantOps/QuantTypes.h

  • This file was moved to mlir/include/mlir/Dialect/Quant/QuantTypes.h.

mlir/include/mlir/Dialect/QuantOps/QuantizeUtils.h

  • This file was moved to mlir/include/mlir/Dialect/Quant/QuantizeUtils.h.

mlir/include/mlir/Dialect/QuantOps/UniformSupport.h

  • This file was moved to mlir/include/mlir/Dialect/Quant/UniformSupport.h.

mlir/include/mlir/InitAllDialects.h

Show All 17 Lines
#include "mlir/Dialect/FxpMathOps/FxpMathOps.h" #include "mlir/Dialect/FxpMathOps/FxpMathOps.h"
#include "mlir/Dialect/GPU/GPUDialect.h" #include "mlir/Dialect/GPU/GPUDialect.h"
#include "mlir/Dialect/LLVMIR/LLVMDialect.h" #include "mlir/Dialect/LLVMIR/LLVMDialect.h"
#include "mlir/Dialect/LLVMIR/NVVMDialect.h" #include "mlir/Dialect/LLVMIR/NVVMDialect.h"
#include "mlir/Dialect/LLVMIR/ROCDLDialect.h" #include "mlir/Dialect/LLVMIR/ROCDLDialect.h"
#include "mlir/Dialect/Linalg/IR/LinalgOps.h" #include "mlir/Dialect/Linalg/IR/LinalgOps.h"
#include "mlir/Dialect/LoopOps/LoopOps.h" #include "mlir/Dialect/LoopOps/LoopOps.h"
#include "mlir/Dialect/OpenMP/OpenMPDialect.h" #include "mlir/Dialect/OpenMP/OpenMPDialect.h"
#include "mlir/Dialect/QuantOps/QuantOps.h" #include "mlir/Dialect/Quant/QuantOps.h"
#include "mlir/Dialect/SDBM/SDBMDialect.h" #include "mlir/Dialect/SDBM/SDBMDialect.h"
#include "mlir/Dialect/SPIRV/SPIRVDialect.h" #include "mlir/Dialect/SPIRV/SPIRVDialect.h"
#include "mlir/Dialect/Shape/IR/Shape.h" #include "mlir/Dialect/Shape/IR/Shape.h"
#include "mlir/Dialect/StandardOps/IR/Ops.h" #include "mlir/Dialect/StandardOps/IR/Ops.h"
#include "mlir/Dialect/VectorOps/VectorOps.h" #include "mlir/Dialect/VectorOps/VectorOps.h"
#include "mlir/IR/Dialect.h" #include "mlir/IR/Dialect.h"
namespace mlir { namespace mlir {
Show All 27 Lines

mlir/include/mlir/InitAllPasses.h

Show All 23 Lines
#include "mlir/Conversion/LinalgToSPIRV/LinalgToSPIRVPass.h" #include "mlir/Conversion/LinalgToSPIRV/LinalgToSPIRVPass.h"
#include "mlir/Conversion/LoopsToGPU/LoopsToGPUPass.h" #include "mlir/Conversion/LoopsToGPU/LoopsToGPUPass.h"
#include "mlir/Conversion/StandardToSPIRV/ConvertStandardToSPIRVPass.h" #include "mlir/Conversion/StandardToSPIRV/ConvertStandardToSPIRVPass.h"
#include "mlir/Dialect/FxpMathOps/Passes.h" #include "mlir/Dialect/FxpMathOps/Passes.h"
#include "mlir/Dialect/GPU/Passes.h" #include "mlir/Dialect/GPU/Passes.h"
#include "mlir/Dialect/LLVMIR/Transforms/LegalizeForExport.h" #include "mlir/Dialect/LLVMIR/Transforms/LegalizeForExport.h"
#include "mlir/Dialect/Linalg/Passes.h" #include "mlir/Dialect/Linalg/Passes.h"
#include "mlir/Dialect/LoopOps/Passes.h" #include "mlir/Dialect/LoopOps/Passes.h"
#include "mlir/Dialect/QuantOps/Passes.h" #include "mlir/Dialect/Quant/Passes.h"
#include "mlir/Dialect/SPIRV/Passes.h" #include "mlir/Dialect/SPIRV/Passes.h"
#include "mlir/Quantizer/Transforms/Passes.h" #include "mlir/Quantizer/Transforms/Passes.h"
#include "mlir/Transforms/LocationSnapshot.h" #include "mlir/Transforms/LocationSnapshot.h"
#include "mlir/Transforms/Passes.h" #include "mlir/Transforms/Passes.h"
#include <cstdlib> #include <cstdlib>
namespace mlir { namespace mlir {
▲ Show 20 Lines • Show All 102 Lines • Show Last 20 Lines

mlir/include/mlir/Quantizer/Support/Configuration.h

Show All 11 Lines
// //
//===----------------------------------------------------------------------===// //===----------------------------------------------------------------------===//
#ifndef MLIR_QUANTIZER_SUPPORT_CONFIGURATION_H #ifndef MLIR_QUANTIZER_SUPPORT_CONFIGURATION_H
#define MLIR_QUANTIZER_SUPPORT_CONFIGURATION_H #define MLIR_QUANTIZER_SUPPORT_CONFIGURATION_H
#include <functional> #include <functional>
#include "mlir/Dialect/QuantOps/QuantTypes.h" #include "mlir/Dialect/Quant/QuantTypes.h"
#include "mlir/IR/Identifier.h" #include "mlir/IR/Identifier.h"
#include "mlir/Quantizer/Support/ConstraintAnalysisGraph.h" #include "mlir/Quantizer/Support/ConstraintAnalysisGraph.h"
#include "mlir/Quantizer/Support/Metadata.h" #include "mlir/Quantizer/Support/Metadata.h"
#include "mlir/Quantizer/Support/Rules.h" #include "mlir/Quantizer/Support/Rules.h"
#include "llvm/ADT/DenseMap.h" #include "llvm/ADT/DenseMap.h"
#include "llvm/ADT/SmallBitVector.h" #include "llvm/ADT/SmallBitVector.h"
#include "llvm/ADT/StringSet.h" #include "llvm/ADT/StringSet.h"
▲ Show 20 Lines • Show All 118 Lines • Show Last 20 Lines

mlir/include/mlir/Quantizer/Support/Metadata.h

Show All 11 Lines
// //
//===----------------------------------------------------------------------===// //===----------------------------------------------------------------------===//
#ifndef MLIR_QUANTIZER_SUPPORT_METADATA_H #ifndef MLIR_QUANTIZER_SUPPORT_METADATA_H
#define MLIR_QUANTIZER_SUPPORT_METADATA_H #define MLIR_QUANTIZER_SUPPORT_METADATA_H
#include <limits> #include <limits>
#include "mlir/Dialect/QuantOps/QuantTypes.h" #include "mlir/Dialect/Quant/QuantTypes.h"
#include "mlir/IR/MLIRContext.h" #include "mlir/IR/MLIRContext.h"
#include "mlir/Quantizer/Support/Rules.h" #include "mlir/Quantizer/Support/Rules.h"
#include "llvm/ADT/SmallBitVector.h" #include "llvm/ADT/SmallBitVector.h"
namespace mlir { namespace mlir {
namespace quantizer { namespace quantizer {
class SolverContext { class SolverContext {
▲ Show 20 Lines • Show All 73 Lines • Show Last 20 Lines

mlir/lib/Dialect/CMakeLists.txt

add_subdirectory(AffineOps) add_subdirectory(AffineOps)
add_subdirectory(FxpMathOps) add_subdirectory(FxpMathOps)
add_subdirectory(GPU) add_subdirectory(GPU)
add_subdirectory(Linalg) add_subdirectory(Linalg)
add_subdirectory(LLVMIR) add_subdirectory(LLVMIR)
add_subdirectory(LoopOps) add_subdirectory(LoopOps)
add_subdirectory(OpenMP) add_subdirectory(OpenMP)
add_subdirectory(QuantOps) add_subdirectory(Quant)
add_subdirectory(SDBM) add_subdirectory(SDBM)
add_subdirectory(Shape) add_subdirectory(Shape)
add_subdirectory(SPIRV) add_subdirectory(SPIRV)
add_subdirectory(StandardOps) add_subdirectory(StandardOps)
add_subdirectory(VectorOps) add_subdirectory(VectorOps)
set(LLVM_OPTIONAL_SOURCES set(LLVM_OPTIONAL_SOURCES
Traits.cpp Traits.cpp
Show All 12 Lines

mlir/lib/Dialect/FxpMathOps/IR/FxpMathOps.cpp

//===- FxpMathOps.cpp - Op implementation for FxpMathOps ------------------===// //===- FxpMathOps.cpp - Op implementation for FxpMathOps ------------------===//
// //
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information. // See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
// //
//===----------------------------------------------------------------------===// //===----------------------------------------------------------------------===//
#include "mlir/Dialect/FxpMathOps/FxpMathOps.h" #include "mlir/Dialect/FxpMathOps/FxpMathOps.h"
#include "mlir/Dialect/QuantOps/QuantTypes.h" #include "mlir/Dialect/Quant/QuantTypes.h"
#include "mlir/IR/MLIRContext.h" #include "mlir/IR/MLIRContext.h"
#include "mlir/IR/StandardTypes.h" #include "mlir/IR/StandardTypes.h"
#include "llvm/ADT/StringRef.h" #include "llvm/ADT/StringRef.h"
#include "llvm/ADT/Twine.h" #include "llvm/ADT/Twine.h"
#include "llvm/Support/MathExtras.h" #include "llvm/Support/MathExtras.h"
using namespace mlir; using namespace mlir;
using namespace mlir::fxpmath; using namespace mlir::fxpmath;
Show All 11 Lines

mlir/lib/Dialect/FxpMathOps/Transforms/UniformKernelUtils.h

//===- UniformKernelUtils.h - Utilities for lowering uniform math - C++ -*-===// //===- UniformKernelUtils.h - Utilities for lowering uniform math - C++ -*-===//
// //
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information. // See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
// //
//===----------------------------------------------------------------------===// //===----------------------------------------------------------------------===//
#ifndef MLIR_FXPMATH_UNIFORM_KERNEL_UTILS_H_ #ifndef MLIR_FXPMATH_UNIFORM_KERNEL_UTILS_H_
#define MLIR_FXPMATH_UNIFORM_KERNEL_UTILS_H_ #define MLIR_FXPMATH_UNIFORM_KERNEL_UTILS_H_
#include "mlir/Dialect/QuantOps/QuantOps.h" #include "mlir/Dialect/Quant/QuantOps.h"
#include "mlir/Dialect/QuantOps/QuantTypes.h" #include "mlir/Dialect/Quant/QuantTypes.h"
#include "mlir/Dialect/QuantOps/UniformSupport.h" #include "mlir/Dialect/Quant/UniformSupport.h"
#include "mlir/IR/Operation.h" #include "mlir/IR/Operation.h"
#include <cmath> #include <cmath>
namespace mlir { namespace mlir {
namespace fxpmath { namespace fxpmath {
namespace detail { namespace detail {
▲ Show 20 Lines • Show All 205 Lines • Show Last 20 Lines

mlir/lib/Dialect/Quant/CMakeLists.txt

  • This file was moved from mlir/lib/Dialect/QuantOps/CMakeLists.txt.
The contents of this file were not changed.

mlir/lib/Dialect/Quant/IR/QuantOps.cpp

  • This file was moved from mlir/lib/Dialect/QuantOps/IR/QuantOps.cpp.
//===- QuantOps.cpp - Quantization Type and Ops Implementation --*- C++ -*-===// //===- QuantOps.cpp - Quantization Type and Ops Implementation --*- C++ -*-===//
// //
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information. // See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
// //
//===----------------------------------------------------------------------===// //===----------------------------------------------------------------------===//
#include "mlir/Dialect/QuantOps/QuantOps.h" #include "mlir/Dialect/Quant/QuantOps.h"
#include "TypeDetail.h" #include "TypeDetail.h"
#include "mlir/Dialect/QuantOps/QuantTypes.h" #include "mlir/Dialect/Quant/QuantTypes.h"
#include "mlir/IR/MLIRContext.h" #include "mlir/IR/MLIRContext.h"
#include "mlir/IR/Matchers.h" #include "mlir/IR/Matchers.h"
#include "mlir/IR/PatternMatch.h" #include "mlir/IR/PatternMatch.h"
#include "mlir/IR/StandardTypes.h" #include "mlir/IR/StandardTypes.h"
#include "llvm/ADT/StringRef.h" #include "llvm/ADT/StringRef.h"
#include "llvm/ADT/Twine.h" #include "llvm/ADT/Twine.h"
#include "llvm/Support/MathExtras.h" #include "llvm/Support/MathExtras.h"
#include <numeric> #include <numeric>
using namespace mlir; using namespace mlir;
using namespace mlir::quant; using namespace mlir::quant;
using namespace mlir::quant::detail; using namespace mlir::quant::detail;
QuantizationDialect::QuantizationDialect(MLIRContext *context) QuantizationDialect::QuantizationDialect(MLIRContext *context)
: Dialect(/*name=*/"quant", context) { : Dialect(/*name=*/"quant", context) {
addTypes<AnyQuantizedType, UniformQuantizedType, addTypes<AnyQuantizedType, UniformQuantizedType,
UniformQuantizedPerAxisType>(); UniformQuantizedPerAxisType>();
addOperations< addOperations<
#define GET_OP_LIST #define GET_OP_LIST
#include "mlir/Dialect/QuantOps/QuantOps.cpp.inc" #include "mlir/Dialect/Quant/QuantOps.cpp.inc"
>(); >();
} }
OpFoldResult StorageCastOp::fold(ArrayRef<Attribute> operands) { OpFoldResult StorageCastOp::fold(ArrayRef<Attribute> operands) {
// Matches x -> [scast -> scast] -> y, replacing the second scast with the // Matches x -> [scast -> scast] -> y, replacing the second scast with the
// value of x if the casts invert each other. // value of x if the casts invert each other.
auto srcScastOp = dyn_cast_or_null<StorageCastOp>(arg().getDefiningOp()); auto srcScastOp = dyn_cast_or_null<StorageCastOp>(arg().getDefiningOp());
if (!srcScastOp || srcScastOp.arg().getType() != getType()) if (!srcScastOp || srcScastOp.arg().getType() != getType())
▲ Show 20 Lines • Show All 47 Lines • ▼ Show 20 Lines if (!isValidQuantizationSpec(outputSpec, outputType)) {
return op.emitOpError() << "has incompatible specification " << outputSpec return op.emitOpError() << "has incompatible specification " << outputSpec
<< " and output type " << outputType; << " and output type " << outputType;
} }
} }
return success(); return success();
} }
#define GET_OP_CLASSES #define GET_OP_CLASSES
#include "mlir/Dialect/QuantOps/QuantOps.cpp.inc" #include "mlir/Dialect/Quant/QuantOps.cpp.inc"

mlir/lib/Dialect/Quant/IR/QuantTypes.cpp

  • This file was moved from mlir/lib/Dialect/QuantOps/IR/QuantTypes.cpp.
//===- QuantOps.cpp - Quantization Type and Ops Implementation --*- C++ -*-===// //===- QuantOps.cpp - Quantization Type and Ops Implementation --*- C++ -*-===//
// //
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information. // See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
// //
//===----------------------------------------------------------------------===// //===----------------------------------------------------------------------===//
#include "mlir/Dialect/QuantOps/QuantTypes.h" #include "mlir/Dialect/Quant/QuantTypes.h"
#include "TypeDetail.h" #include "TypeDetail.h"
#include "mlir/IR/MLIRContext.h" #include "mlir/IR/MLIRContext.h"
#include "mlir/IR/StandardTypes.h" #include "mlir/IR/StandardTypes.h"
#include "llvm/ADT/StringRef.h" #include "llvm/ADT/StringRef.h"
#include "llvm/ADT/Twine.h" #include "llvm/ADT/Twine.h"
#include "llvm/Support/MathExtras.h" #include "llvm/Support/MathExtras.h"
▲ Show 20 Lines • Show All 338 Lines • Show Last 20 Lines

mlir/lib/Dialect/Quant/IR/TypeDetail.h

  • This file was moved from mlir/lib/Dialect/QuantOps/IR/TypeDetail.h.
The contents of this file were not changed.

mlir/lib/Dialect/Quant/IR/TypeParser.cpp

  • This file was moved from mlir/lib/Dialect/QuantOps/IR/TypeParser.cpp.
//===- TypeParser.h - Quantization Type Parser ------------------*- C++ -*-===// //===- TypeParser.h - Quantization Type Parser ------------------*- C++ -*-===//
// //
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information. // See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
// //
//===----------------------------------------------------------------------===// //===----------------------------------------------------------------------===//
#include "mlir/Dialect/QuantOps/QuantOps.h" #include "mlir/Dialect/Quant/QuantOps.h"
#include "mlir/Dialect/QuantOps/QuantTypes.h" #include "mlir/Dialect/Quant/QuantTypes.h"
#include "mlir/IR/DialectImplementation.h" #include "mlir/IR/DialectImplementation.h"
#include "mlir/IR/Location.h" #include "mlir/IR/Location.h"
#include "mlir/IR/StandardTypes.h" #include "mlir/IR/StandardTypes.h"
#include "mlir/IR/Types.h" #include "mlir/IR/Types.h"
#include "llvm/ADT/APFloat.h" #include "llvm/ADT/APFloat.h"
#include "llvm/ADT/StringSwitch.h" #include "llvm/ADT/StringSwitch.h"
#include "llvm/Support/Format.h" #include "llvm/Support/Format.h"
#include "llvm/Support/MathExtras.h" #include "llvm/Support/MathExtras.h"
▲ Show 20 Lines • Show All 364 Lines • Show Last 20 Lines

mlir/lib/Dialect/Quant/Transforms/ConvertConst.cpp

  • This file was moved from mlir/lib/Dialect/QuantOps/Transforms/ConvertConst.cpp.
//===- ConvertConst.cpp - Quantizes constant ops --------------------------===// //===- ConvertConst.cpp - Quantizes constant ops --------------------------===//
// //
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information. // See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
// //
//===----------------------------------------------------------------------===// //===----------------------------------------------------------------------===//
#include "mlir/Dialect/QuantOps/Passes.h" #include "mlir/Dialect/Quant/Passes.h"
#include "mlir/Dialect/QuantOps/QuantOps.h" #include "mlir/Dialect/Quant/QuantOps.h"
#include "mlir/Dialect/QuantOps/QuantizeUtils.h" #include "mlir/Dialect/Quant/QuantizeUtils.h"
#include "mlir/Dialect/QuantOps/UniformSupport.h" #include "mlir/Dialect/Quant/UniformSupport.h"
#include "mlir/Dialect/StandardOps/IR/Ops.h" #include "mlir/Dialect/StandardOps/IR/Ops.h"
#include "mlir/IR/Attributes.h" #include "mlir/IR/Attributes.h"
#include "mlir/IR/Matchers.h" #include "mlir/IR/Matchers.h"
#include "mlir/IR/PatternMatch.h" #include "mlir/IR/PatternMatch.h"
#include "mlir/IR/StandardTypes.h" #include "mlir/IR/StandardTypes.h"
#include "mlir/Pass/Pass.h" #include "mlir/Pass/Pass.h"
using namespace mlir; using namespace mlir;
▲ Show 20 Lines • Show All 92 Lines • Show Last 20 Lines

mlir/lib/Dialect/Quant/Transforms/ConvertSimQuant.cpp

  • This file was moved from mlir/lib/Dialect/QuantOps/Transforms/ConvertSimQuant.cpp.
//===- ConvertSimQuant.cpp - Converts simulated quant ops------------------===// //===- ConvertSimQuant.cpp - Converts simulated quant ops------------------===//
// //
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information. // See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
// //
//===----------------------------------------------------------------------===// //===----------------------------------------------------------------------===//
#include "mlir/Dialect/QuantOps/FakeQuantSupport.h" #include "mlir/Dialect/Quant/FakeQuantSupport.h"
#include "mlir/Dialect/QuantOps/Passes.h" #include "mlir/Dialect/Quant/Passes.h"
#include "mlir/Dialect/QuantOps/QuantOps.h" #include "mlir/Dialect/Quant/QuantOps.h"
#include "mlir/Dialect/QuantOps/UniformSupport.h" #include "mlir/Dialect/Quant/UniformSupport.h"
#include "mlir/IR/Attributes.h" #include "mlir/IR/Attributes.h"
#include "mlir/IR/PatternMatch.h" #include "mlir/IR/PatternMatch.h"
#include "mlir/IR/StandardTypes.h" #include "mlir/IR/StandardTypes.h"
#include "mlir/Pass/Pass.h" #include "mlir/Pass/Pass.h"
using namespace mlir; using namespace mlir;
using namespace mlir::quant; using namespace mlir::quant;
▲ Show 20 Lines • Show All 129 Lines • Show Last 20 Lines

mlir/lib/Dialect/Quant/Utils/FakeQuantSupport.cpp

  • This file was moved from mlir/lib/Dialect/QuantOps/Utils/FakeQuantSupport.cpp.
//===- FakeQuantSupport.cpp - Support utilities for FakeQuant ops ---------===// //===- FakeQuantSupport.cpp - Support utilities for FakeQuant ops ---------===//
// //
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information. // See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
// //
//===----------------------------------------------------------------------===// //===----------------------------------------------------------------------===//
#include "mlir/Dialect/QuantOps/FakeQuantSupport.h" #include "mlir/Dialect/Quant/FakeQuantSupport.h"
#include "mlir/Dialect/QuantOps/QuantTypes.h" #include "mlir/Dialect/Quant/QuantTypes.h"
using namespace mlir; using namespace mlir;
using namespace mlir::quant; using namespace mlir::quant;
static bool getDefaultStorageParams(unsigned numBits, bool narrowRange, static bool getDefaultStorageParams(unsigned numBits, bool narrowRange,
bool isSigned, MLIRContext *ctx, bool isSigned, MLIRContext *ctx,
Type &storageType, int64_t &qmin, Type &storageType, int64_t &qmin,
int64_t &qmax) { int64_t &qmax) {
▲ Show 20 Lines • Show All 157 Lines • Show Last 20 Lines

mlir/lib/Dialect/Quant/Utils/QuantizeUtils.cpp

  • This file was moved from mlir/lib/Dialect/QuantOps/Utils/QuantizeUtils.cpp.
//===- QuantizeUtils.cpp - Support utilities for quantization -------------===// //===- QuantizeUtils.cpp - Support utilities for quantization -------------===//
// //
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information. // See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
// //
//===----------------------------------------------------------------------===// //===----------------------------------------------------------------------===//
#include "mlir/Dialect/QuantOps/QuantizeUtils.h" #include "mlir/Dialect/Quant/QuantizeUtils.h"
#include "mlir/Dialect/QuantOps/UniformSupport.h" #include "mlir/Dialect/Quant/UniformSupport.h"
#include "mlir/IR/Attributes.h" #include "mlir/IR/Attributes.h"
#include "mlir/IR/StandardTypes.h" #include "mlir/IR/StandardTypes.h"
using namespace mlir; using namespace mlir;
using namespace mlir::quant; using namespace mlir::quant;
/// Converts a possible primitive, real expressed value attribute to a /// Converts a possible primitive, real expressed value attribute to a
/// corresponding storage attribute (typically FloatAttr -> IntegerAttr). /// corresponding storage attribute (typically FloatAttr -> IntegerAttr).
▲ Show 20 Lines • Show All 130 Lines • Show Last 20 Lines

mlir/lib/Dialect/Quant/Utils/UniformSupport.cpp

  • This file was moved from mlir/lib/Dialect/QuantOps/Utils/UniformSupport.cpp.
//===- UniformSupport.cpp - Support utilities for uniform quant -----------===// //===- UniformSupport.cpp - Support utilities for uniform quant -----------===//
// //
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information. // See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
// //
//===----------------------------------------------------------------------===// //===----------------------------------------------------------------------===//
#include "mlir/Dialect/QuantOps/UniformSupport.h" #include "mlir/Dialect/Quant/UniformSupport.h"
#include "mlir/IR/StandardTypes.h" #include "mlir/IR/StandardTypes.h"
#include <numeric> #include <numeric>
using namespace mlir; using namespace mlir;
using namespace mlir::quant; using namespace mlir::quant;
static bool isQuantizablePrimitiveType(Type inputType) { static bool isQuantizablePrimitiveType(Type inputType) {
return inputType.isa<FloatType>(); return inputType.isa<FloatType>();
▲ Show 20 Lines • Show All 85 Lines • Show Last 20 Lines

mlir/lib/Dialect/QuantOps/CMakeLists.txt

  • This file was moved to mlir/lib/Dialect/Quant/CMakeLists.txt.

mlir/lib/Dialect/QuantOps/IR/QuantOps.cpp

  • This file was moved to mlir/lib/Dialect/Quant/IR/QuantOps.cpp.

mlir/lib/Dialect/QuantOps/IR/QuantTypes.cpp

  • This file was moved to mlir/lib/Dialect/Quant/IR/QuantTypes.cpp.

mlir/lib/Dialect/QuantOps/IR/TypeDetail.h

  • This file was moved to mlir/lib/Dialect/Quant/IR/TypeDetail.h.

mlir/lib/Dialect/QuantOps/IR/TypeParser.cpp

  • This file was moved to mlir/lib/Dialect/Quant/IR/TypeParser.cpp.

mlir/lib/Dialect/QuantOps/Transforms/ConvertConst.cpp

  • This file was moved to mlir/lib/Dialect/Quant/Transforms/ConvertConst.cpp.

mlir/lib/Dialect/QuantOps/Transforms/ConvertSimQuant.cpp

  • This file was moved to mlir/lib/Dialect/Quant/Transforms/ConvertSimQuant.cpp.

mlir/lib/Dialect/QuantOps/Utils/FakeQuantSupport.cpp

  • This file was moved to mlir/lib/Dialect/Quant/Utils/FakeQuantSupport.cpp.

mlir/lib/Dialect/QuantOps/Utils/QuantizeUtils.cpp

  • This file was moved to mlir/lib/Dialect/Quant/Utils/QuantizeUtils.cpp.

mlir/lib/Dialect/QuantOps/Utils/UniformSupport.cpp

  • This file was moved to mlir/lib/Dialect/Quant/Utils/UniformSupport.cpp.

mlir/lib/Quantizer/Configurations/FxpMathConfig.cpp

Show All 9 Lines
// quantization scheme based on the FxpMathOps (plus a small category of // quantization scheme based on the FxpMathOps (plus a small category of
// extension ops that can be added from other dialects). // extension ops that can be added from other dialects).
// //
//===----------------------------------------------------------------------===// //===----------------------------------------------------------------------===//
#include "mlir/Quantizer/Configurations/FxpMathConfig.h" #include "mlir/Quantizer/Configurations/FxpMathConfig.h"
#include "mlir/Dialect/FxpMathOps/FxpMathOps.h" #include "mlir/Dialect/FxpMathOps/FxpMathOps.h"
#include "mlir/Dialect/QuantOps/QuantOps.h" #include "mlir/Dialect/Quant/QuantOps.h"
#include "mlir/Dialect/QuantOps/QuantTypes.h" #include "mlir/Dialect/Quant/QuantTypes.h"
#include "mlir/Dialect/StandardOps/IR/Ops.h" #include "mlir/Dialect/StandardOps/IR/Ops.h"
#include "mlir/IR/Matchers.h" #include "mlir/IR/Matchers.h"
#include "mlir/IR/StandardTypes.h" #include "mlir/IR/StandardTypes.h"
#include "mlir/Quantizer/Support/ConstraintAnalysisGraph.h" #include "mlir/Quantizer/Support/ConstraintAnalysisGraph.h"
#include "mlir/Quantizer/Support/Metadata.h" #include "mlir/Quantizer/Support/Metadata.h"
#include "mlir/Quantizer/Support/Statistics.h" #include "mlir/Quantizer/Support/Statistics.h"
#include "mlir/Quantizer/Support/UniformConstraints.h" #include "mlir/Quantizer/Support/UniformConstraints.h"
▲ Show 20 Lines • Show All 251 Lines • Show Last 20 Lines

mlir/lib/Quantizer/Support/UniformConstraints.cpp

//===- UniformConstraints.cpp - Constraints for uniform quant -------------===// //===- UniformConstraints.cpp - Constraints for uniform quant -------------===//
// //
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information. // See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
// //
//===----------------------------------------------------------------------===// //===----------------------------------------------------------------------===//
#include "mlir/Quantizer/Support/UniformConstraints.h" #include "mlir/Quantizer/Support/UniformConstraints.h"
#include "mlir/Dialect/QuantOps/QuantTypes.h" #include "mlir/Dialect/Quant/QuantTypes.h"
#include "mlir/IR/Diagnostics.h" #include "mlir/IR/Diagnostics.h"
#include "mlir/IR/Location.h" #include "mlir/IR/Location.h"
#include "mlir/IR/MLIRContext.h" #include "mlir/IR/MLIRContext.h"
#include "mlir/Quantizer/Support/Configuration.h" #include "mlir/Quantizer/Support/Configuration.h"
#include "mlir/Quantizer/Support/ConstraintAnalysisGraph.h" #include "mlir/Quantizer/Support/ConstraintAnalysisGraph.h"
#include "mlir/Quantizer/Support/Metadata.h" #include "mlir/Quantizer/Support/Metadata.h"
#include "mlir/Quantizer/Support/Rules.h" #include "mlir/Quantizer/Support/Rules.h"
#include "mlir/Quantizer/Support/TypeUtils.h" #include "mlir/Quantizer/Support/TypeUtils.h"
▲ Show 20 Lines • Show All 237 Lines • Show Last 20 Lines

mlir/lib/Quantizer/Transforms/AddDefaultStatsTestPass.cpp

//===- AddDefaultStatsTestPass.cpp - Testing pass to add default stats ----===// //===- AddDefaultStatsTestPass.cpp - Testing pass to add default stats ----===//
// //
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information. // See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
// //
//===----------------------------------------------------------------------===// //===----------------------------------------------------------------------===//
// //
// This file defines a testing pass to add default statistics nodes to every // This file defines a testing pass to add default statistics nodes to every
// quantization eligible op. Useful for unit testing. // quantization eligible op. Useful for unit testing.
// //
//===----------------------------------------------------------------------===// //===----------------------------------------------------------------------===//
#include "mlir/Dialect/QuantOps/QuantOps.h" #include "mlir/Dialect/Quant/QuantOps.h"
#include "mlir/Dialect/QuantOps/QuantTypes.h" #include "mlir/Dialect/Quant/QuantTypes.h"
#include "mlir/IR/Attributes.h" #include "mlir/IR/Attributes.h"
#include "mlir/IR/Builders.h" #include "mlir/IR/Builders.h"
#include "mlir/Quantizer/Configurations/FxpMathConfig.h" #include "mlir/Quantizer/Configurations/FxpMathConfig.h"
#include "mlir/Quantizer/Support/Configuration.h" #include "mlir/Quantizer/Support/Configuration.h"
#include "mlir/Quantizer/Support/ConstraintAnalysisGraph.h" #include "mlir/Quantizer/Support/ConstraintAnalysisGraph.h"
#include "mlir/Quantizer/Support/ConstraintAnalysisGraphTraits.h" #include "mlir/Quantizer/Support/ConstraintAnalysisGraphTraits.h"
#include "mlir/Quantizer/Transforms/Passes.h" #include "mlir/Quantizer/Transforms/Passes.h"
#include "mlir/Support/LogicalResult.h" #include "mlir/Support/LogicalResult.h"
▲ Show 20 Lines • Show All 97 Lines • Show Last 20 Lines

mlir/lib/Quantizer/Transforms/InferQuantizedTypesPass.cpp

//===- InferQuantizedTypesPass.cpp - Infers quantized types ---------------===// //===- InferQuantizedTypesPass.cpp - Infers quantized types ---------------===//
// //
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information. // See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
// //
//===----------------------------------------------------------------------===// //===----------------------------------------------------------------------===//
// //
// This file defines the primary pass for instantiating a CAG, running it to // This file defines the primary pass for instantiating a CAG, running it to
// convergence on a module to determine eligible quantized type transforms, and // convergence on a module to determine eligible quantized type transforms, and
// applying those transforms to the IR. // applying those transforms to the IR.
// //
//===----------------------------------------------------------------------===// //===----------------------------------------------------------------------===//
#include "mlir/Dialect/QuantOps/QuantOps.h" #include "mlir/Dialect/Quant/QuantOps.h"
#include "mlir/Dialect/QuantOps/QuantTypes.h" #include "mlir/Dialect/Quant/QuantTypes.h"
#include "mlir/IR/Builders.h" #include "mlir/IR/Builders.h"
#include "mlir/Quantizer/Configurations/FxpMathConfig.h" #include "mlir/Quantizer/Configurations/FxpMathConfig.h"
#include "mlir/Quantizer/Support/Configuration.h" #include "mlir/Quantizer/Support/Configuration.h"
#include "mlir/Quantizer/Support/ConstraintAnalysisGraph.h" #include "mlir/Quantizer/Support/ConstraintAnalysisGraph.h"
#include "mlir/Quantizer/Support/ConstraintAnalysisGraphTraits.h" #include "mlir/Quantizer/Support/ConstraintAnalysisGraphTraits.h"
#include "mlir/Quantizer/Transforms/Passes.h" #include "mlir/Quantizer/Transforms/Passes.h"
#include "mlir/Support/LogicalResult.h" #include "mlir/Support/LogicalResult.h"
#include "llvm/Support/DOTGraphTraits.h" #include "llvm/Support/DOTGraphTraits.h"
▲ Show 20 Lines • Show All 269 Lines • Show Last 20 Lines

mlir/lib/Quantizer/Transforms/RemoveInstrumentationPass.cpp

//===- RemoveInstrumentationPass.cpp - Removes instrumentation ------------===// //===- RemoveInstrumentationPass.cpp - Removes instrumentation ------------===//
// //
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information. // See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
// //
//===----------------------------------------------------------------------===// //===----------------------------------------------------------------------===//
// //
// This file defines a pass to remove any instrumentation ops. It is often one // This file defines a pass to remove any instrumentation ops. It is often one
// of the final steps when performing quantization and is run after any // of the final steps when performing quantization and is run after any
// decisions requiring instrumentation have been made. // decisions requiring instrumentation have been made.
// //
//===----------------------------------------------------------------------===// //===----------------------------------------------------------------------===//
#include "mlir/Dialect/QuantOps/QuantOps.h" #include "mlir/Dialect/Quant/QuantOps.h"
#include "mlir/IR/MLIRContext.h" #include "mlir/IR/MLIRContext.h"
#include "mlir/IR/PatternMatch.h" #include "mlir/IR/PatternMatch.h"
#include "mlir/Pass/Pass.h" #include "mlir/Pass/Pass.h"
#include "mlir/Quantizer/Transforms/Passes.h" #include "mlir/Quantizer/Transforms/Passes.h"
using namespace mlir; using namespace mlir;
using namespace mlir::quantizer; using namespace mlir::quantizer;
using namespace mlir::quant; using namespace mlir::quant;
▲ Show 20 Lines • Show All 45 Lines • Show Last 20 Lines

mlir/test/Dialect/Quant/canonicalize.mlir

  • This file was moved from mlir/test/Dialect/QuantOps/canonicalize.mlir.
The contents of this file were not changed.

mlir/test/Dialect/Quant/convert-const.mlir

  • This file was moved from mlir/test/Dialect/QuantOps/convert-const.mlir.
The contents of this file were not changed.

mlir/test/Dialect/Quant/convert-fakequant-invalid.mlir

  • This file was moved from mlir/test/Dialect/QuantOps/convert-fakequant-invalid.mlir.
The contents of this file were not changed.

mlir/test/Dialect/Quant/convert-fakequant.mlir

  • This file was moved from mlir/test/Dialect/QuantOps/convert-fakequant.mlir.
The contents of this file were not changed.

mlir/test/Dialect/Quant/parse-any-invalid.mlir

  • This file was moved from mlir/test/Dialect/QuantOps/parse-any-invalid.mlir.
The contents of this file were not changed.

mlir/test/Dialect/Quant/parse-any.mlir

  • This file was moved from mlir/test/Dialect/QuantOps/parse-any.mlir.
The contents of this file were not changed.

mlir/test/Dialect/Quant/parse-ops-invalid.mlir

  • This file was moved from mlir/test/Dialect/QuantOps/parse-ops-invalid.mlir.
The contents of this file were not changed.

mlir/test/Dialect/Quant/parse-ops.mlir

  • This file was moved from mlir/test/Dialect/QuantOps/parse-ops.mlir.
The contents of this file were not changed.

mlir/test/Dialect/Quant/parse-uniform-invalid.mlir

  • This file was moved from mlir/test/Dialect/QuantOps/parse-uniform-invalid.mlir.
The contents of this file were not changed.

mlir/test/Dialect/Quant/parse-uniform.mlir

  • This file was moved from mlir/test/Dialect/QuantOps/parse-uniform.mlir.
The contents of this file were not changed.

mlir/test/Dialect/Quant/quant_region.mlir

  • This file was moved from mlir/test/Dialect/QuantOps/quant_region.mlir.
The contents of this file were not changed.

mlir/test/Dialect/QuantOps/canonicalize.mlir

  • This file was moved to mlir/test/Dialect/Quant/canonicalize.mlir.

mlir/test/Dialect/QuantOps/convert-const.mlir

  • This file was moved to mlir/test/Dialect/Quant/convert-const.mlir.

mlir/test/Dialect/QuantOps/convert-fakequant-invalid.mlir

  • This file was moved to mlir/test/Dialect/Quant/convert-fakequant-invalid.mlir.

mlir/test/Dialect/QuantOps/convert-fakequant.mlir

  • This file was moved to mlir/test/Dialect/Quant/convert-fakequant.mlir.

mlir/test/Dialect/QuantOps/parse-any-invalid.mlir

  • This file was moved to mlir/test/Dialect/Quant/parse-any-invalid.mlir.

mlir/test/Dialect/QuantOps/parse-any.mlir

  • This file was moved to mlir/test/Dialect/Quant/parse-any.mlir.

mlir/test/Dialect/QuantOps/parse-ops-invalid.mlir

  • This file was moved to mlir/test/Dialect/Quant/parse-ops-invalid.mlir.

mlir/test/Dialect/QuantOps/parse-ops.mlir

  • This file was moved to mlir/test/Dialect/Quant/parse-ops.mlir.

mlir/test/Dialect/QuantOps/parse-uniform-invalid.mlir

  • This file was moved to mlir/test/Dialect/Quant/parse-uniform-invalid.mlir.

mlir/test/Dialect/QuantOps/parse-uniform.mlir

  • This file was moved to mlir/test/Dialect/Quant/parse-uniform.mlir.

mlir/test/Dialect/QuantOps/quant_region.mlir

  • This file was moved to mlir/test/Dialect/Quant/quant_region.mlir.

mlir/unittests/Dialect/Quant/QuantizationUtilsTest.cpp

  • This file was moved from mlir/unittests/Dialect/QuantOps/QuantizationUtilsTest.cpp.
//===- QuantizationUtilsTest.cpp - unit tests for quantization utils ------===// //===- QuantizationUtilsTest.cpp - unit tests for quantization utils ------===//
// //
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information. // See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
// //
//===----------------------------------------------------------------------===// //===----------------------------------------------------------------------===//
#include "mlir/Dialect/QuantOps/QuantOps.h" #include "mlir/Dialect/Quant/QuantOps.h"
#include "mlir/Dialect/QuantOps/QuantizeUtils.h" #include "mlir/Dialect/Quant/QuantizeUtils.h"
#include "mlir/Dialect/QuantOps/UniformSupport.h" #include "mlir/Dialect/Quant/UniformSupport.h"
#include "mlir/IR/Attributes.h" #include "mlir/IR/Attributes.h"
#include "mlir/IR/StandardTypes.h" #include "mlir/IR/StandardTypes.h"
#include "gmock/gmock.h" #include "gmock/gmock.h"
#include "gtest/gtest.h" #include "gtest/gtest.h"
using namespace mlir; using namespace mlir;
using namespace mlir::quant; using namespace mlir::quant;
▲ Show 20 Lines • Show All 149 Lines • Show Last 20 Lines

mlir/unittests/Dialect/QuantOps/QuantizationUtilsTest.cpp

  • This file was moved to mlir/unittests/Dialect/Quant/QuantizationUtilsTest.cpp.