mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-06-26 03:00:54 +00:00
cf19c3697d
### Description
Run clang-format in CI. Formatted all c/c++, objective-c/c++ files.
Excluded
```
'onnxruntime/core/mlas/**',
'onnxruntime/contrib_ops/cuda/bert/tensorrt_fused_multihead_attention/**',
```
because they contain assembly or is data heavy
### Motivation and Context
Coding style consistency
88 lines
2.6 KiB
C++
88 lines
2.6 KiB
C++
// Copyright (c) Microsoft Corporation. All rights reserved.
|
|
// Licensed under the MIT License.
|
|
#pragma once
|
|
|
|
#include "core/framework/sequential_execution_plan.h"
|
|
#include "core/framework/stream_execution_context.h"
|
|
|
|
namespace onnxruntime {
|
|
|
|
class SessionScope;
|
|
|
|
class BarrierStep : public SequentialExecutionPlan::ExecutionStep {
|
|
public:
|
|
BarrierStep(size_t id, NodeIndex node_index);
|
|
|
|
Status Execute(StreamExecutionContext& ctx,
|
|
size_t /*stream_idx*/,
|
|
SessionScope& /*session_scope*/,
|
|
const bool& /*terminate_flag*/,
|
|
bool& continue_flag) override;
|
|
|
|
std::string ToString() const override;
|
|
|
|
private:
|
|
size_t barrier_id_{0};
|
|
};
|
|
|
|
class WaitOnEPStep : public SequentialExecutionPlan::ExecutionStep {
|
|
public:
|
|
WaitOnEPStep(WaitNotificationFn handle, NotificationIndex idx, NodeIndex node_index);
|
|
|
|
Status Execute(StreamExecutionContext& ctx,
|
|
size_t stream_idx,
|
|
SessionScope& /*session_scope*/,
|
|
const bool& /*terminate_flag*/,
|
|
bool& continue_flag) override;
|
|
|
|
std::string ToString() const override;
|
|
|
|
private:
|
|
WaitNotificationFn wait_handle_;
|
|
NotificationIndex notification_idx_;
|
|
};
|
|
|
|
class LaunchKernelStep : public SequentialExecutionPlan::ExecutionStep {
|
|
public:
|
|
LaunchKernelStep(NodeIndex index);
|
|
|
|
Status Execute(StreamExecutionContext& ctx,
|
|
size_t stream_idx,
|
|
SessionScope& session_scope,
|
|
const bool& terminate_flag,
|
|
bool& continue_flag) override;
|
|
|
|
std::string ToString() const override;
|
|
};
|
|
|
|
class ActivateNotificationStep : public SequentialExecutionPlan::ExecutionStep {
|
|
public:
|
|
ActivateNotificationStep(NotificationIndex notification_index, NodeIndex node_index);
|
|
|
|
Status Execute(StreamExecutionContext& ctx,
|
|
size_t stream_idx,
|
|
SessionScope& /*session_scope*/,
|
|
const bool& /*terminate_flag*/,
|
|
bool& continue_flag) override;
|
|
|
|
virtual std::string ToString() const override;
|
|
|
|
private:
|
|
NotificationIndex notification_idx_;
|
|
};
|
|
|
|
class TriggerDownstreamStep : public SequentialExecutionPlan::ExecutionStep {
|
|
public:
|
|
TriggerDownstreamStep(size_t trigger_point_index, NodeIndex node_index);
|
|
Status Execute(StreamExecutionContext& ctx,
|
|
size_t /*stream_idx*/,
|
|
SessionScope& session_scope,
|
|
const bool& terminate_flag,
|
|
bool& continue_flag) override;
|
|
|
|
virtual std::string ToString() const override;
|
|
|
|
private:
|
|
size_t trigger_point_index_;
|
|
};
|
|
} // namespace onnxruntime
|