28d04c564d
This is a preparation for adding support for more infrastructure around the concept of Action and make tracing Action more of a first class concept. The doc will be updated later in a subsequent revision after the changes are completed. Action belongs to IR because of circular dependency: Actions are dispatched through the MLIRContext but Action will learn to encapsulate IR construct. Differential Revision: https://reviews.llvm.org/D144809
44 lines
1.4 KiB
C++
44 lines
1.4 KiB
C++
//===- DebugCounterTest.cpp - Debug Counter Tests -------------------------===//
|
|
//
|
|
// 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
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "mlir/Debug/Counter.h"
|
|
#include "mlir/Support/TypeID.h"
|
|
#include "gmock/gmock.h"
|
|
|
|
using namespace mlir;
|
|
using namespace mlir::tracing;
|
|
|
|
namespace {
|
|
|
|
struct CounterAction : public ActionImpl<CounterAction> {
|
|
MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID(CounterAction)
|
|
static constexpr StringLiteral tag = "counter-action";
|
|
};
|
|
|
|
TEST(DebugCounterTest, CounterTest) {
|
|
std::unique_ptr<DebugCounter> counter = std::make_unique<DebugCounter>();
|
|
counter->addCounter(CounterAction::tag, /*countToSkip=*/1,
|
|
/*countToStopAfter=*/3);
|
|
|
|
ActionManager manager;
|
|
manager.registerActionHandler(std::move(counter));
|
|
|
|
auto noOp = []() { return; };
|
|
|
|
// The first execution is skipped.
|
|
EXPECT_FALSE(manager.execute<CounterAction>(noOp));
|
|
|
|
// The counter stops after 3 successful executions.
|
|
EXPECT_TRUE(manager.execute<CounterAction>(noOp));
|
|
EXPECT_TRUE(manager.execute<CounterAction>(noOp));
|
|
EXPECT_TRUE(manager.execute<CounterAction>(noOp));
|
|
EXPECT_FALSE(manager.execute<CounterAction>(noOp));
|
|
}
|
|
|
|
} // namespace
|