|
|
@ -16,10 +16,11 @@ static int run_cnt = 0;
|
|
|
|
|
|
|
|
|
|
|
|
class TestOp : public OperatorBase {
|
|
|
|
class TestOp : public OperatorBase {
|
|
|
|
public:
|
|
|
|
public:
|
|
|
|
void InferShape(const std::shared_ptr<pd::Scope>& scope) const override {
|
|
|
|
void InferShape(
|
|
|
|
|
|
|
|
const std::shared_ptr<framework::Scope>& scope) const override {
|
|
|
|
++infer_shape_cnt;
|
|
|
|
++infer_shape_cnt;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
void Run(const std::shared_ptr<pd::Scope>& scope,
|
|
|
|
void Run(const std::shared_ptr<framework::Scope>& scope,
|
|
|
|
const paddle::platform::DeviceContext& dev_ctx) const override {
|
|
|
|
const paddle::platform::DeviceContext& dev_ctx) const override {
|
|
|
|
++run_cnt;
|
|
|
|
++run_cnt;
|
|
|
|
}
|
|
|
|
}
|
|
|
|