atb/operation_infra.h
Definition
class OperationInfra : public Operation {
public:
OperationInfra();
OperationInfra(const OperationInfra &other);
OperationInfra& operator = (const OperationInfra &other);
~OperationInfra();
void SetExecuteStreamId(uint32_t streamId);
uint32_t GetExecuteStreamId() const;
aclrtStream GetExecuteStream(Context *context);
private:
std::unique_ptr<OperationImpl> impl_;
};
Parameters
Member |
Parameter |
Return Value |
Description |
|---|---|---|---|
OperationInfra |
- |
- |
Constructor. |
OperationInfra |
other |
- |
Copy constructor.
|
operator = |
other |
OperationInfra& |
Assignment constructor.
If the execution is successful, the object itself is returned. |
~OperationInfra |
- |
- |
Destructor. |
SetExecuteStreamId |
streamId |
- |
Used to set the ID of the stream used by the operation. The ID is the stream sequence number in the context. This API must be used together with the GetExecuteStream API.
|
GetExecuteStreamId |
- |
uint32_t |
Used to obtain the ID of the stream used by the current operation. If the execution is successful, the ID of the stream is returned. The ID is the stream sequence number in the context. |
GetExecuteStream |
context |
aclrtStream |
Used to obtain the stream used by the current operation.
If the execution is successful, the stream used by the current operation is returned. |